ruby/Ruby-Cogs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Enhance role validation in ItemManager by adding support for role mentions. Update hierarchy_check method to correctly handle role mentions and streamline role name checks, improving user interaction and validation logic.
Some checks are pending
Run pre-commit / Run pre-commit (push) Waiting to run
Details

Browse source
This commit is contained in:
Valerie 2025-05-26 00:08:08 -04:00
parent 85de2c70c1
commit 35e0ac182d
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
shop/shop.py
View file

@ -1495,13 +1495,19 @@ class ItemManager:
return int(cost.content) return int(cost.content)
def hierarchy_check(self, m): def hierarchy_check(self, m):
# Extract role ID from mention if it's a mention
if m.role_mentions:
role = m.role_mentions[0]
if self.ctx.author == m.author:
if self.ctx.author.top_role >= role:
return True
return False
# Regular role name check
roles = [r.name for r in self.ctx.guild.roles if r.name != "Bot"] roles = [r.name for r in self.ctx.guild.roles if r.name != "Bot"]
if self.ctx.author == m.author and m.content in roles: if self.ctx.author == m.author and m.content in roles:
if self.ctx.author.top_role >= discord.utils.get(self.ctx.message.guild.roles, name=m.content): if self.ctx.author.top_role >= discord.utils.get(self.ctx.message.guild.roles, name=m.content):
return True return True
else:
return False
else:
return False return False
async def set_role(self, item=None, shop=None): async def set_role(self, item=None, shop=None):