A context to use temporary directory and don't have to 'remember' to delete it

ctxmgr.py

import os
import tempfile
import shutil
from contextlib import contextmanager


@contextmanager
def mkdtemp(*args, **kwargs):
    d = tempfile.mkdtemp(*args, **kwargs)
    try:
        yield d
    finally:
        shutil.rmtree(d)


with mkdtemp() as tmpdir:
    # here we can use a temporary directory that will be removed
    assert os.path.isdir(tmpdir)

assert not os.path.isdir(tmpdir)