Draft persistence

1 files changed, 6 insertions, 2 deletions

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):