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

FEAT: added consent related code

main.py

@@ -27,7 +27,7 @@ class EONA(discord.Client):
             logger.error("config.py not found")
         try:
             if config["modules"]["birthday"]["enabled"]:
-                from birthday import Birthday
+                from user.birthday import Birthday
                 birthday = Birthday(self)
                 birthday.setup(self.tree)
                 logger.info("Birthday module loaded")

user/birthday.py

@@ -2,7 +2,10 @@ import discord
 from discord.ext import tasks
 import sqlite3
 from logger import logger
-import datetime
+from datetime import datetime
+import os
+from user_data.consent import ConsentManager
+from user_data.data_access import DataAccessManager
 
 
 class Birthday:
@@ -10,24 +13,35 @@ class Birthday:
         self.client = client
         self.db_path = "data/EONA.db"
         self.logger = logger
+        self.consent_manager = ConsentManager(self.db_path)
+        self.data_access_manager = DataAccessManager(self.db_path)
 
         self.ensure_db()
 
     def ensure_db(self):
-        conn = sqlite3.connect(self.db_path)
+        os.makedirs(os.path.dirname(self.db_path), exist_ok=True)
-        cursor = conn.cursor()
+        if not os.path.exists(self.db_path):
-        cursor.execute("""CREATE TABLE IF NOT EXISTS birthdays (
+            self.logger.info("Database does not exist. Creating a new database.")
-            user_id INTEGER,
+            conn = sqlite3.connect(self.db_path)
-            guild_id INTEGER,
+            cursor = conn.cursor()
-            user_name TEXT,
+            cursor.execute("""CREATE TABLE IF NOT EXISTS birthdays (
-            birthday TEXT,
+                user_id INTEGER,
-            PRIMARY KEY (user_id, guild_id)
+                guild_id INTEGER,
-            );""")
+                user_name TEXT,
-        conn.commit()
+                birthday TEXT,
-        conn.close()
+                PRIMARY KEY (user_id, guild_id)
-        self.logger.info("Birthday table ensured")
+                );""")
+            conn.commit()
+            conn.close()
+            self.logger.info("Database and table created successfully.")
+        else:
+            self.logger.info("Database already exists.")
+
+    async def set_birthday(self, interaction: discord.Interaction, user_id, guild_id, birthday):
+        if not self.consent_manager.has_consent(user_id, guild_id):
+            await self.consent_manager.request_consent(interaction)
+            return  # Exit early since consent hasn't been given yet
 
