Add history feature

bot/bot.py

@@ -9,6 +9,7 @@
 
 # Queries
 from database.queries import user_exists, insert_user, deactivate_user
+from database.queries import insert_text
 
 # Variables
 api_token = os.environ.get('API_TOKEN')
@@ -148,12 +149,7 @@ async def left_chat_member_event(chat, member):
 @bot.group_message
 async def group_message_event(chat, message):
     sender = message.get('from')
-    sender_id = sender.get('id')  # noqa
-    sender_name = sender.get('first_name')
-    chat = message.get('chat')
-    group_id = chat.get('id')
     message_date = message.get('date')
-    message_text = message.get('text')
-    logger.info('Got message from group %s at %s', group_id, message_date)
-    logger.info('Sender %s', sender_name)
-    logger.info('Text -> %s', message_text)
+    text = message.get('text')
+    logger.info('Got message from group at %s', message_date)
+    await insert_text(chat.bot.pg_pool, sender, text)

database/queries.py

@@ -51,3 +51,18 @@ async def deactivate_user(pool, user):
 
     finally:
         await pool.release(conn)
+
+
+async def insert_text(pool, sender, text):
+    query = '''
+    insert into history(sender, text)
+    values ($1, $2)
+    '''
+
+    conn = await pool.acquire()
+
+    try:
+        await conn.execute(query, sender, text)
+
+    finally:
+        await pool.release(conn)

database/schema.sql

@@ -15,7 +15,7 @@ create table if not exists users (
 
 
 create table if not exists history (
-  id bigint primary key,
+  id bigserial primary key,
   sender jsonb,
   timestamp timestamptz default timezone('Asia/Tashkent'::text, now()),
   text citext