86 lines
3.5 KiB
Python
86 lines
3.5 KiB
Python
|
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)
|