laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

review changes

Browse Source
This commit is contained in:
Lendemor 2025-01-09 11:22:40 +01:00
parent 41d8cfae57
commit 2825500e26
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
reflex/app.py
View File

@ -15,6 +15,7 @@ import multiprocessing
import platform import platform
import sys import sys
import traceback import traceback
import urllib.parse
import uuid import uuid
from datetime import datetime from datetime import datetime
from pathlib import Path from pathlib import Path
@ -1536,10 +1537,7 @@ class EventNamespace(AsyncNamespace):
sid: The Socket.IO session id. sid: The Socket.IO session id.
environ: The request information, including HTTP headers. environ: The request information, including HTTP headers.
""" """
query_string = environ.get("QUERY_STRING") query_params = urllib.parse.parse_qs(environ.get("QUERY_STRING"))
query_params = dict(
qc.split("=") for qc in query_string.split("&") if "=" in qc
)
await self.link_token_to_sid(sid, query_params.get("token")) await self.link_token_to_sid(sid, query_params.get("token"))
def on_disconnect(self, sid): def on_disconnect(self, sid):
@ -1621,7 +1619,7 @@ class EventNamespace(AsyncNamespace):
token: The client token. token: The client token.
""" """
if token in self.sid_to_token.values() and sid != self.token_to_sid.get(token): if token in self.sid_to_token.values() and sid != self.token_to_sid.get(token):
token = uuid.uuid4().hex token = str(uuid.uuid4())
await self.emit("new_token", token, to=sid) await self.emit("new_token", token, to=sid)
self.token_to_sid[token] = sid self.token_to_sid[token] = sid