X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/a12f751981461907aad0ca4491b07a3866a8a938..6b2c681d29a7f4d3157498b491bdb8371dad4c39:/config.py

diff --git a/config.py b/config.py
index af470cb..c8a461b 100644
--- 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: