Add HTTP and SSE transport support to CLI (#15)

Author: joshuadavidthomas

CHANGELOG.md

@@ -18,6 +18,10 @@ and this project attempts to adhere to [Semantic Versioning](https://semver.org/
 
 ## [Unreleased]
 
+### Added
+
+- Support for HTTP and SSE transport protocols via CLI arguments (`--transport`, `--host`, `--port`, `--path`)
+
 ## [0.4.0]
 
 ### Added

README.md

@@ -92,6 +92,21 @@ There's also a Django management command if you prefer, but that requires adding
 python manage.py mcp_shell
 ```
 
+### Transport
+
+The server supports multiple transport protocols:
+
+```bash
+# Default: STDIO
+python -m mcp_django_shell
+
+# HTTP
+python -m mcp_django_shell --transport http --host 127.0.0.1 --port 8000
+
+# SSE
+python -m mcp_django_shell --transport sse --host 127.0.0.1 --port 8000
+```
+
 ### Client Configuration
 
 Configure your MCP client using one of the examples below. The command is the same for all clients, just expressed in annoyingly different JSON soup.
@@ -152,6 +167,7 @@ It wouldn't be an MCP server README without a gratuitous list of features punctu
 - 🚀 **Zero configuration** - No schemas, no settings, just Django
 - 🤖 **LLM-friendly** - Designed for AI assistants that already know Python
 - 📦 **Minimal dependencies** - Just FastMCP and Django (you already have Django)
+- 🌐 **Multiple transports** - STDIO, HTTP, SSE.. It has it all!
 - 🎯 **Does one thing well** - Runs code. That's it. That's the feature.
 
 Inspired by Armin Ronacher's [Your MCP Doesn't Need 30 Tools: It Needs Code](https://lucumr.pocoo.org/2025/8/18/code-mcps/).

src/mcp_django_shell/cli.py

@@ -26,11 +26,37 @@ def main(argv: Sequence[str] | None = None) -> int:
         action="store_true",
         help="Enable debug logging",
     )
+    parser.add_argument(
+        "--transport",
+        default="stdio",
+        choices=["stdio", "http", "sse"],
+        help="Transport protocol to use (default: stdio)",
+    )
+    parser.add_argument(
+        "--host",
+        default="127.0.0.1",
+        help="Host to bind to for HTTP/SSE transport (default: 127.0.0.1)",
+    )
+    parser.add_argument(
+        "--port",
+        type=int,
+        default=8000,
+        help="Port to bind to for HTTP/SSE transport (default: 8000)",
+    )
+    parser.add_argument(
+        "--path",
+        default="/mcp",
+        help="Path for HTTP transport endpoint (default: /mcp)",
+    )
     args = parser.parse_args(argv)
 
     debug: bool = args.debug
     settings: str | None = args.settings
     pythonpath: str | None = args.pythonpath
+    transport: str = args.transport
+    host: str = args.host
+    port: int = args.port
+    path: str = args.path
 
     if debug:
         logging.basicConfig(
@@ -54,6 +80,14 @@ def main(argv: Sequence[str] | None = None) -> int:
 
     logger.info("Starting MCP Django Shell server")
     logger.debug("Django settings module: %s", django_settings)
+    logger.debug("Transport: %s", transport)
+    if transport in ["http", "sse"]:
+        logger.info(
+            "Server will be available at %s:%s%s",
+            host,
+            port,
+            path if transport == "http" else "",
+        )
 
     def signal_handler(signum: int, _frame: Any):  # pragma: no cover
         logger.info("Received signal %s, shutting down MCP server", signum)
@@ -67,7 +101,16 @@ def signal_handler(signum: int, _frame: Any):  # pragma: no cover
 
         from .server import mcp
 
-        mcp.run()
+        kwargs: dict[str, Any] = {"transport": transport}
+
+        if transport in ["http", "sse"]:
+            kwargs["host"] = host
+            kwargs["port"] = port
+
+        if transport == "http":
+            kwargs["path"] = path
+
+        mcp.run(**kwargs)
 
     except Exception as e:
         logger.error("MCP server crashed: %s", e, exc_info=True)

tests/test_cli.py

@@ -62,3 +62,40 @@ def test_cli_with_debug(monkeypatch):
     result = main(["--debug"])
 
     assert result == 0
+
+
+def test_cli_with_http_transport(monkeypatch):
+    monkeypatch.setenv("DJANGO_SETTINGS_MODULE", "tests.settings")
+
+    mock_mcp = Mock()
+    monkeypatch.setattr("mcp_django_shell.server.mcp", mock_mcp)
+
+    result = main(
+        [
+            "--transport",
+            "http",
+            "--host",
+            "127.0.0.1",
+            "--port",
+            "8000",
+            "--path",
+            "/mcp",
+        ]
+    )
+
+    mock_mcp.run.assert_called_once_with(
+        transport="http", host="127.0.0.1", port=8000, path="/mcp"
+    )
+    assert result == 0
+
+
+def test_cli_with_sse_transport(monkeypatch):
+    monkeypatch.setenv("DJANGO_SETTINGS_MODULE", "tests.settings")
+
+    mock_mcp = Mock()
+    monkeypatch.setattr("mcp_django_shell.server.mcp", mock_mcp)
+
+    result = main(["--transport", "sse", "--host", "0.0.0.0", "--port", "9000"])
+
+    mock_mcp.run.assert_called_once_with(transport="sse", host="0.0.0.0", port=9000)
+    assert result == 0