60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
# currency.py
|
|
import discord
|
|
from discord import app_commands
|
|
import sqlite3
|
|
import random
|
|
|
|
|
|
class Currency:
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.db_path = 'data/selena.db'
|
|
|
|
async def earn_kibble(self, interaction: discord.Interaction):
|
|
guild_id = str(interaction.guild_id)
|
|
user_id = str(interaction.user.id)
|
|
earned = random.randint(1, 10)
|
|
|
|
conn = sqlite3.connect(self.db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute("""
|
|
INSERT INTO guild_currency (guild_id, user_id, balance)
|
|
VALUES (?, ?, ?)
|
|
ON CONFLICT(guild_id, user_id)
|
|
DO UPDATE SET balance = balance + ?
|
|
""", (guild_id, user_id, earned, earned))
|
|
conn.commit()
|
|
cursor.execute("SELECT balance FROM guild_currency WHERE guild_id = ? AND user_id = ?", (guild_id, user_id))
|
|
balance = cursor.fetchone()[0]
|
|
conn.close()
|
|
|
|
await interaction.response.send_message(f"You earned {earned} Kibble! You now have {balance} Kibble.", ephemeral=False)
|
|
|
|
async def balance(self, interaction: discord.Interaction):
|
|
guild_id = str(interaction.guild_id)
|
|
user_id = str(interaction.user.id)
|
|
|
|
conn = sqlite3.connect(self.db_path)
|
|
cursor = conn.cursor()
|
|
cursor.execute("SELECT balance FROM guild_currency WHERE guild_id = ? AND user_id = ?", (guild_id, user_id))
|
|
result = cursor.fetchone()
|
|
conn.close()
|
|
|
|
balance = result[0] if result else 0
|
|
await interaction.response.send_message(f"You have {balance} Kibble.", ephemeral=False)
|
|
|
|
def setup(self, tree: app_commands.CommandTree):
|
|
@tree.command(name="earn_kibble", description="Earn Kibble currency")
|
|
async def earn_kibble_command(interaction: discord.Interaction):
|
|
await self.earn_kibble(interaction)
|
|
|
|
@tree.command(name="balance", description="Check your Kibble balance")
|
|
async def balance_command(interaction: discord.Interaction):
|
|
await self.balance(interaction)
|
|
|
|
if not tree.get_command("earn_kibble"):
|
|
tree.add_command(earn_kibble_command)
|
|
|
|
if not tree.get_command("balance"):
|
|
tree.add_command(balance_command)
|