111 lines
4.1 KiB
Python
111 lines
4.1 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
import datetime
|
|
import random
|
|
import logging
|
|
from typing import Optional
|
|
|
|
# Import database functions from the sibling module
|
|
from . import database
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class GamblingCommands(commands.Cog):
|
|
"""Cog containing gambling-related economy commands."""
|
|
|
|
def __init__(self, bot: commands.Bot):
|
|
self.bot = bot
|
|
|
|
@commands.hybrid_command(
|
|
name="moneyflip",
|
|
aliases=["mf"],
|
|
description="Gamble your money on a coin flip.",
|
|
) # Renamed to avoid conflict
|
|
async def moneyflip(
|
|
self, ctx: commands.Context, amount: int, choice: str
|
|
): # Renamed function
|
|
"""Bets a certain amount on a coin flip (heads or tails)."""
|
|
user_id = ctx.author.id
|
|
command_name = "moneyflip" # Update command name for cooldown tracking
|
|
cooldown_duration = datetime.timedelta(seconds=10) # Short cooldown
|
|
|
|
choice = choice.lower()
|
|
if choice not in ["heads", "tails", "h", "t"]:
|
|
embed = discord.Embed(
|
|
description="❌ Invalid choice. Please choose 'heads' or 'tails'.",
|
|
color=discord.Color.red(),
|
|
)
|
|
await ctx.send(embed=embed, ephemeral=True)
|
|
return
|
|
|
|
if amount <= 0:
|
|
embed = discord.Embed(
|
|
description="❌ Please enter a positive amount to bet.",
|
|
color=discord.Color.red(),
|
|
)
|
|
await ctx.send(embed=embed, ephemeral=True)
|
|
return
|
|
|
|
# Check cooldown
|
|
last_used = await database.check_cooldown(user_id, command_name)
|
|
if last_used:
|
|
now_utc = datetime.datetime.now(datetime.timezone.utc)
|
|
if last_used.tzinfo is None:
|
|
last_used = last_used.replace(tzinfo=datetime.timezone.utc)
|
|
|
|
time_since_last_used = now_utc - last_used
|
|
if time_since_last_used < cooldown_duration:
|
|
time_left = cooldown_duration - time_since_last_used
|
|
embed = discord.Embed(
|
|
description=f"🕒 You're flipping too fast! Try again in **{int(time_left.total_seconds())}s**.",
|
|
color=discord.Color.orange(),
|
|
)
|
|
await ctx.send(embed=embed, ephemeral=True)
|
|
return
|
|
|
|
# Check balance
|
|
user_balance = await database.get_balance(user_id)
|
|
if user_balance < amount:
|
|
embed = discord.Embed(
|
|
description=f"❌ You don't have enough money to bet that much! Your balance is **${user_balance:,}**.",
|
|
color=discord.Color.red(),
|
|
)
|
|
await ctx.send(embed=embed, ephemeral=True)
|
|
return
|
|
|
|
# Set cooldown before proceeding
|
|
await database.set_cooldown(user_id, command_name)
|
|
|
|
# Perform the coin flip
|
|
result = random.choice(["heads", "tails"])
|
|
win = choice.startswith(result[0]) # True if choice matches result
|
|
|
|
if win:
|
|
await database.update_balance(user_id, amount) # Win the amount bet
|
|
current_balance = await database.get_balance(user_id)
|
|
embed = discord.Embed(
|
|
title="Coin Flip: Win!",
|
|
description=f"🪙 The coin landed on **{result}**! You won **${amount:,}**!",
|
|
color=discord.Color.green(),
|
|
)
|
|
embed.add_field(
|
|
name="New Balance", value=f"${current_balance:,}", inline=False
|
|
)
|
|
await ctx.send(embed=embed)
|
|
else:
|
|
await database.update_balance(user_id, -amount) # Lose the amount bet
|
|
current_balance = await database.get_balance(user_id)
|
|
embed = discord.Embed(
|
|
title="Coin Flip: Loss!",
|
|
description=f"🪙 The coin landed on **{result}**. You lost **${amount:,}**.",
|
|
color=discord.Color.red(),
|
|
)
|
|
embed.add_field(
|
|
name="New Balance", value=f"${current_balance:,}", inline=False
|
|
)
|
|
await ctx.send(embed=embed)
|
|
|
|
|
|
# No setup function needed here
|