Merge pull request #23 from HeDeAnTheonlyone/main

add mod sub command "role"

Author: Silabear

.gitignore

@@ -9,6 +9,7 @@ member_data.json
 mute_lengths.json
 modlogs.json
 data/
+member_names.txt
 
 *.vscode/
 *.vscode/*

README.md

@@ -58,6 +58,9 @@ comm_helper_B = 12345
 # report_ping: the id of the role to ping when a report is created
 report_ping = 12345
 
+# roles that moderators can grant and remove
+mod_edit_roles = [1234567890, 9876543210]
+
 #! Misc
 
 # guild: the id of the active guild

modules/moderation/commands/mod.py

@@ -4,7 +4,6 @@
 from datetime import datetime, timedelta
 import variables
 import io
-import re
 from utils.uwufier import Uwuifier
 import utils.modlogs as modlogs
 import matplotlib.pyplot as plt
@@ -33,7 +32,6 @@
     },
 ]
 
-
 def generate_discord_relative_timestamp(seconds):
     # Calculate the future Unix timestamp
     future_timestamp = int((datetime.now() + timedelta(seconds=seconds)).timestamp())
@@ -280,3 +278,46 @@ async def banall(self, inter: disnake.ApplicationCommandInteraction):
             colour=disnake.Colour.dark_red()
         )
         await inter.response.send_message(embed=emb,components=[disnake.ui.Button(style=disnake.ButtonStyle.red,label="CONFIRM")])
+
+    async def role_autocomplete(self, inter: disnake.ApplicationCommandInteraction, string: str):
+        # Filter the roles to only those whose ID is in the whitelist
+        filtered_roles = [
+            role for role in inter.guild.roles 
+            if role.id in variables.mod_edit_roles and string.lower() in role.name.lower()
+        ]
+        
+        # Return the filtered roles as options for autocomplete
+        return [disnake.OptionChoice(name=role.name, value=role.id) for role in filtered_roles]
+
+    def role_list(self):
+        return [disnake.OptionChoice(name=role.name,value=role.id) for role in self.bot.guild.roles if role.id in variables.mod_edit_roles]
+
+    @mod.sub_command("role","grants or removes a (non-vital) role")
+    async def role(
+        self,
+        inter: disnake.ApplicationCommandInteraction,
+        user: disnake.Member,
+        modification: str = commands.Param(
+            default = "add",
+            choices = ["add", "remove"]
+        ),
+        role: disnake.Role = commands.Param()
+    ):
+        if (not role.id in variables.mod_edit_roles):
+            await inter.response.send_message(f"Role {role.name} is not allowed in this command", ephemeral=True)
+            return
+
+        try:
+            if (modification == "add"):
+                if (role in user.roles):
+                    await inter.response.send_message(f"User {user.name} already has the role {role}.", ephemeral=True)
+                else:
+                    await user.add_roles(role)
+            else:
+                if (role in user.roles):
+                    await user.remove_roles(role)
+                else:
+                    await inter.response.send_message(f"User {user.name} doesn't have the role {role}.", ephemeral=True)
+        except:
+            await inter.response.send_message(f"Failed to {modification} role {role.name}, to user {user.name}", ephemeral=True)
+        await inter.response.send_message(f"Role {modification + ('ed' if modification == 'add' else 'd')}", ephemeral=True)

modules/utilities/commands/members.py

@@ -9,6 +9,8 @@ def __init__(self, bot):
     @commands.slash_command(name="members", description="View member list and count")
     async def members(self, inter: disnake.ApplicationCommandInteraction):
 
-        with open("Member Names.txt","w+") as file:
+        with open("member_names.txt","w+") as file:
             file.writelines([member.name + "\n" for member in inter.guild.members])
-            await inter.response.send_message(f"`{inter.guild.member_count}` members",file=disnake.File(file),ephemeral=True)
+        
+        dc_file = disnake.File("member_names.txt")
+        await inter.response.send_message(f"`{inter.guild.member_count}` members",file=dc_file,ephemeral=True)