pygraph-mcp 0.2.2__tar.gz → 0.2.4__tar.gz

{pygraph_mcp-0.2.2 → pygraph_mcp-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygraph-mcp
-Version: 0.2.2
+Version: 0.2.4
 Summary: AST-based Python/Flask codebase indexer for AI agents
 Project-URL: Homepage, https://github.com/shvmgyl15/pygraph
 Project-URL: Repository, https://github.com/shvmgyl15/pygraph

{pygraph_mcp-0.2.2 → pygraph_mcp-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pygraph-mcp"
-version = "0.2.2"
+version = "0.2.4"
 description = "AST-based Python/Flask codebase indexer for AI agents"
 requires-python = ">=3.11"
 authors = [{name = "Shivam Goel", email = "sg15rokz@gmail.com"}]

{pygraph_mcp-0.2.2 → pygraph_mcp-0.2.4}/src/pygraph/extractors/flask.py

@@ -375,6 +375,65 @@ def _extract_add_url_rule_calls(tree: ast.Module, file_path: str) -> list[HTTPRo
     return routes
 
 
+def _extract_flask_restful_routes(tree: ast.Module, file_path: str) -> list[HTTPRoute]:
+    routes: list[HTTPRoute] = []
+
+    class_defs: dict[str, ast.ClassDef] = {}
+    for node in ast.iter_child_nodes(tree):
+        if isinstance(node, ast.ClassDef):
+            class_defs[node.name] = node
+
+    for node in ast.walk(tree):
+        if not isinstance(node, ast.Call):
+            continue
+        func = node.func
+        if not isinstance(func, ast.Attribute) or func.attr != "add_resource":
+            continue
+        if len(node.args) < 2:
+            continue
+
+        cls_node = node.args[0]
+        cls_name = ast.unparse(cls_node) if isinstance(cls_node, ast.Name) else ""
+        if not cls_name:
+            continue
+
+        path_args: list[str] = []
+        for arg in node.args[1:]:
+            p = _string_from_node(arg)
+            if p:
+                path_args.append(p)
+
+        if not path_args:
+            continue
+
+        http_methods: list[str] = []
+        cls_def = class_defs.get(cls_name)
+        if cls_def:
+            for item in cls_def.body:
+                if isinstance(item, (ast.FunctionDef, ast.AsyncFunctionDef)):
+                    method_upper = {
+                        "get": "GET", "post": "POST", "put": "PUT",
+                        "delete": "DELETE", "patch": "PATCH",
+                        "head": "HEAD", "options": "OPTIONS",
+                    }.get(item.name.lower())
+                    if method_upper:
+                        http_methods.append(method_upper)
+        if not http_methods:
+            http_methods = ["GET"]
+
+        for path in path_args:
+            for method in http_methods:
+                routes.append(HTTPRoute(
+                    method=method,
+                    path=path,
+                    handler=f"{file_path}::{cls_name}",
+                    file=file_path,
+                    line=node.lineno,
+                ))
+
+    return routes
+
+
 def extract_flask(source: str, file_path: str) -> dict[str, Any]:
     try:
         tree = ast.parse(source)
@@ -390,7 +449,7 @@ def extract_flask(source: str, file_path: str) -> dict[str, Any]:
         }
 
     return {
-        "routes": _extract_routes(tree, file_path) + _extract_add_url_rule_calls(tree, file_path),
+        "routes": _extract_routes(tree, file_path) + _extract_add_url_rule_calls(tree, file_path) + _extract_flask_restful_routes(tree, file_path),
         "blueprints": _detect_blueprint_assignments(tree),
         "blueprint_registrations": _extract_blueprint_registrations(tree, file_path),
         "template_refs": _extract_template_refs(tree, file_path),

{pygraph_mcp-0.2.2 → pygraph_mcp-0.2.4}/src/pygraph/extractors/implements.py

@@ -55,6 +55,8 @@ def extract_implements(source: str, file_path: str) -> list[ImplementsEdge]:
                     ImplementsEdge(
                         interface=full_base,
                         concrete=concrete_name,
+                        file=file_path,
+                        line=node.lineno,
                     )
                 )
 

{pygraph_mcp-0.2.2 → pygraph_mcp-0.2.4}/src/pygraph/graph/types.py

@@ -155,6 +155,8 @@ class TestEdge:
 class ImplementsEdge:
     interface: str = ""
     concrete: str = ""
+    file: str = ""
+    line: int = 0
 
 
 @dataclass

{pygraph_mcp-0.2.2 → pygraph_mcp-0.2.4}/tests/test_extractors.py

@@ -355,6 +355,76 @@ migrate.init_app(app)
         assert result["blueprints"] == []
         assert result["template_refs"] == []
 
+    def test_flask_restful_add_resource_single_path(self) -> None:
+        src = """
+from flask_restful import Resource, Api
+
+api = Api(app)
+
+class UserResource(Resource):
+    def get(self, id):
+        pass
+    def post(self):
+        pass
+
+api.add_resource(UserResource, '/users/<id>')
+"""
+        from pygraph.extractors.flask import extract_flask
+
+        result = extract_flask(src, "app.py")
+        routes = [r for r in result["routes"] if r.method == "WRAPPER" or r.method in ("GET", "POST", "PUT", "DELETE")]
+        matching = [r for r in routes if r.path == "/users/<id>"]
+        assert len(matching) == 2
+        methods = {r.method for r in matching}
+        assert methods == {"GET", "POST"}
+        assert all(r.handler == "app.py::UserResource" for r in matching)
+
+    def test_flask_restful_add_resource_multi_path(self) -> None:
+        src = """
+from flask_restful import Resource, Api
+
+api = Api(app)
+
+class ItemResource(Resource):
+    def get(self):
+        pass
+    def delete(self):
+        pass
+
+api.add_resource(ItemResource, '/items', '/items/<id>')
+"""
+        from pygraph.extractors.flask import extract_flask
+
+        result = extract_flask(src, "app.py")
+        matching = [r for r in result["routes"] if "ItemResource" in r.handler]
+        assert len(matching) == 4  # 2 paths × 2 methods
+        paths = {r.path for r in matching}
+        assert paths == {"/items", "/items/<id>"}
+        methods = {r.method for r in matching}
+        assert methods == {"GET", "DELETE"}
+
+    def test_flask_restful_no_methods_no_crash(self) -> None:
+        src = """
+api = Api(app)
+api.add_resource(PlainClass, '/plain')
+"""
+        from pygraph.extractors.flask import extract_flask
+
+        result = extract_flask(src, "app.py")
+        matching = [r for r in result["routes"] if r.path == "/plain"]
+        assert len(matching) == 1
+        assert matching[0].method == "GET"
+
+    def test_flask_restful_not_add_resource_not_detected(self) -> None:
+        src = """
+api.add_url_rule('/regular', view_func=handler)
+"""
+        from pygraph.extractors.flask import extract_flask
+
+        result = extract_flask(src, "app.py")
+        assert len(result["routes"]) == 1
+        assert result["routes"][0].path == "/regular"
+
 
 class TestComplexityExtraction:
     def test_base_complexity_one(self) -> None:

{pygraph_mcp-0.2.2 → pygraph_mcp-0.2.4}/uv.lock

@@ -804,7 +804,7 @@ wheels = [
 
 [[package]]
 name = "pygraph-mcp"
-version = "0.2.2"
+version = "0.2.4"
 source = { editable = "." }
 dependencies = [
     { name = "astroid" },