[+] Load database on start

This commit is contained in:
Hykilpikonna
2020-11-22 12:45:47 -05:00
parent 5c89b401d4
commit 0512e83b45
+12 -3
View File
@@ -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'])