link and flake cleanup

hbd/ws.py

@@ -3,6 +3,7 @@
 Provides an asyncio-based WebSocket server and a thread-safe broadcast
 function that other threads or synchronous code can call.
 """
+
 import asyncio
 import json
 import logging
@@ -20,7 +21,6 @@ _verbose = False
 
 
 async def _handler(websocket, path=None):
-    global _connections
     _connections.add(websocket)
     remote_address = websocket.remote_address
     if path is None:
@@ -46,7 +46,10 @@ async def _handler(websocket, path=None):
             if _verbose:
                 logger.debug("received ws data: %s", _)
 
-    except (websockets.exceptions.ConnectionClosedOK, websockets.exceptions.ConnectionClosedError) as e:
+    except (
+        websockets.exceptions.ConnectionClosedOK,
+        websockets.exceptions.ConnectionClosedError,
+    ) as e:
         if _verbose:
             logger.info("ws closed: %r", e)
     except Exception as e:
@@ -59,7 +62,15 @@ async def _handler(websocket, path=None):
         await websocket.wait_closed()
 
 
-async def start(host: str, ws_port: int, wss_port: Optional[int] = None, ssl_context=None, get_hosts: Optional[Callable] = None, get_msgs: Optional[Callable] = None, verbose: bool = False):
+async def start(
+    host: str,
+    ws_port: int,
+    wss_port: Optional[int] = None,
+    ssl_context=None,
+    get_hosts: Optional[Callable] = None,
+    get_msgs: Optional[Callable] = None,
+    verbose: bool = False,
+):
     """Start WebSocket servers and block until cancelled.
 
     This is intended to be awaited inside the main asyncio event loop.
@@ -77,11 +88,13 @@ async def start(host: str, ws_port: int, wss_port: Optional[int] = None, ssl_con
     websockets_logger = logging.getLogger("websockets.server")
     websockets_logger.setLevel(logging.DEBUG if verbose else logging.INFO)
     # regular WebSocket
-    ws_server = websockets.serve(_handler, host, ws_port) #, subprotocols=["hbd"])
+    ws_server = websockets.serve(_handler, host, ws_port)  # , subprotocols=["hbd"])
     servers.append(ws_server)
     # secure WebSocket (optional)
     if wss_port and ssl_context:
-        wss_server = websockets.serve(_handler, host, wss_port, ssl=ssl_context ) #, subprotocols=["hbd"])
+        wss_server = websockets.serve(
+            _handler, host, wss_port, ssl=ssl_context
+        )  # , subprotocols=["hbd"])
         servers.append(wss_server)
 
     # await starting of all servers
@@ -89,7 +102,9 @@ async def start(host: str, ws_port: int, wss_port: Optional[int] = None, ssl_con
         await srv
 
     if _verbose:
-        logger.info("WebSocket server(s) started on port %s (wss %s)", ws_port, wss_port)
+        logger.info(
+            "WebSocket server(s) started on port %s (wss %s)", ws_port, wss_port
+        )
 
     # block forever (until loop is stopped or cancelled)
     await asyncio.Future()
@@ -101,8 +116,6 @@ def broadcast(typ: str, data) -> bool:
     Schedules coroutine(s) on the running loop to send message to all
     connected websockets. Returns False if server was not running.
     """
-    global _loop
-    
     if not _loop:
         return False
     jmsg = json.dumps({"type": typ, "data": data})