First stab at temp_dir context manager.

README.rst

@@ -1,5 +1,5 @@
 Temporary |Version| |Build| |Coverage| |Health|
-===================================================================
+===============================================
 
 |Compatibility| |Implementations| |Format| |Downloads|
 

requirements_install.txt

@@ -0,0 +1 @@
+contextlib2

setup.py

@@ -8,7 +8,8 @@
     license='MIT',
     author='Matthew Tardiff',
     author_email='mattrix@gmail.com',
-    install_requires=(),
+    install_requires=(
+        'contextlib2',),
     tests_require=(
         'nose',
         'flake8',

temporary/__init__.py

@@ -0,0 +1,4 @@
+from temporary.directory import temp_dir
+
+# silence PyFlakes
+assert temp_dir

temporary/directory.py

@@ -0,0 +1,18 @@
+from contextlib2 import contextmanager
+from os import chdir, getcwd
+from shutil import rmtree
+from tempfile import mkdtemp
+
+
+@contextmanager
+def temp_dir(suffix='', prefix='tmp', parent_dir=None, make_cwd=False):
+    prev_cwd = getcwd()
+    abs_path = mkdtemp(suffix, prefix, parent_dir)
+    try:
+        if make_cwd:
+            chdir(abs_path)
+        yield abs_path
+    finally:
+        if make_cwd:
+            chdir(prev_cwd)
+        rmtree(abs_path)

temporary/test/test_directory.py

@@ -0,0 +1,33 @@
+from os.path import exists, isdir
+from nose.tools import eq_
+from temporary import temp_dir
+
+
+#
+# Tests
+#
+
+def test_temp_dir():
+    for x in __ensure_usage():
+        yield x
+
+
+#
+# Test Helpers
+#
+
+def __ensure_usage(**kwargs):
+    def ensure_context_manager():
+        with temp_dir(**kwargs) as d:
+            eq_(isdir(d), True)
+        eq_(exists(d), False)
+
+    def ensure_decorator():
+        @temp_dir(**kwargs)
+        def inner():
+            pass
+
+        inner()
+
+    yield ensure_context_manager
+    yield ensure_decorator