from config import Config def save_message(chat_id: int, message_id: int): if Config.DISABLE_STORAGE: return if True: return from .DB import get_db 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(): if Config.DISABLE_STORAGE: return [] if True: return [] from .DB import get_db db = get_db() cur = db.cursor() cur.execute("SELECT * FROM message") rows = cur.fetchall() cur.close() db.close() return rows def clear_messages(): if Config.DISABLE_STORAGE: return if True: return from .DB import get_db db = get_db() cur = db.cursor() cur.execute("DELETE FROM message") db.commit() cur.close() db.close()