tests passing

Author: samuelcolvin

Makefile

@@ -56,12 +56,9 @@ typecheck-py: ## Typecheck the code
 .PHONY: typecheck
 typecheck: typecheck-ts typecheck-py ## Typecheck all code
 
-.PHONY: test-ts
-test-ts: ## Run TS tests
-	CI=1 npm run test
-
 .PHONY: test
-test: test-ts ## Run all tests
+test: ## Run tests
+	uv run coverage run -m pytest
 
 .PHONY: all
 all: format typecheck test ## run format, typecheck and test

build/build.py

@@ -7,13 +7,13 @@
 root_dir = this_dir.parent
 
 # Define source and destination paths
-src = this_dir / "prepare_env.py"
-dst = root_dir / "mcp_run_python" / "deno" / "prepareEnvCode.ts"
+src = this_dir / 'prepare_env.py'
+dst = root_dir / 'mcp_run_python' / 'deno' / 'prepareEnvCode.ts'
 
 python_code = src.read_text()
 
 # Escape backslashes for JavaScript string literal
-python_code = python_code.replace("\\", "\\\\")
+python_code = python_code.replace('\\', '\\\\')
 
 # Create the JavaScript/TypeScript code
 ts_code = f"""// DO NOT EDIT THIS FILE DIRECTLY, INSTEAD RUN "make build"
@@ -22,4 +22,4 @@
 
 dst.write_text(ts_code)
 
-print(f"Successfully built {dst}")
+print(f'Successfully built {dst}')

build/stubs/pyodide/code.pyi

@@ -1,2 +1 @@
-def find_imports(code: str) -> list[str]:
-    ...
+def find_imports(code: str) -> list[str]: ...

build/stubs/pyodide_js.pyi

@@ -1,7 +1,5 @@
 class _ToPy:
-    def to_py(self) -> dict[str, str]:
-        ...
-
+    def to_py(self) -> dict[str, str]: ...
 
 class _Api:
     _import_name_to_package_name: _ToPy

mcp_run_python/__init__.py

@@ -0,0 +1,8 @@
+from __future__ import annotations as _annotations
+
+from importlib.metadata import version as _metadata_version
+
+from .main import deno_args, run_deno_server
+
+__version__ = _metadata_version('mcp_run_python')
+__all__ = '__version__', 'deno_args', 'run_deno_server'

mcp_run_python/__main__.py

@@ -0,0 +1,6 @@
+"""This means `python -m mcp_run_python` should run the CLI."""
+
+from ._cli import cli
+
+if __name__ == '__main__':
+    cli()

mcp_run_python/_cli.py

@@ -0,0 +1,36 @@
+from __future__ import annotations as _annotations
+
+import argparse
+import sys
+from collections.abc import Sequence
+
+from . import __version__
+from .main import run_deno_server
+
+
+def cli() -> int:  # pragma: no cover
+    """Run the CLI."""
+    sys.exit(cli_logic())
+
+
+def cli_logic(args_list: Sequence[str] | None = None) -> int:
+    parser = argparse.ArgumentParser(
+        prog='mcp-run-python',
+        description=f'mcp-run-python CLI v{__version__}\n\nMCP server for running untrusted Python code.\n',
+        formatter_class=argparse.RawTextHelpFormatter,
+    )
+    parser.add_argument('--version', action='store_true', help='Show version and exit')
+
+    parser.add_argument(
+        'mode',
+        choices=['stdio', 'streamable-http', 'http', 'warmup'],
+        help='Mode to run in ("http" is an alias for "streamable-http")',
+    )
+
+    args = parser.parse_args(args_list)
+    if args.version:
+        print(f'mcp-run-python {__version__}')
+        return 0
+    else:
+        run_deno_server(args.mode.replace('-', '_'))
+        return 0

mcp_run_python/deno.jsonc

@@ -6,7 +6,7 @@
     ".": "./deno/main.ts"
   },
   "tasks": {
-    "run": "deno run -N -R=node_modules -W=node_modules deno/main.ts",
+    "dev": "deno run -N -R=node_modules -W=node_modules deno/main.ts",
     "lint": "deno lint",
     "typecheck": "deno check deno",
     "format": "deno fmt"

mcp_run_python/main.py

@@ -0,0 +1,30 @@
+import subprocess
+import sys
+from pathlib import Path
+from typing import Literal
+
+Mode = Literal['stdio', 'streamable_http', 'warmup']
+THIS_DIR = Path(__file__).parent
+
+
+def run_deno_server(mode: Mode, *, port: int | None = None):
+    print('Running mcp-run-python server...', file=sys.stderr)
+    try:
+        subprocess.run(('deno', *deno_args(mode, port=port)), cwd=THIS_DIR)
+    except KeyboardInterrupt:
+        print('Server stopped.', file=sys.stderr)
+
+
+def deno_args(mode: Mode, *, port: int | None = None) -> list[str]:
+    args = [
+        'run',
+        '-N',
+        f'-R={THIS_DIR / "node_modules"}',
+        f'-W={THIS_DIR / "node_modules"}',
+        '--node-modules-dir=auto',
+        str(THIS_DIR / 'deno/main.ts'),
+        mode,
+    ]
+    if port is not None:
+        args.append(f'--port={port}')
+    return args

pyproject.toml

@@ -1,3 +1,7 @@
+[build-system]
+requires = ["uv_build>=0.8.10"]
+build-backend = "uv_build"
+
 [project]
 name = "mcp-run-python"
 version = "0.0.1"
@@ -25,7 +29,16 @@ classifiers = [
     "Topic :: Internet",
 ]
 requires-python = ">=3.10"
-dependencies = ["mcp>=1.13.1"]
+
+[project.urls]
+Homepage = "https://github.com/pydantic/mcp-run-python"
+Source = "https://github.com/pydantic/mcp-run-python"
+Changelog = "https://github.com/pydantic/mcp-run-python/releases"
+
+[tool.uv.build-backend]
+module-name = "mcp_run_python"
+module-root = ""
+
 
 [dependency-groups]
 dev = [
@@ -34,11 +47,15 @@ dev = [
     "httpx>=0.28.1",
     "inline-snapshot>=0.19.3",
     "micropip>=0.9.0; python_version >= '3.12'",
+    "mcp>=1.13.1",
     "pytest>=8.3.3",
     "pytest-pretty>=1.2.0",
     "ruff>=0.12.11",
 ]
 
+[project.scripts]
+mcp-run-python = "mcp_run_python._cli:cli"
+
 [tool.ruff]
 line-length = 120
 target-version = "py310"