[M] database: Separate from main function

This commit is contained in:
Hykilpikonna
2020-11-22 14:13:50 -05:00
parent 54455ba0dc
commit 74f39bb621
2 changed files with 4 additions and 11 deletions
-9
View File
@@ -8,22 +8,13 @@ import os
from src.commands import *
from src.constants import dbPath, token
from src.database import Database
from src.utils import createCommand
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=token)
+4 -2
View File
@@ -1,6 +1,6 @@
import re
from src.bot import database
from src.database import Database
from src.utils import toJson
from src.request_configuration import RequestConfiguration
@@ -32,6 +32,8 @@ urlValidator = re.compile(
r'(?::\d+)?' # optional port
r'(?:/?|[/?]\S+)$', re.IGNORECASE)
database = Database()
def start(update, context):
chat = update.effective_chat
@@ -73,7 +75,7 @@ def touch(update, context):
database.userRequests[user][name] = RequestConfiguration(url)
database.save()
return "*%s* is successfully created!" % name
return "%s is successfully created!" % name
def rm(update, context):