Work in progress

git-svn-id: http://pyfilesystem.googlecode.com/svn/trunk@60 67cdc799-7952-0410-af00-57a81ceafa0f

1 files changed, 22 insertions, 0 deletions

diff --git a/fs/tempfs.py b/fs/tempfs.py
new file mode 100644
index 0000000..3361966
--- /dev/null
+++ b/fs/tempfs.py
@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+
+from osfs import OSFS
+import tempfile
+from shutil import rmtree
+
+class TempFS(OSFS):
+
+    """Create a Filesystem in a tempory directory (with tempfile.mkdtemp),
+    and removes it when the TempFS object is cleaned up."""
+
+    def __init__(self):
+        self._temp_dir = tempfile.mkdtemp("fstest")
+        OSFS.__init__(self, self._temp_dir)
+
+    def _cleanup(self):
+        if self._temp_dir:
+            rmtree(self._temp_dir)
+            self._temp_dir = ""
+
+    def __del__(self):
+        self._cleanup()