mystiatech/selena
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FEAT: Added XP

Browse Source 
REF: Fixed up currency so the cooldown is presented to the user.
This commit is contained in:
Dan
2024-06-24 23:20:03 -04:00
parent fcc1489b74
commit 49a7588699
4 changed files with 158 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
modules/user/currency.py
View File

@@ -2,7 +2,7 @@ import discord
from discord import app_commands
import random
import sqlite3
import os
from datetime import datetime, timedelta
class Currency:
@@ -10,6 +10,7 @@ class Currency:
self.bot = bot
self.db_path = 'data/selena.db'
self._init_db()
self.cooldowns = {}
def _init_db(self):
with sqlite3.connect(self.db_path) as conn:
@@ -17,7 +18,9 @@ class Currency:
cursor.execute('''
CREATE TABLE IF NOT EXISTS users (
user_id TEXT PRIMARY KEY,
kibble INTEGER DEFAULT 0
kibble INTEGER DEFAULT 0,
xp INTEGER DEFAULT 0,
level INTEGER DEFAULT 1
)
''')
conn.commit()
@@ -25,38 +28,51 @@ class Currency:
def _get_user_kibble(self, user_id):
with sqlite3.connect(self.db_path) as conn:
cursor = conn.cursor()
cursor.execute('SELECT kibble FROM users WHERE user_id = ?',
(user_id,))
cursor.execute('SELECT kibble FROM users WHERE user_id = ?', (user_id,))
row = cursor.fetchone()
return row[0] if row else 0
def _update_user_kibble(self, user_id, amount):
with sqlite3.connect(self.db_path) as conn:
cursor = conn.cursor()
cursor.execute('INSERT OR IGNORE INTO users (user_id, kibble) VALUES (?, ?)',
(user_id, 0))
cursor.execute('UPDATE users SET kibble = kibble + ? WHERE user_id = ?',
(amount, user_id))
cursor.execute('INSERT OR IGNORE INTO users (user_id, kibble) VALUES (?, ?)', (user_id, 0))
cursor.execute('UPDATE users SET kibble = kibble + ? WHERE user_id = ?', (amount, user_id))
conn.commit()
def setup(self, tree):
@tree.command(name='earn_kibble', description='Earn Kibble')
@app_commands.checks.cooldown(1, 30)
async def earn_kibble(interaction: discord.Interaction):
user_id = str(interaction.user.id)
amount = random.choices([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 100],
[0.01, 10, 10, 10, 10, 10, 10, 10, 10,
10, 10, 0.01])[0]
self._update_user_kibble(user_id, amount)
await interaction.response.send_message
(f'{interaction.user.mention} earned {amount} Kibble!')
async def earn_kibble(self, interaction: discord.Interaction):
user_id = str(interaction.user.id)
now = datetime.utcnow()
@tree.command(name='balance', description='Check your Kibble balance')
async def balance(interaction: discord.Interaction):
user_id = str(interaction.user.id)
balance = self._get_user_kibble(user_id)
await interaction.response.send_message
(f'{interaction.user.mention} has {balance} Kibble.')
if user_id in self.cooldowns:
if now < self.cooldowns[user_id]:
retry_after = (self.cooldowns[user_id] - now).total_seconds()
await interaction.response.send_message(
f'{interaction.user.mention}, you are on a cooldown. Please wait {retry_after:.2f} seconds before using this command again.',
ephemeral=True
)
return
self.cooldowns[user_id] = now + timedelta(seconds=30)
amount = random.choices([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 100], [0.01, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 0.01])[0]
self._update_user_kibble(user_id, amount)
await interaction.response.send_message(f'{interaction.user.mention} earned {amount} Kibble!')
async def balance(self, interaction: discord.Interaction):
user_id = str(interaction.user.id)
balance = self._get_user_kibble(user_id)
await interaction.response.send_message(f'{interaction.user.mention} has {balance} Kibble.')
def setup(self, tree):
tree.add_command(app_commands.Command(
name='earn_kibble',
description='Earn Kibble',
callback=self.earn_kibble
))
tree.add_command(app_commands.Command(
name='balance',
description='Check your Kibble balance',
callback=self.balance
))
def setup(bot):