from discord.ext import commands import discord from .utils import checks import re import asyncio class Roles: """Class to handle management of roles on the server""" def __init__(self, bot): self.bot = bot @commands.group(aliases=['roles'], invoke_without_command=True, no_pm=True, pass_context=True) @checks.custom_perms(send_messages=True) async def role(self, ctx): """This command can be used to modify the roles on the server. Pass no subcommands and this will print the roles currently available on this server""" # Simply get a list of all roles in this server and send them server_roles = [role.name for role in ctx.message.server.roles if not role.is_everyone] await self.bot.say("Your server's roles are: ```\n{}```".format("\n".join(server_roles))) @role.command(name='remove', pass_context=True, no_pm=True) @checks.custom_perms(manage_roles=True) async def remove_role(self, ctx): """Use this to remove roles from a number of members""" # No use in running through everything if the bot cannot manage roles if not ctx.message.server.me.permissions_in(ctx.message.channel).manage_roles: await self.bot.say("I can't manage roles in this server, do you not trust me? :c") return server_roles = [role for role in ctx.message.server.roles if not role.is_everyone] # First get the list of all mentioned users members = ctx.message.mentions # If no users are mentioned, ask the author for a list of the members they want to remove the role from if len(members) == 0: await self.bot.say("Please provide the list of members you want to remove a role from") msg = await self.bot.wait_for_message(author=ctx.message.author, channel=ctx.message.channel) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return if len(msg.mentions) == 0: await self.bot.say("I cannot remove a role from someone if you don't provide someone...") return # Override members if everything has gone alright, and then continue members = msg.mentions # This allows the user to remove multiple roles from the list of users, if they want. await self.bot.say("Alright, please provide the roles you would like to remove from this member. " "Make sure the roles, if more than one is provided, are separate by commas. " "Here is a list of this server's roles:" "```\n{}```".format("\n".join([r.name for r in server_roles]))) msg = await self.bot.wait_for_message(author=ctx.message.author, channel=ctx.message.channel) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return # Split the content based on commas, using regex so we can split if a space was not provided or if it was role_names = re.split(', ?', msg.content) roles = [] # This loop is just to get the actual role objects based on the name for role in role_names: _role = discord.utils.get(server_roles, name=role) if _role is not None: roles.append(_role) # If no valid roles were given, let them know that and return if len(roles) == 0: await self.bot.say("Please provide a valid role next time!") return # Otherwise, remove the roles from each member given for member in members: await self.bot.remove_roles(member, *roles) await self.bot.say("I have just removed the following roles:```\n{}``` from the following members:" "```\n{}```".format("\n".join(role_names), "\n".join([m.display_name for m in members]))) @role.command(name='add', pass_context=True, no_pm=True) @checks.custom_perms(manage_roles=True) async def add_role(self, ctx): """Use this to add a role to multiple members. Provide the list of members, and I'll ask for the role If no members are provided, I'll first ask for them""" # No use in running through everything if the bot cannot manage roles if not ctx.message.server.me.permissions_in(ctx.message.channel).manage_roles: await self.bot.say("I can't manage roles in this server, do you not trust me? :c") return # This is exactly the same as removing roles, except we call add_roles instead. server_roles = [role for role in ctx.message.server.roles if not role.is_everyone] members = ctx.message.mentions if len(members) == 0: await self.bot.say("Please provide the list of members you want to add a role to") msg = await self.bot.wait_for_message(author=ctx.message.author, channel=ctx.message.channel) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return if len(msg.mentions) == 0: await self.bot.say("I cannot add a role to someone if you don't provide someone...") return members = msg.mentions await self.bot.say("Alright, please provide the roles you would like to add to this member. " "Make sure the roles, if more than one is provided, are separate by commas. " "Here is a list of this server's roles:" "```\n{}```".format("\n".join([r.name for r in server_roles]))) msg = await self.bot.wait_for_message(author=ctx.message.author, channel=ctx.message.channel) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return role_names = re.split(', ?', msg.content) roles = [] for role in role_names: _role = discord.utils.get(server_roles, name=role) if _role is not None: roles.append(_role) if len(roles) == 0: await self.bot.say("Please provide a valid role next time!") return for member in members: await self.bot.add_roles(member, *roles) await self.bot.say("I have just added the following roles:```\n{}``` to the following members:" "```\n{}```".format("\n".join(role_names), "\n".join([m.display_name for m in members]))) @role.command(name='delete', pass_context=True, no_pm=True) @checks.custom_perms(manage_roles=True) async def delete_role(self, ctx, *, role: discord.Role = None): """This command can be used to delete one of the roles from the server""" # No use in running through everything if the bot cannot manage roles if not ctx.message.server.me.permissions_in(ctx.message.channel).manage_roles: await self.bot.say("I can't delete roles in this server, do you not trust me? :c") return # If no role was given, get the current roles on the server and ask which ones they'd like to remove if role is None: server_roles = [role for role in ctx.message.server.roles if not role.is_everyone] await self.bot.say( "Which role would you like to remove from the server? Here is a list of this server's roles:" "```\n{}```".format("\n".join([r.name for r in server_roles]))) # For this method we're only going to delete one role at a time # This check attempts to find a role based on the content provided, if it can't find one it returns None # We can use that fact to simply use just that as our check check = lambda m: discord.utils.get(server_roles, name=m.content) msg = await self.bot.wait_for_message(author=ctx.message.author, channel=ctx.message.channel, check=check) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return # If we have gotten here, based on our previous check, we know that the content provided is a valid role. # Due to that, no need for any error checking here role = discord.utils.get(server_roles, name=msg.content) await self.bot.delete_role(ctx.message.server, role) await self.bot.say("I have just removed the role {} from this server".format(role.name)) @role.command(name='create', pass_context=True, no_pm=True) @checks.custom_perms(manage_roles=True) async def create_role(self, ctx): """This command can be used to create a new role for this server A prompt will follow asking what settings you would like for this new role I'll then ask if you'd like to set anyone to use this role""" # No use in running through everything if the bot cannot create the role if not ctx.message.server.me.permissions_in(ctx.message.channel).manage_roles: await self.bot.say("I can't create roles in this server, do you not trust me? :c") return # Save a couple variables that will be used repeatedly author = ctx.message.author server = ctx.message.server channel = ctx.message.channel # A couple checks that will be used in the wait_for_message's num_separated_check = lambda m: re.search("\d(,| )", m.content) is not None yes_no_check = lambda m: re.search("(yes|no)", m.content.lower()) is not None members_check = lambda m: len(m.mentions) > 0 # Start the checks for the role, get the name of the role first await self.bot.say( "Alright! I'm ready to create a new role, please respond with the name of the role you want to create") msg = await self.bot.wait_for_message(timeout=60.0, author=author, channel=channel) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return name = msg.content # Print a list of all the permissions available, then ask for which ones need to be active on this new role all_perms = [p for p in dir(discord.Permissions) if isinstance(getattr(discord.Permissions, p), property)] fmt = "\n".join("{}) {}".format(i, perm) for i, perm in enumerate(all_perms)) await self.bot.say("Sounds fancy! Here is a list of all the permissions available. Please respond with just " "the numbers, seperated by commas, of the permissions you want this role to have.\n" "```\n{}```".format(fmt)) # For this we're going to give a couple extra minutes before we timeout, as it might take a bit to figure out which permissions they want msg = await self.bot.wait_for_message(timeout=180.0, author=author, channel=channel, check=num_separated_check) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return # Check if any integer's were provided that are within the length of the list of permissions num_permissions = [int(i) for i in re.split(' ?,?', msg.content) if i.isdigit() and int(i) < len(all_perms)] if len(num_permissions) == 0: await self.bot.say("You did not provide any valid numbers! Try better next time.") return # Check if this role should be in a separate section on the sidebard, i.e. hoisted await self.bot.say("Do you want this role to be in a separate section on the sidebar? (yes or no)") msg = await self.bot.wait_for_message(timeout=60.0, author=author, channel=channel, check=yes_no_check) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return hoist = True if msg.content.lower() == "yes" else False # Check if this role should be able to be mentioned await self.bot.say("Do you want this role to be mentionable? (yes or no)") msg = await self.bot.wait_for_message(timeout=60.0, author=author, channel=channel, check=yes_no_check) if msg is None: await self.bot.say("You took too long. I'm impatient, don't make me wait") return mentionable = True if msg.content.lower() == "yes" else False # Ready to actually create the role # First create a permissions object based on the numbers provided perms = discord.Permissions.none() for index in num_permissions: setattr(perms, all_perms[index], True) payload = { 'name': name, 'permissions': perms, 'hoist': hoist, 'mentionable': mentionable } # Create the role, and wait a second, sometimes it goes too quickly and we get a role with 'new role' to print role = await self.bot.create_role(server, **payload) await asyncio.sleep(1) await self.bot.say("We did it! You just created the new role {}\nIf you want to add this role" " to some people, mention them now".format(role.name)) msg = await self.bot.wait_for_message(timeout=60.0, author=author, channel=channel, check=members_check) # There's no need to mention the users, so don't send a failure message if they didn't, just return if msg is None: return # Otherwise members were mentioned, add the new role to them now for member in msg.mentions: await self.bot.add_roles(member, role) fmt = "\n".join(m.display_name for m in msg.mentions) await self.bot.say("I have just added the role {} to: ```\n{}```".format(name, fmt)) def setup(bot): bot.add_cog(Roles(bot))