Support for temporary directories in python2.7 as context managers (already available in python3.2+)

temporary_directory.py

import shutil
import tempfile


class TemporaryDirectory(object):
    """
    Context manager for tempfile.mkdtemp().

    This class is available in python +v3.2.

    """
    def __enter__(self):
        self.dir_name = tempfile.mkdtemp()
        return self.dir_name

    def __exit__(self, exc_type, exc_value, traceback):
        shutil.rmtree(self.dir_name)


TemporaryDirectory = getattr(tempfile, 'TemporaryDirectory',
                             TemporaryDirectory)