1 import shutil 2 import tempfile 3 4 -class TempDir: 5 - def __init__(self, parent_dir=None, prefix="", suffix=""): 6 self.__path = tempfile.mkdtemp( 7 dir=parent_dir, prefix=prefix, suffix=suffix) 8 9 - def get_path(self): 10 return self.__path 11 12 - def delete(self): 13 shutil.rmtree(self.__path) 14 15 - def __enter__(self): 16 return self 17 18 - def __exit__(self, type_, value, traceback): 19 self.delete() 20