From cea4f9350c574a0b274ec967b3f7f493cd8c6374 Mon Sep 17 00:00:00 2001 From: ben Date: Wed, 23 Nov 2022 22:42:32 +0000 Subject: [PATCH] black --- lnbits/core/views/api.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/lnbits/core/views/api.py b/lnbits/core/views/api.py index b7bc7084b..2cc7aacbf 100644 --- a/lnbits/core/views/api.py +++ b/lnbits/core/views/api.py @@ -701,6 +701,7 @@ async def api_auditor(wallet: WalletTypeInfo = Depends(get_key_type)): ##################UNIVERSAL WEBSOCKET MANAGER######################## + class websocketConnectionManager: def __init__(self): self.active_connections: List[WebSocket] = [] @@ -718,8 +719,10 @@ class websocketConnectionManager: if connection.id == item_id: await connection.send_text(message) + manager = websocketConnectionManager() + @core_app.websocket("/api/v1/ws/{item_id}") async def websocket_connect(websocket: WebSocket, item_id: str): await manager.connect(websocket, item_id) @@ -729,6 +732,7 @@ async def websocket_connect(websocket: WebSocket, item_id: str): except WebSocketDisconnect: manager.disconnect(websocket) + @core_app.post("/api/v1/ws/{item_id}") async def websocket_update(item_id: str, data: str): try: @@ -737,6 +741,7 @@ async def websocket_update(item_id: str, data: str): except: return {"sent": False, "data": data} + @core_app.get("/api/v1/ws/{item_id}/{data}") async def websocket_update(item_id: str, data: str): try: @@ -745,7 +750,6 @@ async def websocket_update(item_id: str, data: str): except: return {"sent": False, "data": data} + async def updater(item_id, data): - return await manager.send_data( - f"{data}", item_id - ) \ No newline at end of file + return await manager.send_data(f"{data}", item_id)