config.py

   1 # Erebus IRC bot - Author: John Runyon
   2 # "Config" class (reading/providing access to bot.config)
   3
   4 from __future__ import print_function
   5 import sys
   6
   7 if sys.version_info.major < 3:
   8         import ConfigParser
   9 else:
  10         import configparser as ConfigParser
  11
  12 class Config(object):
  13         def __init__(self, filename, writeout=True):
  14                 self.__dict__['config'] = ConfigParser.RawConfigParser()
  15                 self.__dict__['filename'] = filename
  16                 self.__dict__['writeout'] = writeout
  17                 self.config.read(filename)
  18
  19         def __getattr__(self, key):
  20                 return self.config.get('erebus', key)
  21
  22         def __setattr__(self, key, value):
  23                 self.config.set('erebus', key, value)
  24
  25         def __getitem__(self, section): #!! READ-ONLY !!
  26                 return {item: self.config.get(section, item) for item in self.config.options(section)}
  27
  28         def level(self, cmd):
  29                 return self.config.get('levels', cmd)
  30
  31         def setlevel(self, cmd, level):
  32                 self.config.set('levels', cmd, level)
  33
  34         def items(self, section='erebus'):
  35                 return self.config.items(section)
  36
  37         def get(self, section, key, default=None):
  38                 try:
  39                         return self.config.get(section, key)
  40                 except:
  41                         return default
  42         def getboolean(self, section, key):
  43                 val = self.get(section, key, False)
  44                 if val == False or val == "0" or val.lower() == "false" or val.strip() == "":
  45                         return False
  46                 else:
  47                         return True
  48
  49         def set(self, section, key, value):
  50                 self.config.set(section, key, value)
  51
  52         def write(self):
  53                 with open(self.filename, 'wb') as configfile:
  54                         self.config.write(configfile)
  55
  56         def __del__(self):
  57                 if self.writeout: self.write()
  58
  59 def setup(fn='bot.config', writeout=True):
  60         return Config(fn, writeout)
  61
  62 if __name__ == '__main__':
  63         import sys
  64         cfg = Config(sys.argv[1], False)
  65
  66         for s in cfg.config.sections():
  67                 for k, v in cfg.items(s):
  68                         print("[%r][%r] = %r" % (s, k, v))