X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/db75daab8e080fef8e9dbe7796492b24945e8707..179c06a914a8364b4bc571a0da3bed467b4d7f76:/config.py diff --git a/config.py b/config.py index 92b08cb..273450a 100644 --- a/config.py +++ b/config.py @@ -5,7 +5,7 @@ import ConfigParser class Config(object): def __init__(self, filename, writeout=True): - self.__dict__['config'] = ConfigParser.SafeConfigParser() + self.__dict__['config'] = ConfigParser.RawConfigParser() self.__dict__['filename'] = filename self.__dict__['writeout'] = writeout self.config.read(filename) @@ -16,9 +16,24 @@ class Config(object): def __setattr__(self, key, value): self.config.set('erebus', key, value) + def level(self, cmd): + return self.config.get('levels', cmd) + + def setlevel(self, cmd, level): + self.config.set('levels', cmd, level) + def items(self, section='erebus'): return self.config.items(section) + def get(self, section, key, default=None): + try: + return self.config.get(section, key) + except: + return default + + def set(self, section, key, value): + self.config.set(section, key, value) + def write(self): with open(self.filename, 'wb') as configfile: self.config.write(configfile) @@ -31,5 +46,8 @@ if __name__ == '__main__': import sys cfg = Config(sys.argv[1], False) - for k, v in cfg.items(): - print k, '=', v + for s in cfg.config.sections(): + for k, v in cfg.items(s): + print s+'.'+k, '=', v +# for k, v in cfg.items(): +# print 'erebus.'+k, '=', v