jprdonnelly/pyscript
1
0
Fork 0
mirror of https://github.com/pyscript/pyscript.git synced 2025-12-19 10:17:23 -05:00
Code Issues Projects Releases Wiki Activity

Make WebSocket lazy attributes definition possible (#2180)

Browse Source
This commit is contained in:
Andrea Giammarchi
2024-09-23 13:19:26 +02:00
committed by GitHub
parent a6b0964185
commit 5aaeebf32c
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pyscript.core/src/stdlib/pyscript/websocket.py
View File

@@ -5,6 +5,7 @@ from pyscript.util import as_bytearray
code = "code"
protocols = "protocols"
reason = "reason"
methods = ["onclose", "onerror", "onmessage", "onopen"]
class EventMessage:
@@ -37,7 +38,7 @@ class WebSocket(object):
socket = js.WebSocket.new(url)
object.__setattr__(self, "_ws", socket)
for t in ["onclose", "onerror", "onmessage", "onopen"]:
for t in methods:
if t in kw:
# Pyodide fails at setting socket[t] directly
setattr(socket, t, create_proxy(kw[t]))
@@ -46,10 +47,11 @@ class WebSocket(object):
return getattr(self._ws, attr)
def __setattr__(self, attr, value):
if attr == "onmessage":
self._ws[attr] = lambda e: value(EventMessage(e))
if attr in methods:
m = lambda e: value(EventMessage(e))
setattr(self._ws, attr, create_proxy(m))
else:
self._ws[attr] = value
setattr(self._ws, attr, value)
def close(self, **kw):
if code in kw and reason in kw: