WS Server

ws_server.py

@@ -0,0 +1,51 @@
+import asyncio
+import logging
+
+import websockets
+from websockets import WebSocketServerProtocol, broadcast
+
+from settings import settings
+
+connected_clients = set()
+
+
+def setup_logging() -> None:
+    logging.basicConfig(
+        level=logging.DEBUG,
+        format="%(asctime)s %(levelname)s - %(message)s",
+    )
+
+
+async def ws_handler(websocket: WebSocketServerProtocol):
+    connected_clients.add(websocket)
+    logging.info(f"Add client: {websocket}")
+
+    try:
+        async for message in websocket:
+            broadcast(connected_clients, message)
+            # await websocket.send(f"Are you talking to me? {message}")
+    finally:
+        connected_clients.remove(websocket)
+        logging.info(f"Remove client: {websocket}")
+
+
+async def main():
+    setup_logging()
+
+    logging.info(
+        f"Starting FairHopper Websockets Server on {settings.ws_server.HOST}:{settings.ws_server.PORT}"
+    )
+
+    async with websockets.serve(
+        ws_handler=ws_handler,
+        host=settings.ws_server.HOST,
+        port=settings.ws_server.PORT,
+    ):
+        await asyncio.Future()  # run forever
+
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except (KeyboardInterrupt, SystemExit):
+        logging.info(f"FairHopper Websockets Server terminated")