52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import discord
|
|
from discord import app_commands
|
|
import sqlite3
|
|
|
|
|
|
class DataPrivacy:
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.db_path = 'data/selena.db'
|
|
|
|
async def fetch_user_data(self, user_id):
|
|
conn = sqlite3.connect(self.db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute("SELECT * FROM user_data WHERE user_id = ?", (user_id,))
|
|
data = cursor.fetchall()
|
|
conn.close()
|
|
return data
|
|
|
|
async def delete_user_data(self, user_id):
|
|
conn = sqlite3.connect(self.db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute("DELETE FROM user_data WHERE user_id = ?", (user_id,))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def setup(self, tree: app_commands.CommandTree):
|
|
@tree.command(name="request_data", description="Request your stored data")
|
|
async def request_data_command(interaction: discord.Interaction):
|
|
user_id = interaction.user.id
|
|
data = await self.fetch_user_data(user_id)
|
|
if data:
|
|
await interaction.response.send_message(f"Your data: {data}", ephemeral=True)
|
|
else:
|
|
await interaction.response.send_message("No data found for your user.", ephemeral=True)
|
|
|
|
@tree.command(name="delete_data", description="Request deletion of your stored data")
|
|
async def delete_data_command(interaction: discord.Interaction):
|
|
user_id = interaction.user.id
|
|
await self.delete_user_data(user_id)
|
|
await interaction.response.send_message("Your data has been deleted.", ephemeral=True)
|
|
|
|
if not tree.get_command("request_data"):
|
|
tree.add_command(request_data_command)
|
|
|
|
if not tree.get_command("delete_data"):
|
|
tree.add_command(delete_data_command)
|
|
|
|
|
|
def setup(bot):
|
|
data_privacy = DataPrivacy(bot)
|
|
data_privacy.setup(bot.tree)
|