koatl-kernel 1.0__tar.gz

koatl_kernel-1.0/.gitignore

@@ -0,0 +1,19 @@
+MANIFEST
+build
+dist
+_build
+*.py[co]
+__pycache__
+*.egg-info
+*~
+*.bak
+.ipynb_checkpoints
+.tox
+.DS_Store
+\#*#
+.#*
+.coverage
+htmlcov
+.cache
+.idea
+data_kernelspec

koatl_kernel-1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Kevin Zhang
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

koatl_kernel-1.0/PKG-INFO

@@ -0,0 +1,40 @@
+Metadata-Version: 2.4
+Name: koatl-kernel
+Version: 1.0
+Summary: Koatl kernel for Jupyter
+Project-URL: Homepage, https://github.com/skkestrel/koatl
+Author-email: Kevin <skkestrel@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2025 Kevin Zhang
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+License-File: LICENSE
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Requires-Dist: ipykernel
+Requires-Dist: jupyter-client
+Requires-Dist: koatl
+Provides-Extra: test
+Requires-Dist: jupyter-kernel-test; extra == 'test'
+Description-Content-Type: text/x-rst
+
+This is the jupyter kernel for koatl.

koatl_kernel-1.0/README.rst

@@ -0,0 +1 @@
+This is the jupyter kernel for koatl.

koatl_kernel-1.0/hatch_build.py

@@ -0,0 +1,44 @@
+import os
+import sys
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+
+
+import argparse
+import json
+import os
+import sys
+import shutil
+
+from jupyter_client.kernelspec import KernelSpecManager
+from tempfile import TemporaryDirectory
+
+kernel_json = {
+    "argv": [sys.executable, "-m", "koatl_kernel", "-f", "{connection_file}"],
+    "display_name": "Koatl",
+    "language": "koatl",
+}
+
+
+class CustomHook(BuildHookInterface):
+    def initialize(self, version, build_data):
+        here = os.path.abspath(os.path.dirname(__file__))
+        sys.path.insert(0, here)
+        prefix = os.path.join(here, "data_kernelspec")
+
+        with TemporaryDirectory() as td:
+            os.chmod(td, 0o755)  # Starts off as 700, not user readable
+            with open(os.path.join(td, "kernel.json"), "w") as f:
+                json.dump(kernel_json, f, sort_keys=True)
+            print("Installing Jupyter kernel spec")
+
+            # Requires logo files in kernel root directory
+            cur_path = os.path.dirname(os.path.realpath(__file__))
+            for logo in ["logo-32x32.png", "logo-64x64.png"]:
+                try:
+                    shutil.copy(os.path.join(cur_path, logo), td)
+                except FileNotFoundError:
+                    print("Custom logo files not found. Default logos will be used.")
+
+            KernelSpecManager().install_kernel_spec(
+                td, "koatl", user=False, prefix=prefix
+            )

koatl_kernel-1.0/koatl_kernel/__init__.py

@@ -0,0 +1,3 @@
+__version__ = "1.0"
+
+from .kernel import KoatlKernel

koatl_kernel-1.0/koatl_kernel/__main__.py

@@ -0,0 +1,4 @@
+from ipykernel.kernelapp import IPKernelApp
+from . import KoatlKernel
+
+IPKernelApp.launch_instance(kernel_class=KoatlKernel)

koatl_kernel-1.0/koatl_kernel/kernel.py

@@ -0,0 +1,67 @@
+from ipykernel.ipkernel import IPythonKernel
+from IPython.core.inputtransformer2 import find_last_indent
+from koatl.notebook import source_code_transformer, import_prelude
+from koatl.notebook.magic import koatl_cell_magic
+from .shell import KoatlShell
+
+
+class KoatlKernel(IPythonKernel):
+    implementation = "Koatl"
+    implementation_version = "1.0"
+    language = "koatl"
+    language_version = "0.1"
+    language_info = {
+        "name": "koatl",
+        "mimetype": "text/x-koatl",
+        "file_extension": ".tl",
+    }
+
+    shell_class = KoatlShell
+
+    @property
+    def banner(self):
+        return "Koatl Kernel - Interactive Koatl Environment"
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+        # TODO: these depend on the python tokenizer which don't work
+
+        # itm.token_transformers.insert(0, KoatlMagicAssign)
+        # itm.token_transformers.insert(1, KoatlEscapedCommand)
+        # itm.token_transformers.insert(2, KoatlSystemAssign)
+
+        itm = self.shell.input_transformer_manager
+
+        itm.do_token_transforms = lambda lines: lines
+        itm.line_transforms.insert(0, koatl_cell_magic)
+        itm.check_complete = self._itm_check_complete
+
+        import_prelude(self.shell.user_ns)
+
+    def _itm_check_complete(self, cell):
+        import re
+
+        lines = cell.splitlines(keepends=True)
+
+        if (
+            lines[-1].endswith(":")
+            or lines[-1].endswith("=>")
+            or lines[-1].endswith("[")
+            or lines[-1].endswith("(")
+        ):
+            return "incomplete", find_last_indent(lines) + 2
+
+        if lines[-1].startswith(" "):
+            if (
+                len(lines) > 2
+                and re.match(r"^\s*$", lines[-2])
+                and re.match(r"^\s*$", lines[-1])
+            ):
+                return "complete", None
+            return "incomplete", find_last_indent(lines)
+
+        return "complete", None
+
+    async def do_execute(self, code, *args, **kwargs):
+        return await super().do_execute(code, *args, **kwargs)

