[+] Load database on start
This commit is contained in:
+12
-3
@@ -1,3 +1,5 @@
|
|||||||
|
import json
|
||||||
|
|
||||||
import telegram
|
import telegram
|
||||||
from telegram.ext import CommandHandler
|
from telegram.ext import CommandHandler
|
||||||
from telegram.ext import Updater
|
from telegram.ext import Updater
|
||||||
@@ -5,16 +7,23 @@ import logging
|
|||||||
import os
|
import os
|
||||||
|
|
||||||
from src.commands import start
|
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
|
# Main
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|
||||||
# Initialize logger
|
# Initialize logger
|
||||||
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
|
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
|
# Create bot
|
||||||
bot = telegram.Bot(token=os.environ['TG_TOKEN'])
|
bot = telegram.Bot(token=os.environ['TG_TOKEN'])
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user