@neat.is/core 0.6.2-dev.20260723 → 0.6.2

package/dist/{chunk-ZZ3VUCWL.js → chunk-AOWANF2K.js}

@@ -4977,7 +4977,7 @@ function keywordArrayStrings(argsNode, key) {
   }
   return [];
 }
-function collectApiRouterPrefixes(root) {
+function collectPythonRouterPrefixes(root) {
   const prefixes = /* @__PURE__ */ new Map();
   walk(root, (node) => {
     if (node.type !== "assignment") return;
@@ -4986,7 +4986,8 @@ function collectApiRouterPrefixes(root) {
     const fn = right.childForFieldName("function");
     if (!fn) return;
     const ctor = fn.type === "attribute" ? fn.childForFieldName("attribute")?.text : fn.text;
-    if (ctor !== "APIRouter") return;
+    const prefixKey = ctor === "APIRouter" ? "prefix" : ctor === "Blueprint" ? "url_prefix" : null;
+    if (!prefixKey) return;
     const left = node.childForFieldName("left");
     if (!left || left.type !== "identifier") return;
     const args = right.childForFieldName("arguments");
@@ -4994,7 +4995,7 @@ function collectApiRouterPrefixes(root) {
     for (let i = 0; i < args.namedChildCount; i++) {
       const arg = args.namedChild(i);
       if (arg?.type !== "keyword_argument") continue;
-      if (arg.childForFieldName("name")?.text !== "prefix") continue;
+      if (arg.childForFieldName("name")?.text !== prefixKey) continue;
       const val = arg.childForFieldName("value");
       const p = val ? pyStaticStringText(val) : null;
       if (p !== null) prefixes.set(left.text, p);
@@ -5002,9 +5003,9 @@ function collectApiRouterPrefixes(root) {
   });
   return prefixes;
 }
-function fastapiRoutesFromSource(source, parser) {
+function pythonRoutesFromSource(source, parser, framework) {
   const tree = parseSource2(parser, source);
-  const prefixes = collectApiRouterPrefixes(tree.rootNode);
+  const prefixes = collectPythonRouterPrefixes(tree.rootNode);
   const out = [];
   walk(tree.rootNode, (node) => {
     if (node.type !== "decorator") return;
@@ -5015,7 +5016,9 @@ function fastapiRoutesFromSource(source, parser) {
     const method = fn.childForFieldName("attribute")?.text?.toLowerCase();
     if (!method) return;
     const isVerb = FASTAPI_METHODS.has(method);
-    if (!isVerb && method !== "api_route") return;
+    const isFlaskRoute = method === "route";
+    const isApiRoute = method === "api_route";
+    if (!isVerb && !isFlaskRoute && !isApiRoute) return;
     const args = call.childForFieldName("arguments");
     const first = args?.namedChild(0);
     if (!first || first.type !== "string") return;
@@ -5026,17 +5029,41 @@ function fastapiRoutesFromSource(source, parser) {
     const pathTemplate = canonicalizeTemplate(prefix + rawPath);
     const line = node.startPosition.row + 1;
     if (isVerb) {
-      out.push({ method: method.toUpperCase(), pathTemplate, line, framework: "fastapi" });
+      out.push({ method: method.toUpperCase(), pathTemplate, line, framework });
       return;
     }
     const methods = keywordArrayStrings(args, "methods");
-    const list = methods.length > 0 ? methods : ["ALL"];
+    const list = methods.length > 0 ? methods : isFlaskRoute ? ["GET"] : ["ALL"];
     for (const m of list) {
+      out.push({ method: m === "ALL" ? "ALL" : m.toUpperCase(), pathTemplate, line, framework });
+    }
+  });
+  return out;
+}
+function djangoRoutesFromSource(source, parser) {
+  const tree = parseSource2(parser, source);
+  const out = [];
+  walk(tree.rootNode, (node) => {
+    if (node.type !== "assignment") return;
+    if (node.childForFieldName("left")?.text !== "urlpatterns") return;
+    const list = node.childForFieldName("right");
+    if (!list || list.type !== "list") return;
+    for (let i = 0; i < list.namedChildCount; i++) {
+      const el = list.namedChild(i);
+      if (el?.type !== "call") continue;
+      if (el.childForFieldName("function")?.text !== "path") continue;
+      const args = el.childForFieldName("arguments");
+      const first = args?.namedChild(0);
+      if (first?.type !== "string") continue;
+      const raw = pyStaticStringText(first);
+      if (raw === null) continue;
+      const second = args?.namedChild(1);
+      if (second?.type === "call" && second.childForFieldName("function")?.text === "include") continue;
       out.push({
-        method: m === "ALL" ? "ALL" : m.toUpperCase(),
-        pathTemplate,
-        line,
-        framework: "fastapi"
+        method: "ALL",
+        pathTemplate: canonicalizeTemplate(raw),
+        line: el.startPosition.row + 1,
+        framework: "django"
       });
     }
   });
@@ -5057,7 +5084,10 @@ async function addRoutes(graph, services) {
     const hasHono = deps["hono"] !== void 0;
     const hasNext = deps["next"] !== void 0;
     const hasFastapi = deps["fastapi"] !== void 0;
-    if (!hasExpress && !hasFastify && !hasHono && !hasNext && !hasFastapi) continue;
+    const hasFlask = deps["flask"] !== void 0;
+    const hasDjango = deps["django"] !== void 0;
+    if (!hasExpress && !hasFastify && !hasHono && !hasNext && !hasFastapi && !hasFlask && !hasDjango)
+      continue;
     const files = await loadSourceFiles(service.dir);
     for (const file of files) {
       if (isTestPath(file.path)) continue;
@@ -5068,7 +5098,8 @@ async function addRoutes(graph, services) {
       let routes;
       try {
         if (isPy) {
-          routes = hasFastapi ? fastapiRoutesFromSource(file.content, pyParser) : [];
+          routes = hasFastapi || hasFlask ? pythonRoutesFromSource(file.content, pyParser, hasFastapi ? "fastapi" : "flask") : [];
+          if (hasDjango) routes = routes.concat(djangoRoutesFromSource(file.content, pyParser));
         } else if (hasNext && (isNextAppRouteFile(relFile) || isNextPagesApiFile(relFile))) {
           routes = nextRoutesFromFile(file.content, relFile, jsParser);
         } else if (hasExpress || hasFastify || hasHono) {
@@ -9995,4 +10026,4 @@ export {
   recordConnectorPoll,
   buildApi
 };
-//# sourceMappingURL=chunk-ZZ3VUCWL.js.map
+//# sourceMappingURL=chunk-AOWANF2K.js.map