koatl_kernel-1.0/koatl_kernel/shell.py

@@ -0,0 +1,62 @@
+import koatl
+import builtins as builtin_mod
+import sys
+import warnings
+from typing import List as ListType, Dict as DictType, Any as AnyType
+from typing import Optional, Sequence, Tuple
+from ipykernel.compiler import XCachingCompiler
+from ipykernel.zmqshell import ZMQInteractiveShell
+from IPython.core.error import InputRejected
+from IPython.core.interactiveshell import *
+from IPython.core.ultratb import FrameInfo
+
+_source_map_override = {}
+_orig_frame_info_init = None
+
+if _orig_frame_info_init is None:
+    _orig_frame_info_init = FrameInfo.__init__
+
+    def _new_init(self, *args, **kwargs):
+        _orig_frame_info_init(self, *args, **kwargs)
+
+        # only override for ipykernel frames
+        if "ipykernel_" in self.filename:
+            old_lineno = self.lineno
+            self.lineno = _source_map_override.get(old_lineno, 0)
+            self._sd.lineno = _source_map_override.get(old_lineno, 0)
+
+    FrameInfo.__init__ = _new_init
+
+
+class KoatlXCachingCompiler(XCachingCompiler):
+    def cache(self, transformed_code, number=0, raw_code=None):
+        return super().cache(self.shell._koatl_cell, number, raw_code)
+
+
+class KoatlShell(ZMQInteractiveShell):
+    def __init__(self, *args, **kwargs):
+        kwargs["compiler_class"] = KoatlXCachingCompiler
+        super().__init__(*args, **kwargs)
+        self.compile.shell = self
+
+    def _get_exc_info(self, exc_tuple=None):
+        ret = super()._get_exc_info(exc_tuple)
+        self._exc_info = ret
+        return ret
+
+    def transform_cell(self, raw_cell):
+        cell = super().transform_cell(raw_cell)
+
+        self._koatl_cell = cell
+        (cell, sourcemap) = koatl.transpile(cell, mode="interactive", sourcemap=True)
+        _source_map_override.clear()
+
+        last_seen_line = 0
+        for i in range(len(cell.splitlines()) + 1):
+            if i in sourcemap:
+                _source_map_override[i] = sourcemap[i]
+                last_seen_line = sourcemap[i]
+            else:
+                _source_map_override[i] = last_seen_line
+
+        return cell

koatl_kernel-1.0/pyproject.toml

@@ -0,0 +1,53 @@
+[build-system]
+requires = [
+    "hatchling>=1.10.0", "ipykernel"
+]
+build-backend = "hatchling.build"
+
+[project]
+name = "koatl-kernel"
+dynamic = [
+    "version",
+]
+description = "Koatl kernel for Jupyter"
+readme = "README.rst"
+license = { file = "LICENSE" }
+authors = [
+    { name = "Kevin", email = "skkestrel@gmail.com" },
+]
+classifiers = [
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: BSD License",
+    "Programming Language :: Python :: 3",
+]
+requires-python = ">=3.8"
+dependencies = [
+    "ipykernel",
+    "jupyter_client",
+    "koatl"
+]
+
+[project.optional-dependencies]
+test = [
+  "jupyter_kernel_test",
+]
+
+[project.urls]
+Homepage = "https://github.com/skkestrel/koatl"
+
+[tool.hatch.version]
+path = "koatl_kernel/__init__.py"
+
+# Used to call hatch_build.py
+[tool.hatch.build.hooks.custom]
+
+[tool.hatch.build.targets.wheel]
+packages = ["/koatl_kernel"]
+
+[tool.hatch.build.targets.sdist]
+include = [
+    "/koatl_kernel",
+]
+
+[tool.hatch.build.targets.wheel.shared-data]
+"data_kernelspec/share" = "share"