Improve type hints

Author: jason0x43

hubitatmaker/hub.py

@@ -43,9 +43,9 @@ def __init__(
         host: str,
         app_id: str,
         access_token: str,
-        port: int = None,
-        event_url: str = None,
-        ssl_context: SSLContext = None
+        port: Optional[int] = None,
+        event_url: Optional[str] = None,
+        ssl_context: Optional[SSLContext] = None,
     ):
         """Initialize a Hubitat hub interface.
 
@@ -274,10 +274,10 @@ async def set_port(self, port: int) -> None:
         if self._server:
             self._server.stop()
         await self._start_server()
-    
+
     async def set_ssl_context(self, ssl_context: Optional[SSLContext]) -> None:
         """Set the SSLContext that the event listener server will use. Passing in a SSLContext object
-        will make the event listener server HTTPS only. Passing in None will revert the server back 
+        will make the event listener server HTTPS only. Passing in None will revert the server back
         to HTTP.
 
         Setting this will stop and restart the event listener server.
@@ -422,10 +422,10 @@ async def _start_server(self) -> None:
         )
         self._server.start()
         _LOGGER.debug(
-            "Listening on %s:%d with SSL %s", 
-            address, 
-            self._server.port, 
-            "disabled" if self.ssl_context is None else "enabled"
+            "Listening on %s:%d with SSL %s",
+            address,
+            self._server.port,
+            "disabled" if self.ssl_context is None else "enabled",
         )
 
         await self.set_event_url(self.event_url)

hubitatmaker/server.py

@@ -3,7 +3,7 @@
 from socket import socket as Socket
 from ssl import SSLContext
 import threading
-from typing import Any, Callable, Dict, List, cast
+from typing import Any, Callable, Dict, List, Optional, cast
 
 from aiohttp import web
 
@@ -13,7 +13,13 @@
 class Server:
     """A handle to a running server."""
 
-    def __init__(self, handle_event: EventCallback, host: str, port: int, ssl_context: SSLContext = None):
+    def __init__(
+        self,
+        handle_event: EventCallback,
+        host: str,
+        port: int,
+        ssl_context: Optional[SSLContext] = None,
+    ):
         """Initialize a Server."""
         self.host = host
         self.port = port
@@ -63,7 +69,9 @@ def _run(self) -> None:
         asyncio.set_event_loop(self._server_loop)
         self._server_loop.run_until_complete(self._runner.setup())
 
-        site = web.TCPSite(self._runner, self.host, self.port, ssl_context=self.ssl_context)
+        site = web.TCPSite(
+            self._runner, self.host, self.port, ssl_context=self.ssl_context
+        )
         self._server_loop.run_until_complete(site.start())
 
         # If the Server was initialized with port 0, determine what port the
@@ -84,7 +92,10 @@ async def _stop(self) -> None:
 
 
 def create_server(
-    handle_event: EventCallback, host: str = "0.0.0.0", port: int = 0, ssl_context: SSLContext = None
+    handle_event: EventCallback,
+    host: str = "0.0.0.0",
+    port: int = 0,
+    ssl_context: Optional[SSLContext] = None,
 ) -> Server:
     """Create a new server."""
     return Server(handle_event, host, port, ssl_context)