Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2024-12-24 11:42:52 +01:00
Code Issues Projects Releases Wiki Activity

Make AtomicFile support context managers.

Browse Source
This commit is contained in:
Valentin Lorentz 2014-07-31 22:52:16 +02:00
parent e42a3dd6ac
commit 5f582e3f52
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/utils/file.py
View File

@ -156,13 +156,19 @@ class AtomicFile(object):
# self.__parent = super(AtomicFile, self) # self.__parent = super(AtomicFile, self)
self._fd = open(self.tempFilename, mode) self._fd = open(self.tempFilename, mode)
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, traceback):
if exc_type:
self.rollback()
else:
self.close()
@property @property
def closed(self): def closed(self):
return self._fd.closed return self._fd.closed
def close(self):
return self._fd.close()
def write(self, data): def write(self, data):
return self._fd.write(data) return self._fd.write(data)