diff --git a/config.ini b/config.ini index b30c71a..ad468e2 100644 --- a/config.ini +++ b/config.ini @@ -1,4 +1,5 @@ [DB] -host=localhost -port=27017 -name=pycms +host = localhost +port = 27017 +name = pycms + diff --git a/main.py b/main.py index 4355af7..885309b 100755 --- a/main.py +++ b/main.py @@ -30,7 +30,8 @@ class Config: """ import configparser config = configparser.ConfigParser() - config.read('config.ini') + if not config.read('config.ini'): + self.createConfig(config) db = config['DB'] @@ -38,11 +39,18 @@ class Config: self.port = int(db['port']) self.dbname = db['name'] - def createConfig(self): + def createConfig(self, config): """ Create config file """ - pass + config['DB'] = {} + db = config['DB'] + db['host'] = 'localhost' + db['port'] = '27017' + db['name'] = 'pycms' + + with open('config.ini', 'w') as cfgfile: + config.write(cfgfile) class Back():