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
|
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
|
