From ecd67cb7a7b5df267ece958fef4e0f2d323e694b Mon Sep 17 00:00:00 2001
From: Jeremy Fincher <jemfinch@users.sourceforge.net>
Date: Sat, 31 Jul 2004 10:58:31 +0000
Subject: [PATCH] Only rollback if we haven't closed.

---
 src/utils.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/utils.py b/src/utils.py
index 3ff4cdd10..aeb031a95 100755
--- a/src/utils.py
+++ b/src/utils.py
@@ -681,6 +681,7 @@ class AtomicFile(file):
     Opens the file in 'w' mode."""
     def __init__(self, filename, allowEmptyOverwrite=False):
         self.filename = filename
+        self.closed = False
         self.rolledback = False
         self.allowEmptyOverwrite = allowEmptyOverwrite
         self.tempFilename = '%s.%s' % (filename, mktemp())
@@ -688,16 +689,18 @@ class AtomicFile(file):
 
     def rollback(self):
         #print 'AtomicFile.rollback'
-        super(AtomicFile, self).close()
-        if os.path.exists(self.tempFilename):
-            print 'AtomicFile: Removing %s.' % self.tempFilename
-            os.remove(self.tempFilename)
-        self.rolledback = True
+        if not self.closed:
+            super(AtomicFile, self).close()
+            if os.path.exists(self.tempFilename):
+                print 'AtomicFile: Removing %s.' % self.tempFilename
+                os.remove(self.tempFilename)
+            self.rolledback = True
 
     def close(self):
         #print 'AtomicFile.close'
         if not self.rolledback:
             #print 'AtomicFile.close: actually closing.'
+            self.closed = True
             super(AtomicFile, self).close()
             size = os.stat(self.tempFilename).st_size
             if size or self.allowEmptyOverwrite: