CodeMirror/Brandon209-Red-bot-Cogs
1
0
Fork 0
mirror of https://github.com/brandons209/Red-bot-Cogs.git synced 2024-05-18 19:32:29 +12:00
Code Issues Projects Releases Wiki Activity

added where if users with subscribed roles loose free roles they charged

Browse source
This commit is contained in:
brandons209 2020-05-07 19:51:55 -04:00
parent f7a2861c3b
commit 79bd86000e
1 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
rolemanagement/core.py
View file

@ -166,13 +166,16 @@ class RoleManagement(
del role_data["subscribed_users"][user_id]
continue
# charge user
cost = await self.config.role(role).cost()
raw_cost = await self.config.role(role).cost()
cost = await self.get_cost(member, role)
currency_name = await bank.get_currency_name(guild)
curr_sub = await self.config.role(role).subscription()
if cost == 0 or curr_sub == 0:
if raw_cost == 0 or curr_sub == 0:
# role is free now or sub is removed, remove stale sub
del role_data["subscribed_users"][user_id]
continue
if cost == 0:
continue
msg = f"Hello! You are being charged {cost} {currency_name} for your subscription to the {role.name} role in {guild.name}."
try:
@ -221,6 +224,17 @@ class RoleManagement(
if len(role_data["subscribed_users"]) == 0:
s_roles.remove(role_id)
async def get_cost(self, member: discord.Member, role: discord.Role):
""" Gets cost of a role for a user """
cost = await self.config.role(role).cost()
free_roles = await self.config.guild(member.guild).free_roles()
for m_role in member.roles:
if m_role.id in free_roles:
return 0
return cost
@commands.guild_only()
@commands.bot_has_permissions(manage_roles=True)
@checks.admin_or_permissions(manage_roles=True)
@ -877,17 +891,20 @@ class RoleManagement(
for m_role in ctx.author.roles:
if m_role.id in free_roles:
await ctx.send(f"You're special, no {currency_name} will be deducted from your account.")
await self.update_roles_atomically(who=ctx.author, give=[role], remove=remove)
await ctx.tick()
return
cost = 0
# await self.update_roles_atomically(who=ctx.author, give=[role], remove=remove)
# await ctx.tick()
# return
try:
await bank.withdraw_credits(ctx.author, cost)
if cost > 0:
await bank.withdraw_credits(ctx.author, cost)
except ValueError:
return await ctx.send(f"You don't have enough {currency_name} (Cost: {cost} {currency_name})")
else:
if subscription > 0:
await ctx.send(f"{role.name} will be renewed every {parse_seconds(subscription)}")
if cost > 0:
await ctx.send(f"{role.name} will be renewed every {parse_seconds(subscription)}")
async with self.config.role(role).subscribed_users() as s:
s[str(ctx.author.id)] = time.time() + subscription
async with self.config.guild(ctx.guild).s_roles() as s: