diff options
Diffstat (limited to 'server/dbconfig.py')
-rw-r--r-- | server/dbconfig.py | 24 |
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 |