Making a temporary directory with a context manager.

make_temp_dir.py

import contextlib
import datetime
import tempfile
import shutil

@contextlib.contextmanager
def make_temp_dir(prefix=""):
    "A context manager for a temporary directory."
    if prefix:
        prefix += "."
    prefix += datetime.datetime.now().strftime("%Y%m%d%H%M%S.")
    temp_dir = tempfile.mkdtemp(prefix=prefix)
    yield temp_dir
    shutil.rmtree(temp_dir)

if __name__ == '__main__':
    with make_temp_dir() as tmpdir:
        print(tmpdir)