Added logic to remove short role name for all members by deleting
This commit is contained in:
@@ -30,6 +30,41 @@ class ShortRoleNameService:
|
||||
self._server = server
|
||||
self._short_role_names = short_role_names
|
||||
|
||||
def _remove_short_role_name_from_name(self, short_role_name: ShortRoleName, name: str) -> str:
|
||||
return (
|
||||
name.replace(f" [{short_role_name.short_name}]", "")
|
||||
.replace(f"[{short_role_name.short_name}] ", "")
|
||||
.replace(f"[{short_role_name.short_name}]", "")
|
||||
)
|
||||
|
||||
async def remove_short_role_from_members(self, short_role_name: ShortRoleName):
|
||||
guild = self._bot.get_guild(short_role_name.server.discord_id)
|
||||
settings: ServerConfig = self._config.get_configuration(f"ServerConfig_{guild.id}")
|
||||
for member in guild.members:
|
||||
try:
|
||||
member_roles = List(discord.Role, member.roles)
|
||||
highest_role: Optional[discord.Role] = (
|
||||
member_roles.last_or_default() if settings.short_role_name_only_set_highest_role else None
|
||||
)
|
||||
member_role_ids = member_roles.select(lambda x: x.id)
|
||||
|
||||
if (
|
||||
short_role_name.role_id not in member_role_ids
|
||||
or highest_role
|
||||
and highest_role.id != short_role_name.role_id
|
||||
):
|
||||
continue
|
||||
|
||||
self._logger.debug(__name__, f"Update member {member.id}")
|
||||
await member.edit(
|
||||
nick=self._remove_short_role_name_from_name(
|
||||
short_role_name, member.nick if member.nick is not None else member.name
|
||||
)
|
||||
)
|
||||
self._logger.debug(__name__, f"Updated member {member.id} {member.name}")
|
||||
except Exception as e:
|
||||
self._logger.error(__name__, f"Renaming member {member.name} failed", e)
|
||||
|
||||
async def check_short_role_names(self, member: discord.Member):
|
||||
self._logger.debug(__name__, f"Started short role name check for {member.id}")
|
||||
settings: ServerConfig = self._config.get_configuration(f"ServerConfig_{member.guild.id}")
|
||||
@@ -53,9 +88,7 @@ class ShortRoleNameService:
|
||||
for short_role_name in self._short_role_names.get_short_role_names_by_server_id(server.id):
|
||||
short_role_name: ShortRoleName = short_role_name
|
||||
|
||||
new_nick = new_nick.replace(f" [{short_role_name.short_name}]", "")
|
||||
new_nick = new_nick.replace(f"[{short_role_name.short_name}] ", "")
|
||||
new_nick = new_nick.replace(f"[{short_role_name.short_name}]", "")
|
||||
new_nick = self._remove_short_role_name_from_name(short_role_name, new_nick)
|
||||
|
||||
if (
|
||||
short_role_name.role_id not in member_role_ids
|
||||
|
Reference in New Issue
Block a user