36 lines
993 B
Python
36 lines
993 B
Python
from collections.abc import Iterator
|
|
from contextlib import contextmanager
|
|
from pathlib import Path
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
from app.native.wndb.inp.exporter import dump_inp
|
|
|
|
|
|
PROJECT_INP_DIRECTORY = Path("db_inp")
|
|
|
|
|
|
@contextmanager
|
|
def temporary_project_inp(
|
|
project_code: str,
|
|
*,
|
|
purpose: str,
|
|
version: str = "2",
|
|
) -> Iterator[Path]:
|
|
"""Export one project model to an isolated INP and remove it afterwards."""
|
|
PROJECT_INP_DIRECTORY.mkdir(parents=True, exist_ok=True)
|
|
with NamedTemporaryFile(
|
|
dir=PROJECT_INP_DIRECTORY,
|
|
prefix=f"{purpose}_",
|
|
suffix=".inp",
|
|
delete=False,
|
|
) as temporary_file:
|
|
path = Path(temporary_file.name).resolve()
|
|
|
|
try:
|
|
dump_inp(project_code, str(path), version)
|
|
if not path.is_file() or path.stat().st_size == 0:
|
|
raise ValueError(f"项目 {project_code!r} 的 INP 导出失败")
|
|
yield path
|
|
finally:
|
|
path.unlink(missing_ok=True)
|