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

added another check

Browse source 
credit add and credit remove will now check if the new balance is below the minimum balance - on that note, added min_bal as a config option
This commit is contained in:
SeaswimmerTheFsh 2023-03-14 19:49:55 -04:00
parent 8f9c774f68
commit 019f68a1e8
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
sugoncredit/sugoncredit.py
View file

@ -10,7 +10,8 @@ class SugonCredit(commands.Cog):
self.config.register_global( self.config.register_global(
bank_name = "Social Credit Enforcement Agency", bank_name = "Social Credit Enforcement Agency",
currency_name = "Social Credit", currency_name = "Social Credit",
max_bal = 1000000000 max_bal = 1000000000,
min_bal = -1000000000
) )
con = sqlite3.connect('credit_db') con = sqlite3.connect('credit_db')
cur = con.cursor() cur = con.cursor()
@ -94,6 +95,7 @@ class SugonCredit(commands.Cog):
bank_name = await self.config.bank_name() bank_name = await self.config.bank_name()
currency_name = await self.config.currency_name() currency_name = await self.config.currency_name()
max_bal = await self.config.max_bal() max_bal = await self.config.max_bal()
min_bal = await self.config_min_bal()
if cur.execute(f'''SELECT user_id FROM credit if cur.execute(f'''SELECT user_id FROM credit
WHERE EXISTS (SELECT user_id FROM credit WHERE {target.id});''')=="FALSE": WHERE EXISTS (SELECT user_id FROM credit WHERE {target.id});''')=="FALSE":
await self.new_user_generation(self, target) await self.new_user_generation(self, target)
@ -109,9 +111,12 @@ class SugonCredit(commands.Cog):
output_amount = (f'{val:,}') output_amount = (f'{val:,}')
output_new_bal = (f'{new_bal:,}') output_new_bal = (f'{new_bal:,}')
output_max_bal = (f'{max_bal:,}') output_max_bal = (f'{max_bal:,}')
output_min_bal = (f'{min_bal:,}')
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 {output_max_bal}. Please try again!") await ctx.send(content=f"You are attempting to set {target.mention}'s balance to above {output_max_bal}. Please try again!")
return return
elif new_bal < min_bal:
await ctx.send(content=f"You are attempting to set {target.mention}'s balance to below {output_min_bal}. Please try again!")
elif ctx.guild.id == 204965774618656769: elif ctx.guild.id == 204965774618656769:
logging_channel = self.bot.get_channel(1082495815878189076) logging_channel = self.bot.get_channel(1082495815878189076)
cur.execute(f'''UPDATE credit cur.execute(f'''UPDATE credit
@ -149,6 +154,7 @@ class SugonCredit(commands.Cog):
bank_name = await self.config.bank_name() bank_name = await self.config.bank_name()
currency_name = await self.config.currency_name() currency_name = await self.config.currency_name()
max_bal = await self.config.max_bal() max_bal = await self.config.max_bal()
min_bal = await self.config_min_bal()
if cur.execute(f'''SELECT user_id FROM credit if cur.execute(f'''SELECT user_id FROM credit
WHERE EXISTS (SELECT user_id FROM credit WHERE {target.id});''')=="FALSE": WHERE EXISTS (SELECT user_id FROM credit WHERE {target.id});''')=="FALSE":
await self.new_user_generation(self, target) await self.new_user_generation(self, target)
@ -162,8 +168,11 @@ class SugonCredit(commands.Cog):
output_amount = (f'{val:,}') output_amount = (f'{val:,}')
output_new_bal = (f'{new_bal:,}') output_new_bal = (f'{new_bal:,}')
output_max_bal = (f'{max_bal:,}') output_max_bal = (f'{max_bal:,}')
output_min_bal = (f'{min_bal:,}')
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 {output_max_bal}. Please try again!") await ctx.send(content=f"You are attempting to set {target.mention}'s balance to above {output_max_bal}. Please try again!")
elif new_bal < min_bal:
await ctx.send(content=f"You are attempting to set {target.mention}'s balance to below {output_min_bal}. Please try again!")
elif ctx.guild.id == 204965774618656769: elif ctx.guild.id == 204965774618656769:
cur.execute(f'''UPDATE credit cur.execute(f'''UPDATE credit
SET balance = {new_bal} SET balance = {new_bal}