refactor: Remove use of anywidget-command kind

banesullivan · Jul 21, 2024 · 9600f14 · 9600f14
1 parent 38582ee
commit 9600f14
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/localtileserver/custom/lts_widget.js b/localtileserver/custom/lts_widget.js
@@ -38,7 +38,7 @@ function render({ model, el, experimental }) {
             tileHandlers[id] = handleTile;
 
             model.send(
-                { id, kind: "anywidget-command", name: "get_tile", msg: {...coords, identifier} },
+                { id, kind: "get_tile", msg: {...coords, identifier} },
                 undefined,
                 [],
             );

diff --git a/localtileserver/custom/widgets.py b/localtileserver/custom/widgets.py
@@ -12,16 +12,14 @@
 THREAD_EXECUTOR = concurrent.futures.ThreadPoolExecutor(max_workers=os.cpu_count())
 
 
-def handle_anywidget_command(widget, msg: str | list | dict, buffers: list[bytes]) -> None:
-    if not isinstance(msg, dict) or msg.get("kind") != "anywidget-command":
-        return
+def handle_custom_message(widget, msg: dict, buffers: list[bytes]) -> None:
+    assert msg["kind"] == "get_tile", f"unexpected message kind: {msg['kind']}"
 
     def target():
         response, buffers = widget.get_tile(msg["msg"])
         widget.send(
             {
                 "id": msg["id"],
-                "kind": "anywidget-command-response",
                 "response": response,
             },
             buffers,
@@ -36,7 +34,7 @@ class TileLayerWidget(anywidget.AnyWidget):
     """Leaflet Tile Layer Widget.
 
     Args:
-        anywidget (_type_): _description_
+        client (TileClient): A TileClient instance.
     """
 
     _esm = THIS_DIR / "lts_widget.js"
@@ -50,7 +48,7 @@ class TileLayerWidget(anywidget.AnyWidget):
     def __init__(self, client: TileClient, **kwargs):
         super().__init__()
         self.client = client  # TODO: weakref?
-        self.on_msg(handle_anywidget_command)
+        self.on_msg(handle_custom_message)
         self._display_kwargs = kwargs
         self.bounds = self.client.bounds()
         self.identifier = uuid.uuid4().hex