From 27334b6e89cdec616ea6ac180ec1047a0bd4f54f Mon Sep 17 00:00:00 2001 From: Dmitrii Morozov Date: Wed, 27 Mar 2024 17:52:56 +0100 Subject: Draft persistence --- tgbot.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'tgbot.py') diff --git a/tgbot.py b/tgbot.py index 97dcbd0..6d4fc52 100755 --- a/tgbot.py +++ b/tgbot.py @@ -2,19 +2,23 @@ import os import time, threading, schedule -from telebot import TeleBot +import telebot from fortniteStatusWrapper import formatFortniteStatus +from persistence import UserRepository from pythonFortniteStatus.FortniteStatus import * if "TELEBOT_BOT_TOKEN" not in os.environ: raise AssertionError("Please configure TELEBOT_BOT_TOKEN as environment variables") -bot = TeleBot(os.environ["TELEBOT_BOT_TOKEN"]) +bot = telebot.TeleBot(os.environ["TELEBOT_BOT_TOKEN"]) fortniteStatus = FortniteStatus() +userRepository = UserRepository('db.sqlite') @bot.message_handler(commands = ['start']) def startCommand(message): bot.reply_to(message, "This bot is doing nothing so far..") + print(message.chat.id) + userRepository.putUser(message.chat.id) @bot.message_handler(commands = ['status']) def fortniteStatusCommand(message): -- cgit v1.2.3