Father, forgive me, for I am adding inline types. (#266)

* Father, forgive me, for I am adding inline types.

* chore: sort imports

Author: sansyrox

robyn/__init__.py

@@ -3,16 +3,17 @@
 import multiprocessing as mp
 import os
 import sys
+from typing import Callable, Optional
 
 from multiprocess import Process
 from watchdog.observers import Observer
-from robyn.events import Events
+
 from robyn.argument_parser import ArgumentParser
 from robyn.dev_event_handler import EventHandler
+from robyn.events import Events
 from robyn.log_colors import Colors
 from robyn.processpool import spawn_process
 from robyn.responses import jsonify, static_file
-
 from robyn.robyn import SocketHeld
 from robyn.router import MiddlewareRouter, Router, WebSocketRouter
 from robyn.ws import WS
@@ -23,7 +24,7 @@
 class Robyn:
     """This is the python wrapper for the Robyn binaries."""
 
-    def __init__(self, file_object):
+    def __init__(self, file_object: str) -> None:
         directory_path = os.path.dirname(os.path.abspath(file_object))
         self.file_path = file_object
         self.directory_path = directory_path
@@ -54,7 +55,7 @@ def _add_route(self, route_type, endpoint, handler, const=False):
         """
         self.router.add_route(route_type, endpoint, handler, const)
 
-    def before_request(self, endpoint):
+    def before_request(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.before_request decorator to add a get route
 
@@ -63,7 +64,7 @@ def before_request(self, endpoint):
 
         return self.middleware_router.add_before_request(endpoint)
 
-    def after_request(self, endpoint):
+    def after_request(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.after_request decorator to add a get route
 
@@ -73,31 +74,31 @@ def after_request(self, endpoint):
         return self.middleware_router.add_after_request(endpoint)
 
     def add_directory(
-        self, route, directory_path, index_file=None, show_files_listing=False
+        self, route: str, directory_path: str, index_file: Optional[str] = None, show_files_listing: bool = False
     ):
         self.directories.append((route, directory_path, index_file, show_files_listing))
 
-    def add_header(self, key, value):
+    def add_header(self, key: str, value: str) -> None:
         self.headers.append((key, value))
 
-    def add_web_socket(self, endpoint, ws):
+    def add_web_socket(self, endpoint: str, ws: WS) -> None:
         self.web_socket_router.add_route(endpoint, ws)
 
-    def _add_event_handler(self, event_type: str, handler):
+    def _add_event_handler(self, event_type: str, handler) -> None:
         logger.debug(f"Add event {event_type} handler")
         if event_type not in {Events.STARTUP, Events.SHUTDOWN}:
             return
 
         is_async = asyncio.iscoroutinefunction(handler)
         self.event_handlers[event_type] = (handler, is_async)
 
-    def startup_handler(self, handler):
+    def startup_handler(self, handler: Callable) -> None:
         self._add_event_handler(Events.STARTUP, handler)
 
-    def shutdown_handler(self, handler):
+    def shutdown_handler(self, handler: Callable) -> None:
         self._add_event_handler(Events.SHUTDOWN, handler)
 
-    def start(self, url="127.0.0.1", port=5000):
+    def start(self, url: str = "127.0.0.1", port: int = 5000):
         """
         Starts the server
 
@@ -157,7 +158,6 @@ def init_processpool(socket):
                 logger.info(f"{Colors.BOLD}{Colors.OKGREEN} Terminating server!! {Colors.ENDC}")
                 for process in process_pool:
                     process.kill()
-
         else:
             event_handler = EventHandler(self.file_path)
             event_handler.start_server_first_time()
@@ -174,7 +174,7 @@ def init_processpool(socket):
                 observer.stop()
                 observer.join()
 
-    def get(self, endpoint, const=False):
+    def get(self, endpoint: str, const: bool = False) -> Callable[..., None]:
         """
         The @app.get decorator to add a get route
 
@@ -186,7 +186,7 @@ def inner(handler):
 
         return inner
 
-    def post(self, endpoint):
+    def post(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.post decorator to add a get route
 
@@ -198,7 +198,7 @@ def inner(handler):
 
         return inner
 
-    def put(self, endpoint):
+    def put(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.put decorator to add a get route
 
@@ -209,7 +209,7 @@ def inner(handler):
 
         return inner
 
-    def delete(self, endpoint):
+    def delete(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.delete decorator to add a get route
 
@@ -221,7 +221,7 @@ def inner(handler):
 
         return inner
 
-    def patch(self, endpoint):
+    def patch(self, endpoint: str) -> Callable[..., None]:
         """
         [The @app.patch decorator to add a get route]
 
@@ -233,7 +233,7 @@ def inner(handler):
 
         return inner
 
-    def head(self, endpoint):
+    def head(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.head decorator to add a get route
 
@@ -245,7 +245,7 @@ def inner(handler):
 
         return inner
 
-    def options(self, endpoint):
+    def options(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.options decorator to add a get route
 
@@ -257,7 +257,7 @@ def inner(handler):
 
         return inner
 
-    def connect(self, endpoint):
+    def connect(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.connect decorator to add a get route
 
@@ -269,7 +269,7 @@ def inner(handler):
 
         return inner
 
-    def trace(self, endpoint):
+    def trace(self, endpoint: str) -> Callable[..., None]:
         """
         The @app.trace decorator to add a get route
 

robyn/__init__.pyi

@@ -2,7 +2,7 @@
 
 
 class ArgumentParser(argparse.ArgumentParser):
-    def __init__(self):
+    def __init__(self) -> None:
         self.parser = argparse.ArgumentParser(
             description="Robyn, a fast async web framework with a rust runtime."
         )
@@ -38,19 +38,19 @@ def __init__(self):
         self.args = self.parser.parse_args()
 
     @property
-    def num_processes(self):
+    def num_processes(self) -> int:
         return self.args.processes
 
     @property
-    def workers(self):
+    def workers(self) -> int:
         return self.args.workers
 
     @property
-    def log_level(self):
+    def log_level(self) -> str:
         return self.args.log_level
 
     @property
-    def is_dev(self):
+    def is_dev(self) -> bool:
         _is_dev = self.args.dev
         if _is_dev and (self.num_processes != 1 or self.workers != 1):
             raise Exception("--processes and --workers shouldn't be used with --dev")

robyn/argument_parser.py

@@ -4,16 +4,16 @@
 
 
 class EventHandler(FileSystemEventHandler):
-    def __init__(self, file_name):
+    def __init__(self, file_name) -> None:
         self.file_name = file_name
         self.processes = []
 
-    def start_server_first_time(self):
+    def start_server_first_time(self) -> None:
         if self.processes:
             raise Exception("Something wrong with the server")
         self.processes.append(subprocess.Popen(["python3", self.file_name], start_new_session=False))
 
-    def on_any_event(self, event):
+    def on_any_event(self, event) -> None:
         """
         This function is a callback that will start a new server on every even change