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

separate the logic

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2025-01-27 13:47:37 -08:00
parent f4a0403d71
commit dfceedc4e3
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
reflex/app.py
View File

@ -418,20 +418,17 @@ class App(MiddlewareMixin, LifespanMixin):
async def modified_send(message):
if message["type"] == "websocket.accept":
subprotocols = scope.get("subprotocols", [])
if not subprotocols:
headers = scope.get("headers", {})
headers_key = b"sec-websocket-protocol"
if headers_key in headers:
subprotocols = (
headers[headers_key].decode().split(", ")
)
else:
subprotocols = None
else:
subprotocols = subprotocols[0]
if scope.get("subprotocols"):
# The following *does* say "subprotocol" instead of "subprotocols", intentionally.
message["subprotocol"] = scope["subprotocols"][0]
message["subprotocol"] = subprotocols
headers = dict(message.get("headers", []))
header_key = b"sec-websocket-protocol"
if subprotocol := headers.get(header_key):
message["headers"] = [
*message.get("headers", []),
(header_key, subprotocol),
]
return await original_send(message)