diff options
-rw-r--r-- | persistence.py | 21 | ||||
-rwxr-xr-x | tgbot.py | 8 |
2 files changed, 27 insertions, 2 deletions
diff --git a/persistence.py b/persistence.py new file mode 100644 index 0000000..eb1903f --- /dev/null +++ b/persistence.py @@ -0,0 +1,21 @@ +import sqlite3 + +class UserRepository: + + conn = None + + def __init__(self, db_path): + self.__initialize() + + def __initialize(self): + cur = self.__getConnection().cursor() + cur.execute("CREATE TABLE IF NOT EXISTS user(chat_id INT)") + + def __getConnection(self): + return sqlite3.connect('db.sqlite') + + def putUser(self, chat_id): + cur = self.__getConnection().cursor() + data_tuple = (chat_id) + print(type(chat_id)) + cur.execute("INSERT INTO user(chat_id) VALUES(?)", (chat_id, )) @@ -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): |