Mikaela
/
Limnoria
mirror of https://github.com/Mikaela/Limnoria.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Only rollback if we haven't closed.

This commit is contained in:
Jeremy Fincher 2004-07-31 10:58:31 +00:00
parent 823bfb040f
commit ecd67cb7a7
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/utils.py
View File

@ -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: