Jerry
28 lines
686 B
Python
28 lines
686 B
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import build123d
|
|
|
|
|
|
def _compound_from_shapes(
|
|
shapes: list[build123d.Shape],
|
|
*,
|
|
label: str = "",
|
|
) -> build123d.Compound:
|
|
return build123d.Compound(obj=shapes, children=shapes, label=label)
|
|
|
|
|
|
def import_as_shape(step_path: Path) -> build123d.Shape:
|
|
imported = build123d.import_step(step_path)
|
|
solids = imported.solids()
|
|
if solids:
|
|
return _compound_from_shapes(solids)
|
|
|
|
if isinstance(imported, build123d.Shape):
|
|
return imported
|
|
|
|
if not imported.children:
|
|
raise RuntimeError(f"No CAD shapes found in {step_path}")
|
|
return _compound_from_shapes(imported.children)
|