txl-remote-kernels 0.1.17__tar.gz → 0.1.19__tar.gz

{txl_remote_kernels-0.1.17 → txl_remote_kernels-0.1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: txl_remote_kernels
-Version: 0.1.17
+Version: 0.1.19
 Summary: TXL plugin for remote kernels
 Project-URL: Source, https://github.com/davidbrochart/jpterm/plugins/remote_kernels
 Author-email: David Brochart <david.brochart@gmail.com>

{txl_remote_kernels-0.1.17 → txl_remote_kernels-0.1.19}/pyproject.toml

@@ -34,6 +34,7 @@ Source = "https://github.com/davidbrochart/jpterm/plugins/remote_kernels"
 
 [project.entry-points."asphalt.components"]
 remote_kernels = "txl_remote_kernels.components:RemoteKernelsComponent"
+remote_kernelspecs = "txl_remote_kernels.components:RemoteKernelspecsComponent"
 
 [tool.hatch.version]
 path = "txl_remote_kernels/__init__.py"

txl_remote_kernels-0.1.19/txl_remote_kernels/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.19"

txl_remote_kernels-0.1.19/txl_remote_kernels/components.py

@@ -0,0 +1,78 @@
+from typing import Any
+from urllib import parse
+
+import httpx
+import y_py as Y
+from asphalt.core import Component, Context
+
+from txl.base import Kernels, Kernelspecs
+
+from .driver import KernelDriver
+
+
+class RemoteKernels(Kernels):
+
+    comm_handlers = []
+
+    def __init__(
+        self,
+        url: str,
+        kernel_name: str | None,
+    ):
+        self.kernel = KernelDriver(url, kernel_name, comm_handlers=self.comm_handlers)
+
+    async def execute(self, ydoc: Y.YDoc, ycell: Y.YMap):
+        await self.kernel.execute(ydoc, ycell)
+
+
+class RemoteKernelspecs(Kernelspecs):
+    def __init__(
+        self,
+        url: str,
+    ):
+        parsed_url = parse.urlparse(url)
+        self.base_url = parse.urljoin(url, parsed_url.path).rstrip("/")
+        self.query_params = parse.parse_qs(parsed_url.query)
+        self.cookies = httpx.Cookies()
+
+    async def get(self) -> dict[str, Any]:
+        async with httpx.AsyncClient() as client:
+            r = await client.get(
+                f"{self.base_url}/api/kernelspecs",
+                params={**self.query_params},
+                cookies=self.cookies,
+            )
+            d = r.json()
+            self.cookies.update(r.cookies)
+            return d
+
+
+class RemoteKernelsComponent(Component):
+    def __init__(self, url: str = "http://127.0.0.1:8000"):
+        super().__init__()
+        self.url = url
+
+    async def start(
+        self,
+        ctx: Context,
+    ) -> None:
+        url = self.url
+
+        class _RemoteKernels(RemoteKernels):
+            def __init__(self, *args, **kwargs):
+                super().__init__(url, *args, **kwargs)
+
+        ctx.add_resource(_RemoteKernels, types=Kernels)
+
+
+class RemoteKernelspecsComponent(Component):
+    def __init__(self, url: str = "http://127.0.0.1:8000"):
+        super().__init__()
+        self.url = url
+
+    async def start(
+        self,
+        ctx: Context,
+    ) -> None:
+        kernelspecs = RemoteKernelspecs(self.url)
+        ctx.add_resource(kernelspecs, types=Kernelspecs)

txl_remote_kernels-0.1.17/txl_remote_kernels/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.17"

txl_remote_kernels-0.1.17/txl_remote_kernels/components.py

@@ -1,39 +0,0 @@
-import y_py as Y
-from asphalt.core import Component, Context
-
-from txl.base import Kernels
-
-from .driver import KernelDriver
-
-
-class RemoteKernels(Kernels):
-
-    comm_handlers = []
-
-    def __init__(
-        self,
-        url: str,
-        kernel_name: str | None,
-    ):
-        self.kernel = KernelDriver(url, kernel_name, comm_handlers=self.comm_handlers)
-
-    async def execute(self, ydoc: Y.YDoc, ycell: Y.YMap):
-        await self.kernel.execute(ydoc, ycell)
-
-
-class RemoteKernelsComponent(Component):
-    def __init__(self, url: str = "http://127.0.0.1:8000"):
-        super().__init__()
-        self.url = url
-
-    async def start(
-        self,
-        ctx: Context,
-    ) -> None:
-        url = self.url
-
-        class _RemoteKernels(RemoteKernels):
-            def __init__(self, *args, **kwargs):
-                super().__init__(url, *args, **kwargs)
-
-        ctx.add_resource(_RemoteKernels, types=Kernels)