import discord from discord.ext import commands from discord import app_commands from datetime import datetime, timedelta import random class Economy(commands.Cog): def __init__(self, bot): self.bot = bot self.db = bot.db @app_commands.command(name="balance", description="Check your balance") async def balance(self, interaction: discord.Interaction, member: discord.Member = None): target = member or interaction.user user_data = await self.db.get_user(target.id, interaction.guild.id) embed = discord.Embed( title=f"💰 {target.name}'s Balance", description=f"**Balance:** ${user_data['money']:,}", color=discord.Color.gold() ) embed.set_thumbnail(url=target.display_avatar.url) await interaction.response.send_message(embed=embed) @app_commands.command(name="daily", description="Claim daily reward") async def daily(self, interaction: discord.Interaction): user_data = await self.db.get_user(interaction.user.id, interaction.guild.id) last_daily = user_data.get('last_daily') if last_daily and datetime.utcnow() - last_daily < timedelta(days=1): next_daily = last_daily + timedelta(days=1) await interaction.response.send_message(f"⏰ You already claimed your daily reward! Come back ", ephemeral=True) return amount = random.randint(500, 1000) await self.db.add_money(interaction.user.id, interaction.guild.id, amount) await self.db.update_user(interaction.user.id, interaction.guild.id, {'last_daily': datetime.utcnow()}) embed = discord.Embed( title="💵 Daily Reward", description=f"You claimed your daily reward of **${amount:,}**!", color=discord.Color.green() ) await interaction.response.send_message(embed=embed) @app_commands.command(name="weekly", description="Claim weekly reward") async def weekly(self, interaction: discord.Interaction): user_data = await self.db.get_user(interaction.user.id, interaction.guild.id) last_weekly = user_data.get('last_weekly') if last_weekly and datetime.utcnow() - last_weekly < timedelta(days=7): next_weekly = last_weekly + timedelta(days=7) await interaction.response.send_message(f"⏰ You already claimed your weekly reward! Come back ", ephemeral=True) return amount = random.randint(3000, 5000) await self.db.add_money(interaction.user.id, interaction.guild.id, amount) await self.db.update_user(interaction.user.id, interaction.guild.id, {'last_weekly': datetime.utcnow()}) embed = discord.Embed( title="💵 Weekly Reward", description=f"You claimed your weekly reward of **${amount:,}**!", color=discord.Color.green() ) await interaction.response.send_message(embed=embed) @app_commands.command(name="work", description="Work for money") async def work(self, interaction: discord.Interaction): user_data = await self.db.get_user(interaction.user.id, interaction.guild.id) last_work = user_data.get('last_work') if last_work and datetime.utcnow() - last_work < timedelta(hours=1): next_work = last_work + timedelta(hours=1) await interaction.response.send_message(f"⏰ You already worked! Come back ", ephemeral=True) return jobs = [ ("as a developer", 300, 500), ("as a teacher", 200, 400), ("as a cashier", 150, 300), ("as a delivery driver", 250, 450), ("as a freelancer", 350, 600) ] job, min_pay, max_pay = random.choice(jobs) amount = random.randint(min_pay, max_pay) await self.db.add_money(interaction.user.id, interaction.guild.id, amount) await self.db.update_user(interaction.user.id, interaction.guild.id, {'last_work': datetime.utcnow()}) embed = discord.Embed( title="💼 Work", description=f"You worked {job} and earned **${amount:,}**!", color=discord.Color.blue() ) await interaction.response.send_message(embed=embed) @app_commands.command(name="beg", description="Beg for money") async def beg(self, interaction: discord.Interaction): if random.randint(1, 100) <= 60: amount = random.randint(10, 100) await self.db.add_money(interaction.user.id, interaction.guild.id, amount) await interaction.response.send_message(f"🙏 Someone gave you **${amount:,}**!") else: await interaction.response.send_message("😔 No one gave you money. Try again later!") @app_commands.command(name="pay", description="Pay someone") async def pay(self, interaction: discord.Interaction, member: discord.Member, amount: int): if member.bot: await interaction.response.send_message("❌ You cannot pay bots", ephemeral=True) return if member == interaction.user: await interaction.response.send_message("❌ You cannot pay yourself", ephemeral=True) return if amount <= 0: await interaction.response.send_message("❌ Amount must be positive", ephemeral=True) return result = await self.db.remove_money(interaction.user.id, interaction.guild.id, amount) if not result: await interaction.response.send_message("❌ You don't have enough money", ephemeral=True) return await self.db.add_money(member.id, interaction.guild.id, amount) embed = discord.Embed( title="💸 Payment Sent", description=f"{interaction.user.mention} paid **${amount:,}** to {member.mention}", color=discord.Color.green() ) await interaction.response.send_message(embed=embed) @app_commands.command(name="rob", description="Rob someone") async def rob(self, interaction: discord.Interaction, member: discord.Member): if member.bot: await interaction.response.send_message("❌ You cannot rob bots", ephemeral=True) return if member == interaction.user: await interaction.response.send_message("❌ You cannot rob yourself", ephemeral=True) return target_data = await self.db.get_user(member.id, interaction.guild.id) if target_data['money'] < 100: await interaction.response.send_message(f"❌ {member.mention} doesn't have enough money to rob", ephemeral=True) return if random.randint(1, 100) <= 50: amount = int(target_data['money'] * random.uniform(0.1, 0.3)) await self.db.remove_money(member.id, interaction.guild.id, amount) await self.db.add_money(interaction.user.id, interaction.guild.id, amount) await interaction.response.send_message(f"💰 You successfully robbed **${amount:,}** from {member.mention}!") else: fine = random.randint(100, 500) await self.db.remove_money(interaction.user.id, interaction.guild.id, fine) await interaction.response.send_message(f"🚔 You got caught! You paid a fine of **${fine:,}**") @app_commands.command(name="baltop", description="View money leaderboard") async def baltop(self, interaction: discord.Interaction): users = await self.db.users.find({'guild_id': interaction.guild.id}).sort('money', -1).limit(10).to_list(length=10) embed = discord.Embed( title="💰 Money Leaderboard", color=discord.Color.gold() ) for i, user_data in enumerate(users, 1): member = interaction.guild.get_member(user_data['user_id']) if member: embed.add_field( name=f"#{i} {member.name}", value=f"**${user_data['money']:,}**", inline=False ) await interaction.response.send_message(embed=embed) async def setup(bot): await bot.add_cog(Economy(bot))