Remove user if was kicked from chat

1 files changed, 13 insertions, 5 deletions

diff --git a/persistence.py b/persistence.py
index d2998e9..941e2f7 100644
--- a/persistence.py
+++ b/persistence.py
@@ -9,7 +9,7 @@ class UserRepository:
     
     def __initialize(self):
         cur = self.__getConnection().cursor()
-        cur.execute("CREATE TABLE IF NOT EXISTS user(chat_id INT)")
+        cur.execute("CREATE TABLE IF NOT EXISTS user(chat_id INT, alias TEXT)")
         cur.execute("CREATE UNIQUE INDEX IF NOT EXISTS chat_id_idx ON user(chat_id)")
     
     def __getConnection(self):
@@ -28,14 +28,22 @@ class UserRepository:
         query = "select * from user"
         cur.execute(query)
         return cur.fetchall()
+    
+    def removeChat(self, chat_id):
+        connection = self.__getConnection()
+        cur = connection.cursor()
+        query = "DELETE FROM user where chat_id = {chat_id}".format(
+            chat_id = chat_id)
+        cur.execute(query)
+        connection.commit()
 
     
-    def putUser(self, chat_id):
+    def putChat(self, chat_id, alias):
         if not self.getUser(chat_id):
             connection = self.__getConnection()
             cur = connection.cursor()
-            query = "INSERT INTO user(chat_id) VALUES({chat_id})".format(chat_id = chat_id)
+            query = "INSERT INTO user(chat_id, alias) VALUES({chat_id}, '{text}')".format(
+                chat_id = chat_id,
+                text = alias)
             cur.execute(query)
             connection.commit()
-        else:
-            print("User {} already exsits, skipping putUser()".format(chat_id))