wslink 2.2.1__py3-none-any.whl → 2.3.0__py3-none-any.whl

wslink/backends/aiohttp/__init__.py

@@ -23,6 +23,8 @@ HTTP_HEADERS: str | None = os.environ.get("WSLINK_HTTP_HEADERS")  # path to json
 if HTTP_HEADERS and Path(HTTP_HEADERS).exists():
     HTTP_HEADERS: dict = json.loads(Path(HTTP_HEADERS).read_text())
 
+STATE_KEY = aiohttp_web.AppKey("state", str)
+
 logger = logging.getLogger(__name__)
 
 # -----------------------------------------------------------------------------
@@ -92,7 +94,7 @@ class WebAppServer(AbstractWebApp):
             self.app.router.add_route("GET", "/", _root_handler)
             self.app.add_routes(routes)
 
-        self.app["state"] = {}
+        self.app[STATE_KEY] = {}
 
     # -------------------------------------------------------------------------
     # Server status

wslink/backends/generic/core.py

@@ -56,7 +56,7 @@ class WsConnection:
 
 
 class WsEndpoint(WslinkHandler):
-    def __init__(self, protocol=None, web_app=None):
+    def __init__(self, protocol, web_app=None):
         super().__init__(protocol, web_app)
 
     async def connect(self):

wslink/backends/jupyter/core.py

@@ -1,49 +1,9 @@
-import asyncio
 from functools import partial
+from wslink.emitter import EventEmitter
 from wslink.backends.generic.core import GenericServer
 from IPython.core.getipython import get_ipython
 
 
-class EventEmitter:
-    def __init__(self):
-        self._listeners = {}
-
-    def clear(self):
-        self._listeners = {}
-
-    def emit(self, event, *args, **kwargs):
-        listeners = self._listeners.get(event)
-        if listeners is None:
-            return
-
-        loop = asyncio.get_running_loop()
-        coroutine_run = (
-            loop.create_task if (loop and loop.is_running()) else asyncio.run
-        )
-
-        for listener in listeners:
-            if asyncio.iscoroutinefunction(listener):
-                coroutine_run(listener(*args, **kwargs))
-            else:
-                listener(*args, **kwargs)
-
-    def add_event_listener(self, event, listener):
-        listeners = self._listeners.get(event)
-        if listeners is None:
-            listeners = set()
-            self._listeners[event] = listeners
-
-        listeners.add(listener)
-
-    def remove_event_listener(self, event, listener):
-        listeners = self._listeners.get(event)
-        if listeners is None:
-            return
-
-        if listener in listeners:
-            listeners.remove(listener)
-
-
 class WsJupyterComm(EventEmitter):
     def __init__(self, kernel=None):
         super().__init__()

wslink/emitter.py

@@ -0,0 +1,63 @@
+from typing import TypeVar, Generic, LiteralString
+
+import asyncio
+import functools
+
+
+T = TypeVar("T", bound=LiteralString)
+
+
+class EventEmitter(Generic[T]):
+    def __init__(self):
+        self._listeners = {}
+
+    def clear(self):
+        self._listeners = {}
+
+    def __call__(self, event: T, *args, **kwargs):
+        self.emit(event, *args, **kwargs)
+
+    def __getattr__(self, name: T):
+        return functools.partial(self.emit, name)
+
+    def emit(self, event: T, *args, **kwargs):
+        listeners = self._listeners.get(event)
+        if listeners is None:
+            return
+
+        loop = asyncio.get_running_loop()
+        coroutine_run = (
+            loop.create_task if (loop and loop.is_running()) else asyncio.run
+        )
+
+        for listener in listeners:
+            if asyncio.iscoroutinefunction(listener):
+                coroutine_run(listener(*args, **kwargs))
+            else:
+                listener(*args, **kwargs)
+
+    def add_event_listener(self, event: T, listener):
+        listeners = self._listeners.get(event)
+        if listeners is None:
+            listeners = set()
+            self._listeners[event] = listeners
+
+        listeners.add(listener)
+
+    def remove_event_listener(self, event: T, listener):
+        listeners = self._listeners.get(event)
+        if listeners is None:
+            return
+
+        if listener in listeners:
+            listeners.remove(listener)
+
+    def has(self, event: T):
+        return self.listeners_count(event) > 0
+
+    def listeners_count(self, event: T):
+        listeners = self._listeners.get(event)
+        if listeners is None:
+            return 0
+
+        return len(listeners)

wslink/protocol.py

@@ -9,6 +9,7 @@ import traceback
 from wslink import schedule_coroutine
 from wslink.publish import PublishManager
 from wslink.chunking import generate_chunks, UnChunker
+from wslink.websocket import ServerProtocol
 
 # from http://www.jsonrpc.org/specification, section 5.1
 METHOD_NOT_FOUND = -32601
