File size: 560 Bytes
e28c9e4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | from __future__ import annotations
import shutil
import uuid
from contextlib import contextmanager
from pathlib import Path
from typing import Iterator
@contextmanager
def workspace_tempdir(prefix: str) -> Iterator[Path]:
base_dir = Path(__file__).resolve().parent.parent / "data" / "test-artifacts"
base_dir.mkdir(parents=True, exist_ok=True)
temp_dir = base_dir / f"{prefix}{uuid.uuid4().hex}"
temp_dir.mkdir(parents=True, exist_ok=True)
try:
yield temp_dir
finally:
shutil.rmtree(temp_dir, ignore_errors=True)
|