diff options
| -rw-r--r-- | libbe/utility.py | 7 |
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 |