feat: 推进一级关系参数化编辑与参数导出

2026-08-13 17:50:20 +08:00
parent 19364d81b5
commit 7d814d2939
21 changed files with 2075 additions and 231 deletions
@@ -0,0 +1,90 @@
+from __future__ import annotations
+
+import json
+from pathlib import Path
+import py_compile
+import sys
+import tempfile
+
+PROJECT_ROOT = Path(__file__).resolve().parent.parent
+if str(PROJECT_ROOT) not in sys.path:
+    sys.path.insert(0, str(PROJECT_ROOT))
+
+from step_editor.parametric_component import export_parametric_component
+
+
+def _assert(condition: bool, message: str) -> None:
+    if not condition:
+        raise AssertionError(message)
+
+
+def main() -> int:
+    with tempfile.TemporaryDirectory(prefix="geom_param_component_export_") as temp_dir:
+        root = Path(temp_dir)
+        source_step = root / "source.step"
+        source_step.write_text("ISO-10303-21;\nEND-ISO-10303-21;\n", encoding="utf-8")
+        parameters = [
+            {
+                "name": "面内长度",
+                "displayName": "面内长度",
+                "type": "number",
+                "ioRole": "input",
+                "default": "10",
+            }
+        ]
+        edits = [
+            {
+                "parameter": "面内长度",
+                "displayName": "面内长度",
+                "targetKind": "feature",
+                "targetId": 0,
+                "uiAction": "resize_face_width_local",
+                "operation": "resize_face_size_local",
+                "args": [0, {"param": "面内长度"}, "width"],
+                "default": 10.0,
+                "valueType": "positive",
+                "scope": "local",
+                "scopeLabel": "局部重建",
+                "sourceStep": str(source_step),
+                "parameterKey": "local_face_width",
+            }
+        ]
+        main_py = export_parametric_component(
+            parameters=parameters,
+            edits=edits,
+            source_step=source_step,
+            component_root=root / "nodes",
+            component_name="测试组件",
+        )
+        text = main_py.read_text(encoding="utf-8")
+        _assert("INPUT_PARAMETERS = " in text, "generated main.py should embed selected input parameter list")
+        _assert("PARAMETERS = INPUT_PARAMETERS + OUTPUT_PARAMETERS" in text, "generated main.py should expose FlowEditor parameters")
+        _assert("NODE_INFO = " in text, "generated main.py should expose FlowEditor node info")
+        _assert("def execute(inputs, params, context):" in text, "generated main.py should expose FlowEditor execute entry")
+        _assert("COMPONENT = " in text, "generated main.py should embed execution config")
+        _assert("step_edit_config.json" not in text, "generated component should not require a sidecar config JSON")
+        _assert(not (main_py.parent / "data.json").exists(), "component directory should not contain data.json")
+        _assert(not (main_py.parent / "step_edit_config.json").exists(), "component directory should not contain step_edit_config.json")
+        py_compile.compile(str(main_py), doraise=True)
+
+        namespace: dict[str, object] = {}
+        exec(compile(text, str(main_py), "exec"), namespace)
+        embedded_input_parameters = namespace.get("INPUT_PARAMETERS")
+        embedded_parameters = namespace.get("PARAMETERS")
+        node_info = namespace.get("NODE_INFO")
+        embedded_component = namespace.get("COMPONENT")
+        _assert(embedded_input_parameters == parameters, f"embedded INPUT_PARAMETERS mismatch: {embedded_input_parameters}")
+        _assert(isinstance(embedded_parameters, list), "embedded PARAMETERS should be a list")
+        _assert(embedded_parameters[: len(parameters)] == parameters, f"embedded input parameter prefix mismatch: {embedded_parameters}")
+        _assert(any(row.get("name") == "output_step" and row.get("ioRole") == "output" for row in embedded_parameters if isinstance(row, dict)), "generated PARAMETERS should include output_step output port")
+        _assert(isinstance(node_info, dict), "NODE_INFO should be a dict")
+        _assert(node_info.get("parameters") == embedded_parameters, "NODE_INFO should point to PARAMETERS")
+        _assert(isinstance(embedded_component, dict), "embedded COMPONENT should be a dict")
+        _assert(embedded_component.get("edits") == edits, f"embedded edits mismatch: {json.dumps(embedded_component, ensure_ascii=False)}")
+
+    print("parametric component export ok")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())