@@ -141,7 +142,7 @@ class AbstractWebApp:
 
 
 class WslinkHandler(object):
-    def __init__(self, protocol=None, web_app=None):
+    def __init__(self, protocol: ServerProtocol, web_app=None):
         self.serverProtocol = protocol
         self.web_app = web_app
         self.functionMap = {}
@@ -153,6 +154,7 @@ class WslinkHandler(object):
         self.pub_manager = PublishManager()
         self.unchunkers = {}
         self.network_monitor = protocol.network_monitor
+        self.log_emitter = protocol.log_emitter
 
         # Build the rpc method dictionary, assuming we were given a serverprotocol
         if self.getServerProtocol():
@@ -253,7 +255,9 @@ class WslinkHandler(object):
 
     async def onMessage(self, is_binary, msg, client_id):
         if not is_binary:
-            logger.critical("wslink is not expecting text message:\n> %s", msg.data)
+            error_message = "wslink is not expecting text message:\n> %s"
+            logger.critical(error_message, msg.data)
+            self.log_emitter.critical(error_message % msg.data)
             return
 
         full_message = self.unchunkers[client_id].process_chunk(msg.data)
@@ -262,7 +266,13 @@ class WslinkHandler(object):
                 await self.onCompleteMessage(full_message, client_id)
 
     async def onCompleteMessage(self, rpc, client_id):
-        logger.debug("wslink incoming msg %s", self.payloadWithSecretStripped(rpc))
+        debug_message = "wslink incoming msg %s"
+        stripped_payload = self.payloadWithSecretStripped(rpc)
+        logger.debug(debug_message, stripped_payload)
+
+        if self.log_emitter.has("debug"):
+            self.log_emitter.debug(debug_message % stripped_payload)
+
         if "id" not in rpc:
             return
 
@@ -295,6 +305,7 @@ class WslinkHandler(object):
 
         # No matching method found
         if not methodName in self.functionMap:
