91 lines
4.2 KiB
Python
91 lines
4.2 KiB
Python
|
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())
|