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

py2dag/parser.py

@@ -509,15 +509,19 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                         })
                         returned_var = const_id
                 elif isinstance(stmt.value, (ast.Constant, ast.List, ast.Tuple, ast.Dict)):
-                    lit = _literal(stmt.value)
-                    const_id = _ssa_new("return_value")
-                    ops.append({
-                        "id": const_id,
-                        "op": "CONST.value",
-                        "deps": [],
-                        "args": {"value": lit},
-                    })
-                    returned_var = const_id
+                    try:
+                        lit = _literal(stmt.value)
+                    except DSLParseError:
+                        returned_var = _emit_assign_from_literal_or_pack("return_value", stmt.value)
+                    else:
+                        const_id = _ssa_new("return_value")
+                        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")
                 return None

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: py2dag
-Version: 0.3.8
+Version: 0.3.10
 Summary: Convert Python function plans to DAG (JSON, pseudo, optional SVG).
 License: MIT
 Author: rvergis
@@ -18,7 +18,16 @@ Requires-Dist: graphviz (>=0.20.3,<0.21.0) ; extra == "svg"
 Description-Content-Type: text/markdown
 
 # py2dag
-Convert Python function plans to a DAG (JSON, pseudo, optional SVG).
+Convert Python function plans to a DAG (JSON, pseudo, optional HTML/SVG).
+
+## Features
+
+- Writes `plan.json` with explicit `nodes` and `edges` for downstream tools.
+- Generates human-readable pseudo code at `plan.pseudo`.
+- Optional exports:
+  - `--html` for an interactive Dagre graph with color-coded nodes.
+  - `--svg` for a Graphviz-rendered SVG.
+- Node colors are chosen deterministically so the same op type is colored consistently across runs.
 
 ## Install
 
@@ -64,17 +73,17 @@ make run FILE=path/to/your_file.py ARGS=--html
 - Run the installed CLI (after `pip install py2dag`):
 
 ```
-py2dag path/to/your_file.py --html
+py2dag path/to/your_file.py --html  # interactive HTML
+py2dag path/to/your_file.py --svg   # Graphviz SVG
 ```
 
-
 - Or directly with Python (inside venv):
 
 ```
 poetry run python cli.py path/to/your_file.py --html
 ```
 
-This generates `plan.json`, `plan.pseudo`, and if `--html` is used, `plan.html`.
+This generates `plan.json`, `plan.pseudo`, and if `--html`/`--svg` is used, `plan.html` or `plan.svg`.
 
 - Function name: By default the tool auto-detects a suitable function in the file. To target a specific function, pass `--func NAME`.
 

py2dag-0.3.10.dist-info/RECORD

@@ -0,0 +1,12 @@
+py2dag/__init__.py,sha256=i8VB44JCVcRJAcvnQtbH8YVRUz5j7dE355iRbikXPGQ,250
+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=9UwUxarXmLXYyiEf2xg2gxcAurhwG9kG6bKbh0FcEFU,37256
+py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
+py2dag-0.3.10.dist-info/LICENSE,sha256=3Qee1EPwej_nusovTbyIQ8LvD2rXHdM0c6LNwk_D8Kc,1067
+py2dag-0.3.10.dist-info/METADATA,sha256=DIOEg22N1JL5m_foB-gdYMxvxytX8yMU54O6HpQuHMA,4024
+py2dag-0.3.10.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+py2dag-0.3.10.dist-info/entry_points.txt,sha256=Q0SHexJJ0z1te4AYL1xTZogx5FrxCCE1ZJ5qntkFMZs,42
+py2dag-0.3.10.dist-info/RECORD,,

py2dag-0.3.8.dist-info/RECORD

@@ -1,12 +0,0 @@
-py2dag/__init__.py,sha256=i8VB44JCVcRJAcvnQtbH8YVRUz5j7dE355iRbikXPGQ,250
-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=WKLT-2qYpcGvlwU9hQ7o9wIFIxU328mlp8ntaAai5KY,37026
-py2dag/pseudo.py,sha256=NJK61slyFLtSjhj8gJDJneUInEpBN57_41g8IfHNPWI,922
-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,,