-    async def set_birthday(self, user_id, guild_id, birthday):
         conn = sqlite3.connect(self.db_path)
         cursor = conn.cursor()
         cursor.execute("""INSERT INTO birthdays (user_id, guild_id, birthday)
@@ -37,7 +51,10 @@ class Birthday:
         conn.commit()
         conn.close()
         self.logger.info(f"{user_id}! Your birthday was set to {birthday}")
+        await interaction.response.send_message(f"Your birthday has been set to {birthday}.", ephemeral=True)
 
+    async def delete_birthday(self, user_id, guild_id):
+        self.data_access_manager.delete_data(user_id, guild_id)
 
     async def get_birthday(self, user_id, guild_id):
         conn = sqlite3.connect(self.db_path)
@@ -69,6 +86,24 @@ class Birthday:
         self.logger.info("Birthday module loaded")
 
     def setup(self, tree: discord.app_commands.CommandTree):
+        @tree.command(name="consent", description="Give consent for data storage.")
+        async def consent_command(interaction: discord.Interaction):
+            self.consent_manager.give_consent(interaction.user.id, interaction.guild.id)
+            await interaction.response.send_message(f"{interaction.user.mention}, your consent has been recorded.", ephemeral=True)
+
+        @tree.command(name="view_data", description="View your stored data.")
+        async def view_data_command(interaction: discord.Interaction):
+            data = self.data_access_manager.view_user_data(interaction.user.id, interaction.guild.id)
+            if data:
+                await interaction.response.send_message(f"{interaction.user.mention}, your stored data: {data}", ephemeral=True)
+            else:
+                await interaction.response.send_message(f"{interaction.user.mention}, no data found.", ephemeral=True)
+
+        @tree.command(name="delete_data", description="Delete your stored data.")
+        async def delete_data_command(interaction: discord.Interaction):
+            self.data_access_manager.delete_user_data(interaction.user.id, interaction.guild.id)
+            await interaction.response.send_message(f"{interaction.user.mention}, your data has been deleted.", ephemeral=True)
+
         @tree.command(name="set_birthday", description="Set your birthday (YYYY-MM-DD)")
         async def set_birthday_command(interaction: discord.Interaction):
             await interaction.response.send_modal(BirthdayModal(self))
@@ -83,6 +118,12 @@ class Birthday:
             else:
                 await interaction.response.send_message("Your birthday has not been set. Please use `/set_birthday` to set your birthday.", ephemeral=True)
 
+        if not tree.get_command("consent"):
+            tree.add_command(consent_command)
+        if not tree.get_command("view_data"):
+            tree.add_command(view_data_command)
+        if not tree.get_command("delete_data"):
+            tree.add_command(delete_data_command)
         if not tree.get_command("set_birthday"):
             tree.add_command(set_birthday_command)
         if not tree.get_command("get_birthday"):
@@ -100,11 +141,14 @@ class BirthdayModal(discord.ui.Modal, title="Set your birthday"):
         user_id = str(interaction.user.id)
         guild_id = str(interaction.guild.id)
         try:
-            birthday = self.birthday.value
+            # Validate the date format
-            await self.birthday_module.set_birthday(user_id, guild_id, birthday)
+            birthday_str = self.birthday.value
-            await interaction.response.send_message(f"Your birthday has been set to {birthday}", ephemeral=True)
+            birthday = datetime.strptime(birthday_str, "%Y-%m-%d").date()
-        except ValueError as e:
+
-            await interaction.response.send_message(str(e), ephemeral=True)
+            # Proceed to set the birthday if valid
+            await self.birthday_module.set_birthday(interaction, user_id, guild_id, birthday_str)
+        except ValueError:
+            await interaction.response.send_message("Invalid date format. Please enter the date in YYYY-MM-DD format.", ephemeral=True)
 
 
 def setup(client):

user_data/consent.py

@@ -0,0 +1,50 @@
+import sqlite3
+import os
+from logger import logger
+
+
+class ConsentManager:
+    def __init__(self, db_path="data/EONA.db"):
+        self.db_path = db_path
+        self.ensure_db()
+
+    def ensure_db(self):
+        os.makedirs(os.path.dirname(self.db_path), exist_ok=True)
+
+        conn = sqlite3.connect(self.db_path)
+        cursor = conn.cursor()
+        cursor.execute("""CREATE TABLE IF NOT EXISTS consents (
+            user_id INTEGER,
+            guild_id INTEGER,
+            consent INTEGER,
+            PRIMARY KEY (user_id, guild_id)
+            );""")
+        conn.commit()
+        conn.close()
+        logger.info("Consent table ensured")
+
+    async def request_consent(self, interaction):
+        await interaction.response.send_message(
+            "To use this feature, please consent to our data policy. "
+            "Type `/consent` to agree, or `/decline` to disagree.",
+            ephemeral=True
+        )
+
+    def give_consent(self, user_id, guild_id):
+        conn = sqlite3.connect(self.db_path)
+        cursor = conn.cursor()
+        cursor.execute("""INSERT INTO consents (user_id, guild_id, consent)
+                       VALUES (?, ?, 1) ON CONFLICT(user_id, guild_id)
+                       DO UPDATE SET consent = 1;""",
+                       (user_id, guild_id))
+        conn.commit()
+        conn.close()
+        logger.info(f"User {user_id} consented to data storage.")
+
+    def has_consent(self, user_id, guild_id):
+        conn = sqlite3.connect(self.db_path)
+        cursor = conn.cursor()
+        cursor.execute("SELECT consent FROM consents WHERE user_id = ? AND guild_id = ?", (user_id, guild_id))
+        row = cursor.fetchone()
+        conn.close()
+        return row and row[0] == 1  # Returns True if consent is given

user_data/data_access.py

@@ -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