From 8b04adef2b2764baf0d5839077bf19de6b93651d Mon Sep 17 00:00:00 2001 From: mitsuha_s Date: Sat, 21 May 2022 10:48:33 +0000 Subject: [PATCH] add add_user method --- database_interaction.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/database_interaction.py b/database_interaction.py index 48aadc9..2586e86 100644 --- a/database_interaction.py +++ b/database_interaction.py @@ -6,7 +6,11 @@ class Database(): self.cur.execute('CREATE TABLE IF NOT EXISTS users (id INTEGER, telegram_id NUMERIC, PRIMARY KEY(id))') self.cur.execute('CREATE TABLE IF NOT EXISTS feeds (id INTEGER, link TEXT, PRIMARY KEY(id))') self.cur.execute('CREATE TABLE IF NOT EXISTS subscribes (user_id INTEGER, feed_id INTEGER, FOREIGN KEY(user_id) REFERENCES users(id), FOREIGN KEY(feed_id) REFERENCES feeds(id))') - self.cur.execute('CREATE TABLE IF NOT EXISTS feeds_last_items (feed_id INTEGER, link TEXT, title TEXT, description TEXT, date NUMERIC, FOREIGN KEY(feed_id) REFERENCES feeds(id))') + self.cur.execute('CREATE TABLE IF NOT EXISTS feeds_last_items (feed_id INTEGER, link TEXT, title TEXT, description TEXT, date NUMERIC, FOREIGN KEY(feed_id) REFERENCES feeds(id))') + + def add_user(self, telegram_id): + self.cur.execute('INSERT INTO users (telegram_id) VALUES (:telegram_id)', {'telegram_id': telegram_id}) + self.conn.commit() def add_rss_feed(): pass @@ -25,4 +29,3 @@ class Database(): def storing_feed_content(): pass -