Fixed "'NoneType' object is not callable" error in libbe/utility/Dir.__del__

1 files changed, 5 insertions, 2 deletions

diff --git a/libbe/utility.py b/libbe/utility.py
index 3f69a95..7c1d10a 100644
--- a/libbe/utility.py
+++ b/libbe/utility.py
@@ -101,11 +101,14 @@ class Dir (object):
     "A temporary directory for testing use"
     def __init__(self):
         self.path = tempfile.mkdtemp(prefix="BEtest")
-        self.shutil = shutil # save local reference for __del__
+        self.rmtree = shutil.rmtree # save local reference for __del__
+        self.removed = False
     def __del__(self):
         self.cleanup()
     def cleanup(self):
-        self.shutil.rmtree(self.path)
+        if self.removed == False:
+            self.rmtree(self.path)
+            self.removed = True
     def __call__(self):
         return self.path