diff options
Diffstat (limited to 'persistence.py')
-rw-r--r-- | persistence.py | 18 |
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)) |