From 11ed17f2ea64efe2e97017da76c3060d058b0b82 Mon Sep 17 00:00:00 2001 From: Anorak_1 Date: Tue, 8 Aug 2023 17:26:45 +0200 Subject: [PATCH] Created dbutils.py for easy database manipulation --- dbutil.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 dbutil.py diff --git a/dbutil.py b/dbutil.py new file mode 100644 index 0000000..71f17f0 --- /dev/null +++ b/dbutil.py @@ -0,0 +1,23 @@ +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 + +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() \ No newline at end of file