File size: 924 Bytes
bbae060 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | from __future__ import annotations
import shutil
import tempfile
from ._errors import OperationFailed
from ._osfs import OSFS
class TempFS(OSFS):
def __init__(self, auto_clean: bool = True, ignore_clean_errors: bool = True):
self.auto_clean = auto_clean
self.ignore_clean_errors = ignore_clean_errors
self._temp_dir = tempfile.mkdtemp("__temp_fs__")
self._cleaned = False
super().__init__(self._temp_dir)
def close(self):
if self.auto_clean:
self.clean()
super().close()
def clean(self):
if self._cleaned:
return
try:
shutil.rmtree(self._temp_dir)
except Exception as e:
if not self.ignore_clean_errors:
raise OperationFailed(
f"failed to remove temporary directory: {self._temp_dir!r}"
) from e
self._cleaned = True
|