discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

fixed-replyto

Browse Source
This commit is contained in:
tonyrewin
2022-12-12 09:50:53 +03:00
parent c812db810b
commit 4d1759c188
5 changed files with 35 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
services/inbox/presence.py
View File

@@ -11,9 +11,9 @@ async def set_online_status(user_id, status):
await redis.execute("SREM", "users-online", user_id)
async def on_connect(websocket, params):
async def on_connect(req, params):
if not isinstance(params, dict):
websocket.scope["connection_params"] = {}
req.scope["connection_params"] = {}
return
token = params.get('token')
if not token:
@@ -21,12 +21,12 @@ async def on_connect(websocket, params):
else:
payload = await SessionToken.verify(token)
if payload and payload.user_id:
websocket.scope["user_id"] = payload.user_id
req.scope["user_id"] = payload.user_id
await set_online_status(payload.user_id, True)
async def on_disconnect(websocket):
user_id = websocket.scope.get("user_id")
async def on_disconnect(req):
user_id = req.scope.get("user_id")
await set_online_status(user_id, False)