cswimr/GalaxyCogs
cswimr/GalaxyCogs
1
1
Fork 0
Code Issues 2 Pull requests Projects Wiki Activity Actions

added error handling for ints + some other things

Browse source
This commit is contained in:
Seaswimmer 2023-03-05 23:24:13 -05:00
parent 825a541b0d
commit 13843a1ee8
1 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
sugoncredit/sugoncredit.py
View file

@ -34,6 +34,11 @@ class SugonCredit(commands.Cog):
@commands.mod() @commands.mod()
async def add(self, ctx, target: discord.Member, amount: int): async def add(self, ctx, target: discord.Member, amount: int):
"""Adds credits to an account.""" """Adds credits to an account."""
try:
val = int(amount)
except ValueError:
await ctx.send(content="``amount`` must be a number! Please try again.")
return
bank_name = await bank.get_bank_name(ctx.guild) bank_name = await bank.get_bank_name(ctx.guild)
currency_name = await bank.get_currency_name(ctx.guild) currency_name = await bank.get_currency_name(ctx.guild)
current_bal = await bank.get_balance(target) current_bal = await bank.get_balance(target)
@ -42,6 +47,9 @@ class SugonCredit(commands.Cog):
if new_bal > max_bal: if new_bal > max_bal:
await ctx.send(content=f"You are attempting to set {target.mention}'s balance to above {max.bal}. Please try again!") await ctx.send(content=f"You are attempting to set {target.mention}'s balance to above {max.bal}. Please try again!")
return return
elif new_bal < 0:
await ctx.send(content=f"You are attempting to set {target.mention}'s balance to below 0. Please try again!")
return
else: else:
embed=discord.Embed(title=f"{bank_name} - Add", color=await self.bot.get_embed_color(None), description=f"{target.mention}'s {currency_name} balance has been increased by {amount}.\nCurrent balance is {new_bal}.") embed=discord.Embed(title=f"{bank_name} - Add", color=await self.bot.get_embed_color(None), description=f"{target.mention}'s {currency_name} balance has been increased by {amount}.\nCurrent balance is {new_bal}.")
await bank.deposit_credits(target, amount=amount) await bank.deposit_credits(target, amount=amount)
@ -52,12 +60,17 @@ class SugonCredit(commands.Cog):
@commands.mod() @commands.mod()
async def remove(self, ctx, target: discord.Member, amount: int): async def remove(self, ctx, target: discord.Member, amount: int):
"""Removes credits from an account.""" """Removes credits from an account."""
try:
val = int(amount)
except ValueError:
await ctx.send(content="``amount`` must be a number. Please try again!")
return
bank_name = await bank.get_bank_name(ctx.guild) bank_name = await bank.get_bank_name(ctx.guild)
currency_name = await bank.get_currency_name(ctx.guild) currency_name = await bank.get_currency_name(ctx.guild)
current_bal = await bank.get_balance(target) current_bal = await bank.get_balance(target)
new_bal = current_bal - amount new_bal = current_bal - amount
if new_bal < 1: if new_bal < 0:
await ctx.send(content=f"You are attempting to set {target.mention}'s balance to below 1. Please try again!") await ctx.send(content=f"You are attempting to set {target.mention}'s balance to below 0. Please try again!")
return return
else: else:
embed=discord.Embed(title=f"{bank_name} - Remove", color=await self.bot.get_embed_color(None), description=f"{target.mention}'s {currency_name} balance has been decreased by {amount}.\nCurrent balance is {new_bal}.") embed=discord.Embed(title=f"{bank_name} - Remove", color=await self.bot.get_embed_color(None), description=f"{target.mention}'s {currency_name} balance has been decreased by {amount}.\nCurrent balance is {new_bal}.")