Add Guild event handlers

2019-12-16 15:38:35 -09:00
parent 665ce34735
commit 3e8bbe8695
2 changed files with 48 additions and 2 deletions
@@ -0,0 +1,46 @@
+import discord
+from discord.ext import commands
+import logging
+import json
+
+guild_logger = logging.getLogger('guilds')
+
+
+class GuildEvents(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    async def on_guild_join(self, guild):
+        default_config = {
+            'prefixes': [self.bot.default_prefix],
+            'id': guild.id
+        }
+        admin_role = guild.roles[-1] if len(guild.roles) > 1 else 0
+        resp = await self.bot.aio_session.post(f'{self.bot.api_base}/guilds',
+                                               headers=self.bot.auth_header,
+                                               json=default_config)
+        guild_logger.info(f'Guild Joined {guild.name}: {await resp.json()}')
+        if resp.status == 200:
+            for role in guild.roles:
+                role_dict = {
+                    'id': str(role.id),
+                    'role_type': 100 if role == admin_role else 0
+                }
+                role_resp = await self.bot.aio_session.post(f'{self.bot.api_base}/guilds/{guild.id}/roles',
+                                                            headers=self.bot.auth_header,
+                                                            json=role_dict)
+                guild_logger.info(f'Role Added {role.name}: {await role_resp.json()}')
+        guild_logger.info(f'All Entries Created for {guild.name}')
+        await guild.me.edit(nick=f'[{self.bot.default_prefix}] Geeksbot')
+
+    async def on_guild_remove(self, guild):
+        resp = await self.bot.aio_session.delete(f'{self.bot.api_base}/guilds/{guild.id}',
+                                                 headers=self.bot.auth_header)
+        if resp.status == 200:
+            guild_logger.info(f'Left Guild {guild.name}: {await resp.json()}')
+        else:
+            guild_logger.error(f'Error Deleting Guild {guild.name}: {await resp.json()}')
+
+
+def setup(bot):
+    bot.add_cog(GuildEvents(bot))