py2dag 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

py2dag/parser.py

@@ -108,7 +108,7 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
             deps: List[str] = []
             for n in ast.walk(node):
                 if isinstance(n, ast.Name) and isinstance(n.ctx, ast.Load):
-                    if n.id not in callees and n.id not in deps:
+                    if n.id not in callees and n.id not in deps and n.id in latest:
                         deps.append(n.id)
             return deps
 
@@ -603,7 +603,10 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                 raise DSLParseError("Only assignments, control flow, settings/output calls, and return are allowed in function body")
 
         # Parse body sequentially; still require a resulting output
-        for i, stmt in enumerate(fn.body):  # type: ignore[attr-defined]
+        body = list(getattr(fn, "body", []))  # type: ignore[attr-defined]
+        if body and isinstance(body[0], ast.Expr) and isinstance(getattr(body[0], "value", None), ast.Constant) and isinstance(getattr(body[0].value, "value", None), str):  # type: ignore[attr-defined]
+            body = body[1:]
+        for stmt in body:
             _parse_stmt(stmt)
 
         if not outputs:

{py2dag-0.3.0.dist-info → py2dag-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: py2dag
-Version: 0.3.0
+Version: 0.3.2
 Summary: Convert Python function plans to DAG (JSON, pseudo, optional SVG).
 License: MIT
 Author: rvergis

{py2dag-0.3.0.dist-info → py2dag-0.3.2.dist-info}/RECORD

@@ -3,10 +3,10 @@ py2dag/cli.py,sha256=q8ocafpGtkrls1mx9QiFU1wL6KKeeJhT0Oklk6SwHB0,15151
 py2dag/colors.py,sha256=kVMVWUKJY1uLQFQux5XKiEcofDFpNPQoAEzpMJWs2b0,604
 py2dag/export_dagre.py,sha256=Og-oVjKzc7bRgeU7rgLHDjiY6wIcNyr2nIYlSAQzKoM,6041
 py2dag/export_svg.py,sha256=3ZmZqxIUidSbuKIIh5kFqhfVnoqZ6PbEQpMih4AX9xo,3793
-py2dag/parser.py,sha256=Z0X_IyHF1EzM_HPxP9-eNMBGPoHuyhRAqJmX3zXH92k,30399
+py2dag/parser.py,sha256=e3_w2Yhs2pG4FWah56E5tMzgCYqjKy9mKWmlEj6_46U,30676
 py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
-py2dag-0.3.0.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
-py2dag-0.3.0.dist-info/METADATA,sha256=9AQIxET2CsznM7lzAGM47GBQE79oHTPjNHepQnHtVSM,3549
-py2dag-0.3.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-py2dag-0.3.0.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
-py2dag-0.3.0.dist-info/RECORD,,
+py2dag-0.3.2.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
+py2dag-0.3.2.dist-info/METADATA,sha256=khrjr13YasQph1VsQPXD1Hp5MEpuoJ60oDdqvqHglHc,3549
+py2dag-0.3.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+py2dag-0.3.2.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
+py2dag-0.3.2.dist-info/RECORD,,