15 lines
307 B
Python
15 lines
307 B
Python
|
from contextlib import contextmanager
|
||
|
|
import shutil
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
try:
|
||
|
|
from tempfile import TemporaryDirectory
|
||
|
|
except ImportError:
|
||
|
|
@contextmanager
|
||
|
|
def TemporaryDirectory():
|
||
|
|
name = tempfile.mkdtemp()
|
||
|
|
try:
|
||
|
|
yield name
|
||
|
|
finally:
|
||
|
|
shutil.rmtree(name)
|