tmpdir_provider.py 291 B

123456789101112
  1. from tempfile import TemporaryDirectory
  2. from typing import Optional
  3. __all__ = ["TmpDirProvider"]
  4. class TmpDirProvider:
  5. def __init__(self, dir: Optional[str] = None):
  6. self.dir = dir
  7. def __call__(self) -> TemporaryDirectory:
  8. return TemporaryDirectory(dir=self.dir)