Fixed header

[erebus.git] / config.py

diff --git a/config.py b/config.py
index af470cb4a63f9f85e8fbf9ffb543d48f8120f8f8..c8a461bd2572843901ab049b3cafe799745b1807 100644 (file)
--- a/config.py
+++ b/config.py
@@ -1,7 +1,9 @@
+# Erebus IRC bot - Author: John Runyon
+# "Config" class (reading/providing access to bot.config)
+
 import ConfigParser
 
 class Config(object):
-#      config = ConfigParser.SafeConfigParser()
        def __init__(self, filename, writeout=True):
                self.__dict__['config'] = ConfigParser.SafeConfigParser()
                self.__dict__['filename'] = filename
@@ -14,8 +16,14 @@ class Config(object):
        def __setattr__(self, key, value):
                self.config.set('erebus', key, value)
 
-       def items(self):
-               return self.config.items('erebus')
+       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 write(self):
                with open(self.filename, 'wb') as configfile: