FEAT: Added a currency system (still needs named)

main.py

@@ -22,6 +22,7 @@ class Selena(discord.Client):
             "modules.admin.logger_module",
             "modules.media.spotify_module",
             "modules.user.birthday_module",
+            "modules.money.currency_module",
             # Add other modules here
         ]
         for extension in extensions:

modules/money/currency_db.py

@@ -0,0 +1,18 @@
+import sqlite3
+
+
+def initialize_db():
+    conn = sqlite3.connect('currency.db')
+    cursor = conn.cursor()
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS currency (
+            user_id TEXT PRIMARY KEY,
+            balance INTEGER
+        )
+    ''')
+    conn.commit()
+    conn.close()
+
+
+def get_connection():
+    return sqlite3.connect('currency.db')

modules/money/currency_module.py

@@ -0,0 +1,85 @@
+import discord
+from discord import app_commands
+import sqlite3
+import logging
+import random
+from .currency_db import initialize_db, get_connection
+
+logger = logging.getLogger(__name__)
+initialize_db()
+
+
+class CurrencyModule:
+    def __init__(self, bot):
+        self.bot = bot
+        self.add_commands()
+
+    def add_commands(self):
+        @app_commands.command(
+            name="earn_currency", description="Earn currency"
+        )
+        async def earn_currency(interaction: discord.Interaction):
+            await interaction.response.defer()
+            try:
+                amount = random.choices([random.randint(1, 10), 100], [0.99, 0.01])[0]
+                conn = get_connection()
+                cursor = conn.cursor()
+                cursor.execute('INSERT OR IGNORE INTO currency (user_id, balance) VALUES (?, ?)',
+                               (str(interaction.user.id), 0))
+                cursor.execute('UPDATE currency SET balance = balance + ? WHERE user_id = ?',
+                               (amount, str(interaction.user.id)))
+                conn.commit()
+                cursor.execute('SELECT balance FROM currency WHERE user_id = ?', (str(interaction.user.id),))
+                new_balance = cursor.fetchone()[0]
+                conn.close()
+                await interaction.followup.send(
+                    embed=discord.Embed(
+                        title="Earn Currency",
+                        description=f"You have earned {amount} currency. Your new balance is {new_balance}.",
+                        color=discord.Color.green()
+                    )
+                )
+                logger.info(f"User {interaction.user.id} earned {amount} currency. New balance: {new_balance}")
+            except Exception as e:
+                await interaction.followup.send(f"An error occurred: {e}")
+                logger.error(f"Error in earn_currency command: {e}")
+
+        @app_commands.command(
+            name="check_balance", description="Check your currency balance"
+        )
+        async def check_balance(interaction: discord.Interaction):
+            await interaction.response.defer()
+            try:
+                conn = get_connection()
+                cursor = conn.cursor()
+                cursor.execute('SELECT balance FROM currency WHERE user_id = ?', (str(interaction.user.id),))
+                result = cursor.fetchone()
+                conn.close()
+                if result:
+                    await interaction.followup.send(
+                        embed=discord.Embed(
+                            title="Check Balance",
+                            description=f"Your current balance is {result[0]} currency.",
+                            color=discord.Color.green()
+                        )
+                    )
+                    logger.info(f"User {interaction.user.id} checked balance: {result[0]}")
+                else:
+                    await interaction.followup.send(
+                        embed=discord.Embed(
+                            title="Check Balance",
+                            description="You have no currency.",
+                            color=discord.Color.red()
+                        )
+                    )
+                    logger.info(f"User {interaction.user.id} has no currency.")
+            except Exception as e:
+                await interaction.followup.send(f"An error occurred: {e}")
+                logger.error(f"Error in check_balance command: {e}")
+
+        self.bot.tree.add_command(earn_currency)
+        self.bot.tree.add_command(check_balance)
+
+
+async def setup(bot):
+    CurrencyModule(bot)