parent
11ed17f2ea
commit
838600c369
36
dbutil.py
36
dbutil.py
|
@ -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()
|
Loading…
Reference in New Issue