[RoleTools] Add support for assigning roles to thread members

TrustyJAID · Oct 5, 2024 · e459085 · e459085
1 parent 10433f3
commit e459085
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/roletools/roletools.py b/roletools/roletools.py
@@ -349,7 +349,7 @@ async def giverole(
         self,
         ctx: Context,
         role: RoleHierarchyConverter,
-        *who: Union[discord.Role, discord.TextChannel, discord.Member, str],
+        *who: Union[discord.Role, discord.TextChannel, discord.Thread, discord.Member, str],
     ) -> None:
         """
         Gives a role to designated members.
@@ -383,7 +383,16 @@ async def giverole(
         async with ctx.typing():
             members = []
             for entity in who:
-                if isinstance(entity, discord.TextChannel) or isinstance(entity, discord.Role):
+                if isinstance(entity, discord.Thread):
+                    try:
+                        thread_members = await entity.fetch_members()
+                        for m in thread_members:
+                            if mem := ctx.guild.get_member(m.id):
+                                members.append(mem)
+                    except Exception:
+                        log.error("Could not find members of thread in %s", entity)
+
+                elif isinstance(entity, discord.TextChannel) or isinstance(entity, discord.Role):
                     members += entity.members
                 elif isinstance(entity, discord.Member):
                     members.append(entity)

diff --git a/roletools/select.py b/roletools/select.py
@@ -183,7 +183,11 @@ async def create_select_menu(
             for item in replacement_view.children:
                 if item.custom_id == custom_id:
                     replacement_view.remove_item(item)
-            replacement_view.add_item(select_menus)
+            try:
+                replacement_view.add_item(select_menus)
+            except ValueError:
+                log.error("Error editing old menu on Select Menu %s", custom_id)
+                continue
 
         select_menus.update_options(ctx.guild)
         view = RoleToolsView(self, timeout=180.0)