py2dag 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

py2dag/parser.py

@@ -62,7 +62,7 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
         versions: Dict[str, int] = {}
         latest: Dict[str, str] = {}
         context_suffix: str = ""
-        ctx_counts: Dict[str, int] = {"if": 0, "loop": 0, "while": 0}
+        ctx_counts: Dict[str, int] = {"if": 0, "loop": 0, "while": 0, "except": 0}
 
         def _ssa_new(name: str) -> str:
             if not VALID_NAME_RE.match(name):
@@ -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
 
@@ -593,6 +593,29 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                         "args": {"var": var},
                     })
                 return None
+            elif isinstance(stmt, ast.Try):
+                pre_versions = dict(versions)
+                pre_latest = dict(latest)
+                for inner in stmt.body:
+                    _parse_stmt(inner)
+                # else block executes only if no exception
+                for inner in getattr(stmt, "orelse", []):
+                    _parse_stmt(inner)
+                post_versions = versions
+                post_latest = latest
+                for handler in getattr(stmt, "handlers", []):
+                    saved_ctx = context_suffix
+                    ctx_counts["except"] = ctx_counts.get("except", 0) + 1
+                    context_suffix = f"except{ctx_counts['except']}"
+                    saved_versions, saved_latest = versions, latest
+                    versions, latest = dict(pre_versions), dict(pre_latest)
+                    for inner in handler.body:
+                        _parse_stmt(inner)
+                    versions, latest = post_versions, post_latest
+                    context_suffix = saved_ctx
+                for inner in getattr(stmt, "finalbody", []):
+                    _parse_stmt(inner)
+                return None
             elif isinstance(stmt, ast.Break):
                 ssa = _ssa_new("break")
                 ops.append({"id": ssa, "op": "CTRL.break", "deps": [], "args": {}})

{py2dag-0.3.1.dist-info → py2dag-0.3.3.dist-info}/METADATA

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

{py2dag-0.3.1.dist-info → py2dag-0.3.3.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=bpwOZVsJUS3fRQTFfm2CfTxqlTWUYaLYDOHJRA2Yj34,30657
+py2dag/parser.py,sha256=WN2ZbOZdc05mOr60Q-IDA9TPsTX6tKb_VuDdBEBFeq8,31862
 py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
-py2dag-0.3.1.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
-py2dag-0.3.1.dist-info/METADATA,sha256=GhuNW_nWmz0P0eNjX6Zs2y65AglkhUT1lplPeDzxmaU,3549
-py2dag-0.3.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-py2dag-0.3.1.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
-py2dag-0.3.1.dist-info/RECORD,,
+py2dag-0.3.3.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
+py2dag-0.3.3.dist-info/METADATA,sha256=Y6H9SC22ONmNes0_2ELcSAOtJlbsEpXKr0lzi2f2zss,3549
+py2dag-0.3.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+py2dag-0.3.3.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
+py2dag-0.3.3.dist-info/RECORD,,