Draft persistence

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