+            self.log_emitter.error(f"Method not found: {methodName}")
             with self.network_monitor:
                 await self.sendWrappedError(
                     rpcid,
@@ -328,6 +339,7 @@ class WslinkHandler(object):
             logger.error("Exception raised")
             logger.error(repr(e_inst))
             logger.error(captured_trace)
+            self.log_emitter.exception(e_inst)
             with self.network_monitor:
                 await self.sendWrappedError(
                     rpcid,

wslink/websocket.py

@@ -6,9 +6,12 @@ ServerProtocol to hook all the needed LinkProtocols together.
 
 import logging
 import asyncio
+from typing import Literal
+from dataclasses import dataclass
 
-from . import register as exportRpc
-from . import schedule_callback
+from wslink import register as exportRpc
+from wslink import schedule_callback
+from wslink.emitter import EventEmitter
 
 logger = logging.getLogger(__name__)
 
@@ -117,6 +120,9 @@ class NetworkMonitor:
             await self.event.wait()
 
 
+LogEmitterEvents = Literal["exception", "error", "critical", "info", "debug"]
+
+
 class ServerProtocol(object):
     """
     Defines the core server protocol for wslink. Gathers a list of LinkProtocol
@@ -125,6 +131,7 @@ class ServerProtocol(object):
 
     def __init__(self):
         self.network_monitor = NetworkMonitor()
+        self.log_emitter = EventEmitter[LogEmitterEvents]()
         self.linkProtocols = []
         self.secret = None
         self.initialize()
@@ -169,7 +176,9 @@ class ServerProtocol(object):
         try:
             self.linkProtocols.remove(protocol)
         except ValueError as e:
-            logger.error("Link protocol missing from registered list.")
+            error_message = "Link protocol missing from registered list."
+            logger.error(error_message)
+            self.log_emitter("error", error_message)
 
     def getLinkProtocols(self):
         return self.linkProtocols

{wslink-2.2.1.dist-info → wslink-2.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wslink
-Version: 2.2.1
+Version: 2.3.0
 Summary: Python/JavaScript library for communicating over WebSocket
 Home-page: https://github.com/kitware/wslink
 Author: Kitware, Inc.

{wslink-2.2.1.dist-info → wslink-2.3.0.dist-info}/RECORD

@@ -1,25 +1,26 @@
 wslink/LICENSE,sha256=I44UH7kDVqxDLnnlOWw_hFL2Fz7RjQ_4vPzZv9NYgTU,1483
 wslink/__init__.py,sha256=qdLGVhzpEXBoraTnLr5S-8bX9oK8pZCQWe8rZRlwAco,2946
 wslink/chunking.py,sha256=1DJlGG6fjknGFrqPOtqUUc5tCrijldP7Kdx56d5e3Wg,7337
+wslink/emitter.py,sha256=4nm5fOwfvSb5tHq5cZtUsF8YNpuphl05QZQzNHljfRU,1669
 wslink/launcher.py,sha256=8VMs3juObLkyGYQFNLjMoo4qFpKIcxWz0kS-af-DKO4,21170
-wslink/protocol.py,sha256=T3F_Y_pZ4qXUApjpDoi9zXDkDvp3qnlKQICj2jnPlsA,16744
+wslink/protocol.py,sha256=IUvdYzXaaKP4qllJps6uQ8zvbipbH6MVwU6o6f60Kuc,17265
 wslink/publish.py,sha256=Xyv9piZT4HxO5l_SA7zeReH8t6tisVHUgU57Hjzgkcg,1595
 wslink/relay.py,sha256=E8Lzu2Ay7KbOheN1-ArAZawo8lLqdDgJXOZSBuMknYs,86
 wslink/server.py,sha256=yvhCjpzPOfhbZrpDvW9i4H_uSyuQAe3ZOP-BRBmgHQA,9326
 wslink/ssl_context.py,sha256=hNOJJCdrStws1Qf6vPvY4vTk9Bf8J5d90W3fS0cRv8o,2290
 wslink/uri.py,sha256=woCQ4yChUqTMg9IT6YYDtUYeKmCg7OUCEgeBGA-19DY,384
-wslink/websocket.py,sha256=wOuLfYeM72Ikzc9YA0qZjj6EfSXXQp-8p0uP-H9gapE,6004
+wslink/websocket.py,sha256=T47xgY6fgQQqDqcxmZpamIEvhQtvw_TWGeF2Es1dIpI,6350
 wslink/backends/__init__.py,sha256=cyJGjm-YyBSyOEX81owyTbJ3YnrA6dB7--B4LnsEtHI,1214
-wslink/backends/aiohttp/__init__.py,sha256=Bc3uPVrhNXnn5Xt3Tck2Th-dsysz1VAnQdh8WzRn1y0,9253
+wslink/backends/aiohttp/__init__.py,sha256=SSK5a_lxuEZGyW_mMMipgvAhJQQLhxAImR3jRokFchM,9301
 wslink/backends/aiohttp/launcher.py,sha256=gHNMvtgNHEwBN_QBRDSCrTp2B4K1PsfV81rKaHi7Cxo,8897
 wslink/backends/aiohttp/relay.py,sha256=oZAzIQTpsQaObWXaa-_VtoTOUQALC_QLDd9UvWspYaU,13311
 wslink/backends/generic/__init__.py,sha256=Qu65gWsd2xCSsxybnDtEDI5vMjHN-F5jgPZOyNIxnGs,112
-wslink/backends/generic/core.py,sha256=JuLeT6by2NA7y3qQL_aZCYEsAREZmshDZAvSnO1T_bo,4363
+wslink/backends/generic/core.py,sha256=-tlNgXdvEr8BrwosxBWlUJVk5yiGMLfbK3EWs5VEKNM,4358
 wslink/backends/jupyter/__init__.py,sha256=Qu65gWsd2xCSsxybnDtEDI5vMjHN-F5jgPZOyNIxnGs,112
-wslink/backends/jupyter/core.py,sha256=PCQN-uZPFROnRv8B5dNwnwHV67o4Bpme3_Z6V-zbOUA,3864
+wslink/backends/jupyter/core.py,sha256=F8R3uH4m6RHCrHHRiA5UAgDOLdyGbpuCSAgCZxANREk,2794
 wslink/backends/tornado/__init__.py,sha256=Qu65gWsd2xCSsxybnDtEDI5vMjHN-F5jgPZOyNIxnGs,112
 wslink/backends/tornado/core.py,sha256=tPMkkhWuO_ovkisVim0zcegwZKEAG4IRUdd_O_0a_R0,2157
-wslink-2.2.1.dist-info/METADATA,sha256=Ng1NUSdljWz0EDJg7yppAY29AZ2G-V3XbJDU0etWiLQ,3120
-wslink-2.2.1.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-wslink-2.2.1.dist-info/top_level.txt,sha256=N0d8eqvhwhfW1p1yPTmvxlbzhjz7ZyhBfysNvaFqpQY,7
-wslink-2.2.1.dist-info/RECORD,,
+wslink-2.3.0.dist-info/METADATA,sha256=hisMCyrZAwxLjXz24KZ74-VaFyHvusXzfpnzU0U5wJM,3120
+wslink-2.3.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+wslink-2.3.0.dist-info/top_level.txt,sha256=N0d8eqvhwhfW1p1yPTmvxlbzhjz7ZyhBfysNvaFqpQY,7
+wslink-2.3.0.dist-info/RECORD,,

{wslink-2.2.1.dist-info → wslink-2.3.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.44.0)
+Generator: bdist_wheel (0.45.1)
 Root-Is-Purelib: true
 Tag: py3-none-any