django-orm-lens 1.3.0__tar.gz → 1.4.0__tar.gz

{django_orm_lens-1.3.0 → django_orm_lens-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-orm-lens
-Version: 1.3.0
+Version: 1.4.0
 Summary: Static analysis + MCP server for Django models. Sidebar tree, ER diagrams, and JSON output for terminals and AI coding agents.
 Author: frowningdev
 License: MIT

{django_orm_lens-1.3.0 → django_orm_lens-1.4.0}/django_orm_lens/mcp_server.py

@@ -37,6 +37,7 @@ from typing import Any, Dict, List, Union
 from .cli import _build_mermaid
 from .migrations_parser import describe_migration_dependency
 from .models import WorkspaceIndex, find_model, find_user_model, resolve_related_tail
+from .query_analyzer import scan_for_nplusone as _scan_for_nplusone
 from .query_analyzer import suggest_indexes as _suggest_indexes
 from .signals_parser import signal_graph as _signal_graph
 from .workspace import WorkspaceError, get_index, resolve_workspace
@@ -300,6 +301,24 @@ def _tool_signal_graph(args: Dict[str, Any]) -> str:
     return json.dumps(result, indent=2)
 
 
+def _tool_nplusone_scan(args: Dict[str, Any]) -> str:
+    """Static scan for Django ORM N+1 anti-patterns across the workspace.
+
+    Walks every ``.py`` file, finds ``for ... in <queryset>:`` loops, and
+    flags attribute-chain accesses against the loop target that touch a
+    related object (FK / O2O / M2M / reverse FK) without a matching
+    ``select_related`` / ``prefetch_related`` clause on the source queryset.
+    Uses the parsed WorkspaceIndex to classify relations; falls back to
+    a schema-less heuristic when a model is unknown. Zero-runtime, no DB.
+    """
+    ws = _resolve(args)
+    if isinstance(ws, WorkspaceError):
+        return _error_json(ws)
+    index = get_index(ws)
+    findings = _scan_for_nplusone(str(ws), index)
+    return json.dumps([f.to_dict() for f in findings], indent=2)
+
+
 # ---------------------------------------------------------------------------
 # Tool registry — descriptions ship in tools/list so agents see them
 # ---------------------------------------------------------------------------
@@ -377,6 +396,17 @@ TOOLS = {
             "Django boot." + _WORKSPACE_HINT
         ),
     },
+    "nplusone_scan": {
+        "handler": _tool_nplusone_scan,
+        "description": (
+            "Static scan for Django ORM N+1 anti-patterns. Walks every .py "
+            "file, finds 'for x in <queryset>:' loops that touch related "
+            "objects (FK/O2O/M2M/reverse) without matching select_related "
+            "or prefetch_related. Returns findings with file, line, "
+            "queryset_var, accessed relations, suggested_fix, and confidence "
+            "(high/medium). Zero-runtime, no DB, no Django boot." + _WORKSPACE_HINT
+        ),
+    },
 }
 
 
@@ -478,6 +508,10 @@ def main() -> int:
             @server.tool(name=name, description=description)
             def signal_graph(workspace_root: str = "") -> str:
                 return handler({"workspace_root": workspace_root})
+        elif name == "nplusone_scan":
+            @server.tool(name=name, description=description)
+            def nplusone_scan(workspace_root: str = "") -> str:
+                return handler({"workspace_root": workspace_root})
 
     for name, spec in TOOLS.items():
         _register(name, spec["description"], spec["handler"])

{django_orm_lens-1.3.0 → django_orm_lens-1.4.0}/django_orm_lens.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-orm-lens
-Version: 1.3.0
+Version: 1.4.0
 Summary: Static analysis + MCP server for Django models. Sidebar tree, ER diagrams, and JSON output for terminals and AI coding agents.
 Author: frowningdev
 License: MIT

{django_orm_lens-1.3.0 → django_orm_lens-1.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "django-orm-lens"
-version = "1.3.0"
+version = "1.4.0"
 description = "Static analysis + MCP server for Django models. Sidebar tree, ER diagrams, and JSON output for terminals and AI coding agents."
 readme = "README.md"
 requires-python = ">=3.9"

{django_orm_lens-1.3.0 → django_orm_lens-1.4.0}/tests/test_mcp_server.py

@@ -34,6 +34,7 @@ from django_orm_lens.mcp_server import (
     _tool_find_relations,
     _tool_list_apps,
     _tool_list_models,
+    _tool_nplusone_scan,
     _tool_signal_graph,
     _tool_suggest_indexes,
 )
@@ -66,7 +67,7 @@ def _write_django_project(root: Path) -> None:
 class ToolsRegistryTest(unittest.TestCase):
     """The ``TOOLS`` dict is what feeds ``tools/list`` — schema-level asserts."""
 
-    def test_all_nine_tools_registered(self) -> None:
+    def test_all_ten_tools_registered(self) -> None:
         expected = {
             "list_apps",
             "list_models",
@@ -77,6 +78,7 @@ class ToolsRegistryTest(unittest.TestCase):
             "describe_migration_dependency",
             "suggest_indexes",
             "signal_graph",
+            "nplusone_scan",
         }
         self.assertEqual(set(TOOLS), expected)
 
@@ -213,6 +215,9 @@ class ErrorEnvelopeTest(unittest.TestCase):
     def test_signal_graph_envelope(self) -> None:
         self._assert_envelope(_tool_signal_graph({"workspace_root": str(self.tmp)}))
 
+    def test_nplusone_scan_envelope(self) -> None:
+        self._assert_envelope(_tool_nplusone_scan({"workspace_root": str(self.tmp)}))
+
     def test_missing_workspace_returns_not_found_envelope(self) -> None:
         raw = _tool_list_apps({"workspace_root": str(self.tmp / "no-such-dir")})
         self._assert_envelope(raw, expected_code="WORKSPACE_NOT_FOUND")
@@ -266,6 +271,34 @@ class HappyPathTest(unittest.TestCase):
         self.assertIsInstance(payload, list)
         self.assertEqual(len(payload), 1)
 
+    def test_nplusone_scan_flags_intentional_loop(self) -> None:
+        # Plant a synthetic views.py with a textbook N+1 pattern —
+        # for book in Book.objects.all(): print(book.author.name) — and
+        # confirm the tool surfaces at least one finding with the shape
+        # the CHANGELOG documents (file, line, queryset_var, accessed,
+        # suggested_fix, confidence).
+        views = self.tmp / "myapp" / "views.py"
+        views.write_text(
+            "from myapp.models import Book\n"
+            "def index(request):\n"
+            "    books = Book.objects.all()\n"
+            "    for book in books:\n"
+            "        print(book.author.name)\n",
+            encoding="utf-8",
+        )
+        raw = _tool_nplusone_scan({"workspace_root": str(self.tmp)})
+        payload = json.loads(raw)
+        self.assertIsInstance(payload, list, f"expected list, got {type(payload).__name__}: {payload!r}")
+        self.assertGreaterEqual(
+            len(payload), 1, f"no findings emitted for a textbook N+1 pattern: {payload!r}"
+        )
+        f = payload[0]
+        for key in ("file", "line", "loop_var", "queryset_var", "accessed",
+                    "suggested_fix", "confidence"):
+            self.assertIn(key, f, f"finding missing key {key!r}: {f!r}")
+        self.assertIn(f["confidence"], ("high", "medium"))
+        self.assertIn("select_related", f["suggested_fix"])
+
 
 class SentinelDataclassTest(unittest.TestCase):
     """The WorkspaceError sentinel is what powers the envelope. Keep the