py2dag 0.3.9__tar.gz → 0.3.11__tar.gz

{py2dag-0.3.9 → py2dag-0.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: py2dag
-Version: 0.3.9
+Version: 0.3.11
 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.9 → py2dag-0.3.11}/README.md

@@ -1,5 +1,14 @@
 # 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
 
@@ -45,17 +54,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.9 → py2dag-0.3.11}/py2dag/parser.py

@@ -298,8 +298,14 @@ def parse(source: str, function_name: Optional[str] = None) -> Dict[str, Any]:
                             tmp_id = _emit_assign_from_call(f"{var_name}_field", v_node)
                             deps.append(tmp_id)
                         else:
-                            # Synthesize const for literal value
-                            lit_val = _literal(v_node)
+                            # Synthesize const for literal or arbitrary expression value
+                            try:
+                                lit_val = _literal(v_node)
+                            except DSLParseError:
+                                try:
+                                    lit_val = ast.unparse(v_node)  # type: ignore[attr-defined]
+                                except Exception:
+                                    lit_val = v_node.__class__.__name__
                             tmp = _ssa_new(f"{var_name}_lit")
                             ops.append({
                                 "id": tmp,
@@ -509,15 +515,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.9 → py2dag-0.3.11}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "py2dag"
-version = "0.3.9"
+version = "0.3.11"
 description = "Convert Python function plans to DAG (JSON, pseudo, optional SVG)."
 authors = ["rvergis"]
 license = "MIT"