diff --git a/app.py b/app.py index fbf2f5f..2b22391 100644 --- a/app.py +++ b/app.py @@ -26,11 +26,11 @@ async def set_variable(project_id,var): if not vars.get(project_id,''): vars[project_id] = {} vars[project_id][var.get('name','')] = var.get('value','') - await websockets.broadcast(clients.get(project_id,[]),json.dumps(var)) + websockets.broadcast(clients.get(project_id,[]),json.dumps(var)) -async def disconnect(project_id,user): - print("disconnecting",user) - clients.get(project_id,[]).pop(user) +async def disconnect(project_id,ws): + print(f"disconnecting {ws.host}:{ws.port}") + clients.get(project_id,[]).remove(ws) async def process(ws, path):