1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
import discord
from discord.ext import commands
from discord import app_commands
import random
from bot import InsufficientFundsException
from database import Database
class CoinFlip(commands.Cog):
def __init__(self, bot: commands.Bot):
self.bot = bot
self.economy = Database(bot)
async def check_bet(
self,
interaction: discord.Interaction,
bet
):
bet = int(bet)
if bet <= 0:
raise commands.errors.BadArgument()
current = (await self.economy.get_entry(interaction.user.id))[1]
if bet > current:
raise InsufficientFundsException()
@app_commands.command()
@app_commands.checks.cooldown(1, 2)
@app_commands.describe(bet="The amount of money you want to bet")
async def coinflip(
self,
interaction: discord.Interaction,
bet: int
):
"Bet money on a coinflip (heads=win, tails=lose)"
await self.check_bet(interaction, bet)
if random.randint(0, 2) == 0:
await self.economy.add_money(interaction.user.id, bet*2)
embed = discord.Embed(
title=f"You won ${bet*2:,}!",
description="You flipped heads!",
color=discord.Color.green()
)
return await interaction.response.send_message(embed=embed)
else:
await self.economy.add_money(interaction.user.id, bet*-1)
embed = discord.Embed(
title=f"You lost ${bet:,}!",
description="You flipped tails!",
color=discord.Color.red()
)
return await interaction.response.send_message(embed=embed)
async def setup(bot: commands.Bot):
await bot.add_cog(CoinFlip(bot))
|
