From 539d53d2099b26824a5e02a1fa4c6d79187b81e5 Mon Sep 17 00:00:00 2001 From: Lulzette Date: Fri, 17 Sep 2021 15:50:18 +0300 Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=B8=D0=BB=20?= =?UTF-8?q?=D1=87=D1=82=D0=B5=D0=BD=D0=B8=D0=B5=20=D0=BA=D0=BE=D0=BD=D1=84?= =?UTF-8?q?=D0=B8=D0=B3=D0=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config.ini | 4 ++++ main.py | 22 +++++++++++++++++----- 2 files changed, 21 insertions(+), 5 deletions(-) create mode 100644 config.ini diff --git a/config.ini b/config.ini new file mode 100644 index 0000000..b30c71a --- /dev/null +++ b/config.ini @@ -0,0 +1,4 @@ +[DB] +host=localhost +port=27017 +name=pycms diff --git a/main.py b/main.py index 937e6db..4355af7 100755 --- a/main.py +++ b/main.py @@ -15,18 +15,30 @@ class Config: """ Init init """ - mongoclient = pymongo.MongoClient('localhost', 27017) - database = mongoclient['pycms'] + self.readConfig() + + mongoclient = pymongo.MongoClient(self.host, self.port) + database = mongoclient[self.dbname] + + # TODO: Create table if not exists posts = database['posts'] self.posts = posts - def readConfig(): + def readConfig(self): """ Read config file, if not exists - call Init.createConfig() """ - pass + import configparser + config = configparser.ConfigParser() + config.read('config.ini') - def createConfig(): + db = config['DB'] + + self.host = db['host'] + self.port = int(db['port']) + self.dbname = db['name'] + + def createConfig(self): """ Create config file """