vuer 0.0.18__py3-none-any.whl → 0.0.19__py3-none-any.whl

vuer/server.py

@@ -2,10 +2,12 @@ import asyncio
 from asyncio import sleep
 from collections import deque, defaultdict
 from functools import partial
-from typing import cast, Callable, Coroutine
+from typing import cast, Callable, Coroutine, Dict
 from uuid import uuid4
 
+from aiohttp.web_request import Request
 from aiohttp.web_response import Response
+from aiohttp.web_ws import WebSocketResponse
 from msgpack import packb, unpackb
 from params_proto import Proto, PrefixProto
 from websockets import ConnectionClosedError
@@ -55,6 +57,24 @@ class VuerSession:
         self.downlink_queue = que_maker()
         self.uplink_queue = que_maker()
 
+    @property
+    def socket(self):
+        """Getter for the websocket object.
+
+        this is useful for closing the socket session from the client side.
+
+        Example Usage::
+
+            @app.spawn(start=True):
+            async def main(session: VuerSession):
+                print("doing something...")
+                await sleep(1.0)
+
+                print("I am done! closing the socket.")
+                session.socket.close()
+        """
+        return self.vuer.ws[self.CURRENT_WS_ID]
+
     def __matmul__(self, event: ServerEvent):
         """
         Send a ServerEvent to the client.
@@ -294,7 +314,7 @@ class Vuer(PrefixProto, Server):
         # todo: can remove
         self.page = Page()
 
-        self.ws = {}
+        self.ws: Dict[str, WebSocketResponse] = {}
         self.socket_handler: SocketHandler = None
         self.spawned_coroutines = []
 
@@ -510,7 +530,7 @@ class Vuer(PrefixProto, Server):
             else:
                 await sleep(0.0)
 
-    async def downlink(self, request, ws):
+    async def downlink(self, request: Request, ws: WebSocketResponse):
         """
         The websocket handler for receiving messages from the client.
 
@@ -530,7 +550,17 @@ class Vuer(PrefixProto, Server):
         self._add_task(self.uplink(vuer_proxy))
 
         if self.socket_handler is not None:
-            task = self._add_task(self.socket_handler(vuer_proxy))
+
+            async def handler():
+                try:
+                    await self.socket_handler(vuer_proxy)
+                except Exception:
+                    await self.close_ws(ws_id)
+                    return
+
+                await self.close_ws(ws_id)
+
+            task = self._add_task(handler())
 
         if hasattr(generator, "__anext__"):
             serverEvent = await generator.__anext__()

