H1K0-dev/skazanull
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(bot): use message timestamp instead of current time when adding quote

Browse Source
This commit is contained in:
Masahiko AMANO 2025-06-25 02:00:19 +03:00
parent dc2bae0da6
commit b045b53519
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
bot/src/db.py
View File

@ -53,8 +53,9 @@ class Database:
quote = msg.text.strip().split("\n\n") quote = msg.text.strip().split("\n\n")
text = quote[0].strip() text = quote[0].strip()
author = "" if len(quote) == 1 else quote[1].strip() author = "" if len(quote) == 1 else quote[1].strip()
timestamp = msg.forward_origin.date if msg.forward_origin else msg.date
with self.conn.cursor() as cursor: with self.conn.cursor() as cursor:
cursor.execute("INSERT INTO quotes(text, author, creator_id) VALUES(%s, NULLIF(%s, ''), (SELECT id FROM users WHERE telegram_id=%s)) RETURNING id", (quote[0].strip(), author, msg.from_user.id)) cursor.execute("INSERT INTO quotes(text, author, creator_id, datetime) VALUES(%s, NULLIF(%s, ''), (SELECT id FROM users WHERE telegram_id=%s), to_timestamp(%s)::timestamptz) RETURNING id", (quote[0].strip(), author, msg.from_user.id, timestamp))
return cursor.fetchone()[0] return cursor.fetchone()[0]
def quotes_count(self): def quotes_count(self):