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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
import discord
from discord.ext import commands
from discord import app_commands
import random
from bot import InsufficientFundsException
from global_variables import BOT_COLOR, BONUS_COOLDOWN
from database import Database
class GamblingHelpers(commands.Cog):
def __init__(self, bot: commands.Bot) -> None:
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()
# This is only here temporarily, because I accidentally deleted the database
@commands.command()
@commands.is_owner()
@commands.dm_only()
async def refund(
self,
ctx: commands.Context,
id,
amount: int
):
user = await self.bot.fetch_user(id)
if user:
await self.economy.add_money(id, amount)
await ctx.send(f"Refunded {user} ${amount:,}")
else:
await ctx.send("User not found")
@app_commands.command()
@app_commands.checks.cooldown(1, BONUS_COOLDOWN*3600)
async def add(
self,
interaction: discord.Interaction
):
f"Add $10,000 to your balance every {BONUS_COOLDOWN} hours"
amount = 10000
await self.economy.add_money(interaction.user.id, amount)
embed = discord.Embed(
title="I've added $10,000 to you balance",
description=f"Come back again in {BONUS_COOLDOWN} hours.",
color=BOT_COLOR
)
await interaction.response.send_message(embed=embed, ephemeral=True)
@app_commands.command()
@app_commands.checks.cooldown(1, 120)
async def work(
self,
interaction: discord.Interaction
):
"Work for a randomized amount of money every 2 minutes"
a = random.randint(500, 2500)
b = random.randint(500, 2500)
if a == b:
num = 50000000
else:
num = a + b
await self.economy.add_money(interaction.user.id, num)
embed = discord.Embed(
title=f"You worked and earned ${num:,}",
description="Come back again in 2 minutes.",
color=BOT_COLOR
)
await interaction.response.send_message(embed=embed, ephemeral=True)
@app_commands.command()
async def leaderboard(
self,
interaction: discord.Interaction
):
"Show the global currency leaderboard"
entries = await self.economy.top_entries(5)
embed = discord.Embed(
title="Global Economy Leaderboard:",
description="",
color=discord.Color.gold()
)
for i, entry in enumerate(entries):
id = entry[0]
name = f"<@{id}>"
embed.description += f"**`{i+1}.` {name}**\n${entry[1]:,}\n\n"
await interaction.response.send_message(embed=embed)
@app_commands.command()
@app_commands.describe(user="The user to give money to")
@app_commands.describe(amount="The amount of money to give the user")
async def give(
self,
interaction: discord.Interaction,
user: discord.Member,
amount: app_commands.Range[int, 1, None]
):
"Give money to another user"
if user == interaction.user:
embed = discord.Embed(
title="Self Error",
description="You cannot give money to yourself, please try again with a different user.",
color=BOT_COLOR
)
return await interaction.response.send_message(embed=embed, ephemeral=True)
else:
await self.check_bet(interaction, amount)
await self.economy.add_money(user.id, amount)
await self.economy.add_money(interaction.user.id, amount*-1)
embed = discord.Embed(
title="Gift Success",
description=f"You have successfully given {user.mention} ${amount:,}!",
color=discord.Color.green()
)
return await interaction.response.send_message(embed=embed)
async def setup(bot: commands.Bot):
await bot.add_cog(GamblingHelpers(bot))
|