{vuer-0.0.18.dist-info → vuer-0.0.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vuer
-Version: 0.0.18
+Version: 0.0.19
 Home-page: https://github.com/geyang/vuer
 Author: Ge Yang<ge.ike.yang@gmail.com>
 Author-email: ge.ike.yang@gmail.com

{vuer-0.0.18.dist-info → vuer-0.0.19.dist-info}/RECORD

@@ -3,14 +3,14 @@ vuer/base.py,sha256=R-PtD39ouihiEGmSfJL26chlU0XkFJP-JmBNYzDdEvM,3068
 vuer/events.py,sha256=8AAzkfbm5jHkeMh17Z57n8YTxF4mheWC3Qbya4XkQSc,7291
 vuer/schemas.py,sha256=aZOocE02gO43SLLfeYSxxac0x38TmyTcfs_iFh5hsJ4,18152
 vuer/serdes.py,sha256=gD2iA9Yypu1QjocneOT3Nc0y6q_mdHn9zW1ko5j3I7c,2600
-vuer/server.py,sha256=OsXSq6iScFTG-yvAXWAik-EwNwEKdd1F84nUqBwyrjE,19955
+vuer/server.py,sha256=gIYbWwc-25JwtK9eeojQJtPSRMrCZ8rRcmZToDMnrpg,20851
 vuer/types.py,sha256=N2KLa0fl6z8Jm6cJwZoD8Vqrins_AayG5BCGk1p3Eek,1279
 vuer/__pycache__/__init__.cpython-38.pyc,sha256=rS-9zzAjFEXnObk5q3E5Rex_uvoJ-Ccfz7Hs9DGNVgs,188
 vuer/__pycache__/base.cpython-38.pyc,sha256=Z5EFlRX9RqZsQUszvqPtAhLv_5QG5wlUXW7JbFd_XDc,3824
 vuer/__pycache__/events.cpython-38.pyc,sha256=zx3bXeJixqOyCFe2nA7qpq6jiCJ49kRaXO-xONGUyeQ,9616
 vuer/__pycache__/schemas.cpython-38.pyc,sha256=hvY9Aak8zE-zKcWiwuNe6DghOw9qH_zSe_FtkBOAPPk,23234
 vuer/__pycache__/serdes.cpython-38.pyc,sha256=KMxTjPEWuSGn2bqBAl5OLIDSCSoqfPDGfk3fvNnRDYA,2253
-vuer/__pycache__/server.cpython-38.pyc,sha256=kAsC8t-JgTOOccZMaXZMEqgTSV3C3T5Oyj1sl-QABM4,18696
+vuer/__pycache__/server.cpython-38.pyc,sha256=z44pZ0u3x6ZLwDOh2skBFlvNEnGO2re4cdhRAippzy0,19636
 vuer/__pycache__/types.cpython-38.pyc,sha256=IhlXtkT-XWM0V1420FDuoqIYnpvRvekYVkGqEK7fAV8,1819
 vuer/addons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vuer/addons/nerf_vuer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -29,9 +29,9 @@ vuer/schemas/__pycache__/__init__.cpython-38.pyc,sha256=NRnVrpIDBKne93xOUY1-Wpav
 vuer/schemas/__pycache__/drei_components.cpython-38.pyc,sha256=g_ufcKxf-XKfZLdUV-HqKnjIrgxGWFv51aHLWQgH-ws,7712
 vuer/schemas/__pycache__/html_components.cpython-38.pyc,sha256=q0DMFwNkYbnaH1A8w3BowMiQAlmGpFWOOKsjLVE6CIk,8215
 vuer/schemas/__pycache__/scene_components.cpython-38.pyc,sha256=eRxB-Tp24yqBR4zbLMGX_4o1P_D-tW1nrDeBPSJKoU4,18135
-vuer-0.0.18.dist-info/LICENSE,sha256=MGF-inVBUaGe2mEjqT0g6XsHIXwoNXgNHqD7Z1MzR0k,1063
-vuer-0.0.18.dist-info/METADATA,sha256=cbpGK1fH2Wx_FmSLe_X9M59Dk0E_V-7v5FCS-KRfvx0,4277
-vuer-0.0.18.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-vuer-0.0.18.dist-info/entry_points.txt,sha256=J_NM6fbpipmD9oP7cdxd1UyBR8mVEQVx0xjlE_56yss,41
-vuer-0.0.18.dist-info/top_level.txt,sha256=ermmVkwvGFAK4gfSgDIwOmKpxwpqNt-oo7gVQQUSHok,5
-vuer-0.0.18.dist-info/RECORD,,
+vuer-0.0.19.dist-info/LICENSE,sha256=MGF-inVBUaGe2mEjqT0g6XsHIXwoNXgNHqD7Z1MzR0k,1063
+vuer-0.0.19.dist-info/METADATA,sha256=IY8XfDtcj6-Z8xww_0qN2r9Nh-1_j01_63eGS2OLoto,4277
+vuer-0.0.19.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+vuer-0.0.19.dist-info/entry_points.txt,sha256=J_NM6fbpipmD9oP7cdxd1UyBR8mVEQVx0xjlE_56yss,41
+vuer-0.0.19.dist-info/top_level.txt,sha256=ermmVkwvGFAK4gfSgDIwOmKpxwpqNt-oo7gVQQUSHok,5
+vuer-0.0.19.dist-info/RECORD,,