Files
myfirstprogram/storage/message_storage.py
T

31 lines
575 B
Python

from .DB import get_db
def save_message(chat_id: int, message_id: int):
db = get_db()
cur = db.cursor()
cur.execute("INSERT INTO message VALUES (?, ?)", (int(chat_id), int(message_id)))
db.commit()
cur.close()
db.close()
def load_messages():
db = get_db()
cur = db.cursor()
cur.execute("SELECT * FROM message")
rows = cur.fetchall()
cur.close()
db.close()
return rows
def clear_messages():
db = get_db()
cur = db.cursor()
cur.execute("DELETE FROM message")
db.commit()
cur.close()
db.close()