X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/460fefe4c08241418b04407c5e43542970a0b43b..3cec5bdc665d310ac628e80b007dd2d71ae7d7bd:/config.py diff --git a/config.py b/config.py index b050892..9755f84 100644 --- a/config.py +++ b/config.py @@ -12,7 +12,7 @@ else: class Config(object): def __init__(self, filename, writeout=True): - self.__dict__['config'] = ConfigParser.RawConfigParser() + self.__dict__['config'] = ConfigParser.RawConfigParser(delimiters=('=',)) self.__dict__['filename'] = filename self.__dict__['writeout'] = writeout self.config.read(filename) @@ -47,7 +47,9 @@ class Config(object): return default def getboolean(self, section, key, default=False): val = self.get(section, key, default) - if val == False or val == "0" or val.lower() == "false": + if type(val) is bool: + return val + elif val == "0" or val.lower() == "false": return False else: return True @@ -58,8 +60,13 @@ class Config(object): self.config.set(section, key, str(value)) if self.writeout: self.write() + def delete(self, section, key): + if self.config.has_section(section): + self.config.remove_option(section, key) + if self.writeout: self.write() + def write(self): - with open(self.filename+'.tmp', 'wb') as configfile: + with open(self.filename+'.tmp', 'w') as configfile: self.config.write(configfile) os.rename(configfile.name, self.filename)