Moved app message functions under MessageDB class

for hinting
main
Anorak_1 2023-08-08 17:33:37 +02:00
parent 11ed17f2ea
commit 838600c369
1 changed files with 19 additions and 17 deletions

View File

@ -2,22 +2,24 @@ import json
import os import os
import sqlite3 import sqlite3
def add_application_msg(msg_id: str, author_id: str, guild_id: str) -> None:
data = (msg_id, author_id, guild_id)
con = sqlite3.connect("applications.db")
cur = con.cursor()
cur.execute("INSERT INTO app_msg_db VALUES (?, ?, ?)", data)
con.commit()
def get_application_msg(msg_id: str): class MessageDB():
con = sqlite3.connect("applications.db") def add_application_msg(msg_id: str, author_id: str, guild_id: str) -> None:
cur = con.cursor() data = (msg_id, author_id, guild_id)
cur.execute(f"SELECT user_id, guild_id FROM app_msg_db WHERE msg_id={msg_id}") con = sqlite3.connect("applications.db")
user_id, guild_id = cur.fetchone() cur = con.cursor()
return user_id, guild_id cur.execute("INSERT INTO app_msg_db VALUES (?, ?, ?)", data)
con.commit()
def remove_application_msg(msg_id: str) -> None: def get_application_msg(msg_id: str) -> tuple[str, str]:
con = sqlite3.connect("applications.db") con = sqlite3.connect("applications.db")
cur = con.cursor() cur = con.cursor()
cur.execute(f"DELETE FROM app_msg_db WHERE msg_id={msg_id}") cur.execute(f"SELECT user_id, guild_id FROM app_msg_db WHERE msg_id={msg_id}")
con.commit() user_id, guild_id = cur.fetchone()
return user_id, guild_id
def remove_application_msg(msg_id: str) -> None:
con = sqlite3.connect("applications.db")
cur = con.cursor()
cur.execute(f"DELETE FROM app_msg_db WHERE msg_id={msg_id}")
con.commit()