py2dag 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

py2dag/parser.py

@@ -139,12 +139,29 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
             elif isinstance(stmt, ast.Return):
                 if i != len(fn.body) - 1:  # type: ignore[index]
                     raise DSLParseError("return must be the last statement")
-                if not isinstance(stmt.value, ast.Name):
-                    raise DSLParseError("return must return a variable name")
-                var = stmt.value.id
-                if var not in defined:
-                    raise DSLParseError(f"Undefined return variable: {var}")
-                returned_var = var
+                if isinstance(stmt.value, ast.Name):
+                    var = stmt.value.id
+                    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
+                    const_id_base = "return_value"
+                    const_id = const_id_base
+                    n = 1
+                    while const_id in defined:
+                        const_id = f"{const_id_base}_{n}"
+                        n += 1
+                    ops.append({
+                        "id": const_id,
+                        "op": "CONST.value",
+                        "deps": [],
+                        "args": {"value": lit},
+                    })
+                    returned_var = const_id
+                else:
+                    raise DSLParseError("return must return a variable name or literal")
             else:
                 raise DSLParseError("Only assignments, expression calls, and a final return are allowed in function body")
 
@@ -156,7 +173,9 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
         if len(ops) > 200:
             raise DSLParseError("Too many operations")
 
-        plan: Dict[str, Any] = {"version": 1, "ops": ops, "outputs": outputs}
+        # Include the parsed function name for visibility/debugging
+        fn_name = getattr(fn, "name", None)  # type: ignore[attr-defined]
+        plan: Dict[str, Any] = {"version": 1, "function": fn_name, "ops": ops, "outputs": outputs}
         if settings:
             plan["settings"] = settings
         return plan

{py2dag-0.1.5.dist-info → py2dag-0.1.7.dist-info}/METADATA

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

{py2dag-0.1.5.dist-info → py2dag-0.1.7.dist-info}/RECORD

@@ -2,10 +2,10 @@ 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=2sYscG3hFHK7RxJ6rNlO1w9GnxF-9hZG6SMAs_oGDu0,8849
+py2dag/parser.py,sha256=1U16-GOQMa8oriJLePJ2TQlRZsrC9A4apy5V-V10M1A,9774
 py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
-py2dag-0.1.5.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
-py2dag-0.1.5.dist-info/METADATA,sha256=yEDK8WOlw2SYEa5J_3H7OINAfFLmPLgvi0UzJ50O9Cg,3537
-py2dag-0.1.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-py2dag-0.1.5.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
-py2dag-0.1.5.dist-info/RECORD,,
+py2dag-0.1.7.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
+py2dag-0.1.7.dist-info/METADATA,sha256=MEkl1D4hDnS69pFadkzMpLkULnek1jCo4R7_of6PqjE,3537
+py2dag-0.1.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+py2dag-0.1.7.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
+py2dag-0.1.7.dist-info/RECORD,,