X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/d1ea05b06625b91b6d4fc9580fff461b47b3570c..0956623592806f24abdcdf0c937e1dbdc31a38ac:/config.py

diff --git a/config.py b/config.py
index d274509..5f35061 100644
--- a/config.py
+++ b/config.py
@@ -1,3 +1,6 @@
+# Erebus IRC bot - Author: John Runyon
+# "Config" class (reading/providing access to bot.config)
+
 import ConfigParser
 
 class Config(object):
@@ -13,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)