reupload

1 files changed, 62 insertions, 0 deletions

diff --git a/code/cogs/coinflip.py b/code/cogs/coinflip.py
new file mode 100644
index 0000000..9771cbd
--- /dev/null
+++ b/code/cogs/coinflip.py
@@ -0,0 +1,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))
\ No newline at end of file