py2dag 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

py2dag/parser.py

@@ -14,6 +14,20 @@ def _literal(node: ast.AST) -> Any:
     """Return a Python literal from an AST node or raise DSLParseError."""
     if isinstance(node, ast.Constant):
         return node.value
+    if isinstance(node, ast.JoinedStr):
+        parts: List[str] = []
+        for value in node.values:
+            if isinstance(value, ast.Constant):
+                parts.append(str(value.value))
+            elif isinstance(value, ast.FormattedValue):
+                try:
+                    expr = ast.unparse(value.value)  # type: ignore[attr-defined]
+                except Exception:
+                    expr = "?"
+                parts.append("{" + expr + "}")
+            else:
+                raise DSLParseError("Keyword argument values must be JSON-serialisable literals")
+        return "".join(parts)
     if isinstance(node, (ast.List, ast.Tuple)):
         return [_literal(elt) for elt in node.elts]
     if isinstance(node, ast.Dict):
@@ -722,11 +736,28 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
         for stmt in body:
             _parse_stmt(stmt)
 
+        # If no explicit return was encountered, emit a terminal break node so
+        # that the plan represents function completion.
+        break_id: Optional[str] = None
+        if returned_var is None:
+            break_id = _ssa_new("break")
+            ops.append({"id": break_id, "op": "CTRL.break", "deps": [], "args": {}})
+
+        # If no outputs were produced, synthesise a default return of `None` so
+        # that parsing succeeds for empty functions.
+        if returned_var is None and not outputs:
+            const_id = _ssa_new("return_value")
+            ops.append({
+                "id": const_id,
+                "op": "CONST.value",
+                "deps": [],
+                "args": {"value": None},
+            })
+            returned_var = const_id
+
         if not outputs:
-            if returned_var is not None:
-                outputs.append({"from": returned_var, "as": "return"})
-            else:
-                raise DSLParseError("At least one output() call required")
+            outputs.append({"from": returned_var if returned_var is not None else break_id, "as": "return"})
+
         if len(ops) > 2000:
             raise DSLParseError("Too many operations")
 

{py2dag-0.3.6.dist-info → py2dag-0.3.8.dist-info}/METADATA

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

{py2dag-0.3.6.dist-info → py2dag-0.3.8.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=DM4Zi5c9YWX_6o6GsyeEyp1sNO5R0e9oiPLj6IJFz-U,35726
+py2dag/parser.py,sha256=WKLT-2qYpcGvlwU9hQ7o9wIFIxU328mlp8ntaAai5KY,37026
 py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
-py2dag-0.3.6.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
-py2dag-0.3.6.dist-info/METADATA,sha256=BiIraaTzBmP-yCzdmoX7mQ945OR0r2juBol_ouEtwYk,3549
-py2dag-0.3.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-py2dag-0.3.6.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
-py2dag-0.3.6.dist-info/RECORD,,
+py2dag-0.3.8.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
+py2dag-0.3.8.dist-info/METADATA,sha256=nDc9Jce8EqZbUy3vum4zOwJJ20We7geFRYKfFGyMbSE,3549
+py2dag-0.3.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+py2dag-0.3.8.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
+py2dag-0.3.8.dist-info/RECORD,,