parent
11ed17f2ea
commit
838600c369
36
dbutil.py
36
dbutil.py
|
@ -2,22 +2,24 @@ import json
|
|||
import os
|
||||
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):
|
||||
con = sqlite3.connect("applications.db")
|
||||
cur = con.cursor()
|
||||
cur.execute(f"SELECT user_id, guild_id FROM app_msg_db WHERE msg_id={msg_id}")
|
||||
user_id, guild_id = cur.fetchone()
|
||||
return user_id, guild_id
|
||||
class MessageDB():
|
||||
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 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()
|
||||
def get_application_msg(msg_id: str) -> tuple[str, str]:
|
||||
con = sqlite3.connect("applications.db")
|
||||
cur = con.cursor()
|
||||
cur.execute(f"SELECT user_id, guild_id FROM app_msg_db WHERE msg_id={msg_id}")
|
||||
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()
|
Loading…
Reference in New Issue