From 0512e83b455dd6c18be44b62b2b0df4057339226 Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Sun, 22 Nov 2020 12:45:47 -0500 Subject: [PATCH] [+] Load database on start --- src/bot.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/src/bot.py b/src/bot.py index f1de74a..3caca88 100644 --- a/src/bot.py +++ b/src/bot.py @@ -1,3 +1,5 @@ +import json + import telegram from telegram.ext import CommandHandler from telegram.ext import Updater @@ -5,16 +7,23 @@ import logging import os from src.commands import start +from src.database import Database -# Read token from an environment variable -token=os.environ['TG_TOKEN'] +token = os.environ['TG_TOKEN'] +dbPath = 'database.json' +database = Database() # Main if __name__ == '__main__': - # Initialize logger logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO) + # Check database + if os.path.isfile(dbPath): + f = open(dbPath, 'r') + database = json.loads(f.read()) + f.close() + # Create bot bot = telegram.Bot(token=os.environ['TG_TOKEN'])