py2dag 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

py2dag/parser.py

@@ -101,6 +101,15 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                         "deps": deps,
                         "args": {"template": template},
                     })
+                elif isinstance(value, (ast.Constant, ast.List, ast.Tuple, ast.Dict)):
+                    # Allow assigning JSON-serialisable literals directly
+                    lit = _literal(value)
+                    ops.append({
+                        "id": var_name,
+                        "op": "CONST.value",
+                        "deps": [],
+                        "args": {"value": lit},
+                    })
                 else:
                     raise DSLParseError("Right hand side must be a call or f-string")
                 defined.add(var_name)
@@ -144,9 +153,9 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                     if var not in defined:
                         raise DSLParseError(f"Undefined return variable: {var}")
                     returned_var = var
-                elif isinstance(stmt.value, ast.Constant):
-                    # Support returning a literal (e.g., "DONE"): synthesize a const op
-                    lit = stmt.value.value
+                elif isinstance(stmt.value, (ast.Constant, ast.List, ast.Tuple, ast.Dict)):
+                    # Support returning a JSON-serialisable literal (str/num/bool/None, list/tuple, dict)
+                    lit = _literal(stmt.value)
                     const_id_base = "return_value"
                     const_id = const_id_base
                     n = 1
@@ -162,6 +171,9 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                     returned_var = const_id
                 else:
                     raise DSLParseError("return must return a variable name or literal")
+            elif isinstance(stmt, (ast.For, ast.AsyncFor, ast.While)):
+                # Ignore control flow blocks; only top-level linear statements are modeled
+                continue
             else:
                 raise DSLParseError("Only assignments, expression calls, and a final return are allowed in function body")
 

{py2dag-0.1.8.dist-info → py2dag-0.1.10.dist-info}/METADATA

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

py2dag-0.1.10.dist-info/RECORD

@@ -0,0 +1,11 @@
+py2dag/__init__.py,sha256=i8VB44JCVcRJAcvnQtbH8YVRUz5j7dE355iRbikXPGQ,250
+py2dag/cli.py,sha256=BCBi5mNxOqeEN8uEMt_hiDx0iSt7ZE3y74cGXREzZ2I,1296
+py2dag/export_dagre.py,sha256=S244wUxBMuM9qXD8bklaslkMp3rcbBGWvMhwWVBzBF0,3487
+py2dag/export_svg.py,sha256=YyjqOuj8GhUTDWP70SKnnSWAKI1PvJwyOhHLwB29uNM,1812
+py2dag/parser.py,sha256=nj3QE_6iPBktdBQFt5DF0YYitPKVQRmtcuLxG9WyQjE,10444
+py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
+py2dag-0.1.10.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
+py2dag-0.1.10.dist-info/METADATA,sha256=0VGU_tJB5PeSIuZVr2ZTUkxqvXCgcMnl5QUHdOzSN2w,3550
+py2dag-0.1.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+py2dag-0.1.10.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
+py2dag-0.1.10.dist-info/RECORD,,

py2dag-0.1.8.dist-info/RECORD

@@ -1,11 +0,0 @@
-py2dag/__init__.py,sha256=i8VB44JCVcRJAcvnQtbH8YVRUz5j7dE355iRbikXPGQ,250
-py2dag/cli.py,sha256=BCBi5mNxOqeEN8uEMt_hiDx0iSt7ZE3y74cGXREzZ2I,1296
-py2dag/export_dagre.py,sha256=S244wUxBMuM9qXD8bklaslkMp3rcbBGWvMhwWVBzBF0,3487
-py2dag/export_svg.py,sha256=YyjqOuj8GhUTDWP70SKnnSWAKI1PvJwyOhHLwB29uNM,1812
-py2dag/parser.py,sha256=1U16-GOQMa8oriJLePJ2TQlRZsrC9A4apy5V-V10M1A,9774
-py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
-py2dag-0.1.8.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
-py2dag-0.1.8.dist-info/METADATA,sha256=PvQOVUwFviTr7G_m_sUsQeBGQtbedpZfxjbEiVsZvko,3549
-py2dag-0.1.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-py2dag-0.1.8.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
-py2dag-0.1.8.dist-info/RECORD,,