Peter Surda
8cb0953401
- shared.writeKeysFile -> BMConfigParser().save - makes it easier to get rid of circular imports
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import ConfigParser
|
|
import datetime
|
|
import shutil
|
|
import os
|
|
|
|
from singleton import Singleton
|
|
import state
|
|
|
|
|
|
@Singleton
|
|
class BMConfigParser(ConfigParser.SafeConfigParser):
|
|
def set(self, section, option, value=None):
|
|
if self._optcre is self.OPTCRE or value:
|
|
if not isinstance(value, basestring):
|
|
raise TypeError("option values must be strings")
|
|
return ConfigParser.ConfigParser.set(self, section, option, value)
|
|
|
|
def get(self, section, option, raw=False, vars=None):
|
|
if section == "bitmessagesettings" and option == "timeformat":
|
|
try:
|
|
return ConfigParser.ConfigParser.get(self, section, option, raw, vars)
|
|
except ConfigParser.InterpolationError:
|
|
return ConfigParser.ConfigParser.get(self, section, option, True, vars)
|
|
return ConfigParser.ConfigParser.get(self, section, option, True, vars)
|
|
|
|
def safeGetBoolean(self, section, field):
|
|
if self.has_option(section, field):
|
|
try:
|
|
return self.getboolean(section, field)
|
|
except ValueError:
|
|
return False
|
|
return False
|
|
|
|
def safeGetInt(self, section, field):
|
|
if self.has_option(section, field):
|
|
try:
|
|
return self.getint(section, field)
|
|
except ValueError:
|
|
return 0
|
|
return 0
|
|
|
|
def safeGet(self, section, option, default = None):
|
|
if self.has_option(section, option):
|
|
return self.get(section, option)
|
|
else:
|
|
return default
|
|
|
|
def items(self, section, raw=False, vars=None):
|
|
return ConfigParser.ConfigParser.items(self, section, True, vars)
|
|
|
|
def save(self):
|
|
fileName = os.path.join(state.appdata, 'keys.dat')
|
|
fileNameBak = fileName + "." + datetime.datetime.now().strftime("%Y%j%H%M%S%f") + '.bak'
|
|
# create a backup copy to prevent the accidental loss due to the disk write failure
|
|
try:
|
|
shutil.copyfile(fileName, fileNameBak)
|
|
# The backup succeeded.
|
|
fileNameExisted = True
|
|
except:
|
|
# The backup failed. This can happen if the file didn't exist before.
|
|
fileNameExisted = False
|
|
# write the file
|
|
with open(fileName, 'wb') as configfile:
|
|
self.write(configfile)
|
|
# delete the backup
|
|
if fileNameExisted:
|
|
os.remove(fileNameBak)
|