make db configurable

1 files changed, 24 insertions, 0 deletions

diff --git a/server/dbconfig.py b/server/dbconfig.py
new file mode 100644
index 0000000..e5eb42c
--- /dev/null
+++ b/server/dbconfig.py
@@ -0,0 +1,24 @@
+
+import sys
+import ConfigParser
+
+class DBConfig(object):
+
+    def __init__(self, configfile):
+        self.config = ConfigParser.ConfigParser()
+        if len(self.config.read(configfile)) == 0:
+            sys.stderr.write('Cannot read ' + configfile)
+            sys.exit(1)
+
+    def get_config(self):
+        ret = dict()
+        try:
+            ret['DB'] = self.config.get('MYSQL', 'DB')
+            ret['USER'] = self.config.get('MYSQL', 'USER')
+            ret['PASS'] = self.config.get('MYSQL', 'PASS')
+
+        except ConfigParser.NoSectionError, ConfigParser.NoOptionError:
+            sys.stderr.write('Invalid db config')
+            sys.exit(1)
+
+        return ret