Exclusive Role initial commit

exclusiverole/__init__.py

@@ -0,0 +1,5 @@
+from .exclusiverole import ExclusiveRole
+
+
+def setup(bot):
+    bot.add_cog(ExclusiveRole(bot))

exclusiverole/exclusiverole.py

@@ -0,0 +1,84 @@
+import discord
+from redbot.core import Config, checks, commands
+
+
+class ExclusiveRole:
+    """
+    Custom commands
+    Creates commands used to display text and adjust roles
+    """
+
+    def __init__(self, bot):
+        self.bot = bot
+        self.config = Config.get_conf(self, identifier=9999114111108101)
+        default_guild = {
+            "role_set": set()
+        }
+
+        self.config.register_guild(**default_guild)
+
+    @commands.group(no_pm=True)
+    async def exclusive(self, ctx):
+        """Base command for managing exclusive roles"""
+
+        if not ctx.invoked_subcommand:
+            pass
+
+    @exclusive.command(name="add")
+    @checks.mod_or_permissions(administrator=True)
+    async def exclusive_add(self, ctx, role: discord.Role):
+        """Adds an exclusive role"""
+        if role.id in (await self.config.guild(ctx.guild).role_list()):
+            await ctx.send("That role is already exclusive")
+            return
+
+        async with self.config.guild(ctx.guild).role_set() as rs:
+            rs.add(role.id)
+
+        await self.check_guild(ctx.guild)
+
+        await ctx.send("Exclusive role added")
+
+    @exclusive.command(name="delete")
+    @checks.mod_or_permissions(administrator=True)
+    async def exclusive_delete(self, ctx, role: discord.Role):
+        """Deletes an exclusive role"""
+        if role.id not in (await self.config.guild(ctx.guild).role_list()):
+            await ctx.send("That role is not exclusive")
+            return
+
+        async with self.config.guild(ctx.guild).role_set() as rs:
+            rs.remove(role.id)
+
+        await ctx.send("Exclusive role removed")
+
+    async def check_guild(self, guild: discord.Guild):
+        role_set = await self.config.guild(guild).role_set()
+        for member in guild.members:
+            try:
+                await self.remove_non_exclusive_roles(member, role_set=role_set)
+            except discord.Forbidden:
+                pass
+
+    async def remove_non_exclusive_roles(self, member: discord.Member, role_set=None):
+        if role_set is None:
+            role_set = await self.config.guild(member.guild).role_set()
+
+        member_set = set([role.id for role in member.roles])
+        to_remove = member_set - role_set
+
+        if to_remove and member_set & role_set:
+            await member.remove_roles(*to_remove, "Exclusive roles")
+
+    async def on_member_update(self, before: discord.Member, after: discord.Member):
+        if before.roles == after.roles:
+            return
+
+        role_set = await self.config.guild(after.guild).role_set()
+        member_set = set([role.id for role in after.roles])
+
+        if role_set & member_set and member_set - role_set:
+            try:
+                await self.remove_non_exclusive_roles(after, role_set=role_set)
+            except discord.Forbidden:
+                pass

exclusiverole/info.json

@@ -0,0 +1,22 @@
+{
+  "author": [
+    "Bobloy"
+  ],
+  "bot_version": [
+    3,
+    0,
+    0
+  ],
+  "description": "Assign roles to be exclusive, preventing other roles from being added",
+  "hidden": false,
+  "install_msg": "Thank you for installing ExclusiveRole. Get started with `[p]help ExclusiveRole`",
+  "requirements": [],
+  "short": "Set roles to be exclusive",
+  "tags": [
+    "fox",
+    "bobloy",
+    "utility",
+    "tools",
+    "roles"
+  ]
+}