new module for context managers, keeping py <2.4 syntax compat for distribution

1 files changed, 20 insertions, 0 deletions

diff --git a/contexts.py b/contexts.py
new file mode 100644
index 0000000..968c9be
--- /dev/null
+++ b/contexts.py
@@ -0,0 +1,20 @@
+try:
+    from contextlib import contextmanager
+except ImportError:
+    # py < 2.5
+    pass
+else:
+    
+    import tempfile
+    import shutil
+
+    def tempdir():
+        try:
+            path = tempfile.mkdtemp()
+            yield path
+        finally:
+            shutil.rmtree(path)
+
+    # keep py < 2.4 syntax compat to avoid distribution pb
+    tempdir = contextmanager(tempdir)
+