Files
python-appimage/python_appimage/tmp.py
Valentin Niess dc3acadb9a Initial commit
2020-03-29 11:59:23 +02:00

25 lines
496 B
Python

from contextlib import contextmanager as contextmanager
import os
import tempfile
from .fs import remove_tree
from .log import debug
__all__ = ['TemporaryDirectory']
@contextmanager
def TemporaryDirectory():
'''Create a temporary directory (Python 2 wrapper)
'''
tmpdir = tempfile.mkdtemp(prefix='python-appimage-')
debug('MKDIR', tmpdir)
pwd = os.getcwd()
os.chdir(tmpdir)
try:
yield tmpdir
finally:
os.chdir(pwd)
remove_tree(tmpdir)