tmp contexthandler

tmpdir_contexthandler.py

import os
import shutil
import tempfile
from contextlib import contextmanager

@contextmanager
def tmpdir(suffix='', prefix='tmp', dir=None):
    dirname = tempfile.mkdtemp(suffix=suffix, prefix=prefix, dir=dir)
    yield dirname
    shutil.rmtree(dirname)

with tmpdir() as d:
    print(d)