nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

more portable solution of saving in a temp file

Browse source
This commit is contained in:
Romain Bignon 2010-07-01 11:55:51 +02:00
commit 46d00b4c79
1 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
weboob/tools/config/yamlconfig.py
View file

@ -18,7 +18,6 @@
from __future__ import with_statement
import sys
import os
import tempfile
import logging
@ -50,14 +49,11 @@ class YamlConfig(IConfig):
self.values = {}
def save(self):
if sys.version_info[:2] <= (2, 5):
with open(self.path, 'w') as f:
yaml.dump(self.values, f)
else:
# write in a temporary file to avoid corruption problems
with tempfile.NamedTemporaryFile(delete=False) as f:
yaml.dump(self.values, f)
os.rename(f.name, self.path)
# write in a temporary file to avoid corruption problems
fd, path = tempfile.mkstemp(dir=os.path.dirname(self.path))
with os.fdopen(fd, 'w') as f:
yaml.dump(self.values, f)
os.rename(path, self.path)
def get(self, *args, **kwargs):
default = None