14 lines
307 B
Python
14 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)
|