REF: Moved around the code a bit to make more sense

FEAT: added consent related code
This commit is contained in:
Dan
2024-08-12 18:38:48 -04:00
parent efc4ed6baf
commit deb1143990
4 changed files with 139 additions and 20 deletions

25
user_data/data_access.py Normal file
View File

@ -0,0 +1,25 @@
import sqlite3
import os
from logger import logger
class DataAccessManager:
def __init__(self, db_path="data/EONA.db"):
self.db_path = db_path
def delete_user_data(self, user_id, guild_id):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("DELETE FROM birthdays WHERE user_id = ? AND guild_id = ?", (user_id, guild_id))
cursor.execute("DELETE FROM consents WHERE user_id = ? AND guild_id = ?", (user_id, guild_id))
conn.commit()
conn.close()
logger.info(f"User {user_id} requested data deletion.")
def view_user_data(self, user_id, guild_id):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("SELECT * FROM birthdays WHERE user_id = ? AND guild_id = ?", (user_id, guild_id))
birthday = cursor.fetchone()
conn.close()
return birthday