BanGame initial commit

bangame/__init__.py

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

bangame/bangame.py

@@ -0,0 +1,113 @@
+import asyncio
+
+import discord
+from redbot.core import Config, checks, commands
+from typing import Any
+
+Cog: Any = getattr(commands, "Cog", object)
+
+
+class BanGame(Cog):
+    """
+    Ban anyone playing the chosen games
+    """
+
+    def __init__(self, bot):
+        self.bot = bot
+        self.config = Config.get_conf(self, identifier=66971107197109101)
+        default_guild = {"banned_games": [], "do_ban": False}
+
+        self.config.register_guild(**default_guild)
+
+    @commands.guild_only()
+    @commands.group(aliases=["exclusiverole"])
+    async def bangame(self, ctx):
+        """Base command for managing exclusive roles"""
+
+        if not ctx.invoked_subcommand:
+            pass
+
+    @bangame.command(name="toggleban")
+    @checks.mod_or_permissions(administrator=True)
+    async def bangame_toggleban(self, ctx):
+        """Toggles kicking and banning"""
+
+        do_ban = not self.config.guild(ctx.guild).do_ban()
+        await self.config.guild(ctx.guild).do_ban.set(do_ban)
+
+        await ctx.send("Members will now be {} for playing a banned game".format("Banned" if do_ban else "Kicked"))
+
+    @bangame.command(name="add")
+    @checks.mod_or_permissions(administrator=True)
+    async def bangame_add(self, ctx, game):
+        """Adds a banned game"""
+        if game in (await self.config.guild(ctx.guild).banned_games()):
+            await ctx.send("That game is already banned")
+            return
+
+        async with self.config.guild(ctx.guild).banned_games() as bg:
+            bg.append(game)
+
+        await self.check_guild(ctx.guild)
+
+        await ctx.send("Banned game added: {}".format(game))
+
+    @bangame.command(name="delete")
+    @checks.mod_or_permissions(administrator=True)
+    async def bangame_delete(self, ctx, game):
+        """Deletes a banned game"""
+        if game not in (await self.config.guild(ctx.guild).banned_games()):
+            await ctx.send("That game is not banned")
+            return
+
+        async with self.config.guild(ctx.guild).banned_games() as bg:
+            bg.remove(game)
+
+        await ctx.send("{} is no longer banned".format(game))
+
+    @bangame.command(name="list")
+    @checks.mod_or_permissions(administrator=True)
+    async def bangame_list(self, ctx):
+        """List current banned games"""
+        banned_games = await self.config.guild(ctx.guild).banned_games()
+
+        out = "**Banned Games**\n\n"
+
+        for game in banned_games:
+            out += "{}\n".format(game)
+
+        await ctx.send(out)
+
+    async def check_guild(self, guild: discord.Guild):
+        game_set = set(await self.config.guild(guild).banned_games())
+        for member in guild.members:
+            try:
+                await self.ban_or_kick_banned_games(member, game_set=game_set)
+            except discord.Forbidden:
+                pass
+
+    async def ban_or_kick_banned_games(self, member: discord.Member, game_set=None):
+        if game_set is None:
+            game_set = set(await self.config.guild(member.guild).banned_games())
+
+        if member.activity is not None and member.activity.name in game_set:
+            do_ban = await self.config.guild(member.guild).do_ban()
+
+            if do_ban:
+                await member.ban(reason="Plays {}".format(member.activity.name))
+            else:
+                await member.kick(reason="Plays {}".format(member.activity.name))
+
+    async def on_member_update(self, before: discord.Member, after: discord.Member):
+        if before.activity == after.activity:
+            return
+
+        await asyncio.sleep(1)
+
+        game_set = set(await self.config.guild(after.guild).banned_games())
+
+        if after.activity is not None and after.activity.name in game_set:
+            try:
+                await self.ban_or_kick_banned_games(after, game_set=game_set)
+            except discord.Forbidden:
+                pass

bangame/info.json

@@ -0,0 +1,21 @@
+{
+  "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]load exclusiverole` and `[p]help ExclusiveRole`",
+  "short": "Set roles to be exclusive",
+  "tags": [
+    "fox",
+    "bobloy",
+    "utility",
+    "tools",
+    "roles"
+  ]
+}