Add cog reload|unload|load commands

2018-05-20 21:41:06 +02:00
parent baf1359d60
commit 9891c42ed9
2 changed files with 84 additions and 2 deletions
@@ -0,0 +1,82 @@
+#!/usr/bin/python
+# -*- coding: <encoding name> -*-
+
+from discord.ext import commands
+import discord
+import traceback
+
+class Upload:
+    """
+    CogName should be the name of the cog
+    """
+    def __init__(self, bot):
+        self.bot = bot
+        print('upload loaded')
+
+    @commands.command()
+    async def reload(self, ctx, *, extension: str):
+        """Reload an extension."""
+
+        if ctx.author.id not in self.bot.ownerlist:
+            return await ctx.send('Only my creator can use me like this :blush:', delete_after=1)
+
+        extension = extension.lower()
+        try:
+            self.bot.unload_extension("src.cogs.{}".format(extension))
+            self.bot.load_extension("src.cogs.{}".format(extension))
+        except Exception as e:
+            traceback.print_exc()
+            await ctx.send(f'Could not reload `{extension}` -> `{e}`')
+        else:
+            await ctx.send(f'Reloaded `{extension}`.')
+
+    @commands.command()
+    async def reloadall(self, ctx):
+        """Reload all extensions."""
+        if ctx.author.id not in self.bot.ownerlist:
+            return await ctx.send('Only my creator can use me like this :blush:', delete_after=1)
+
+        try:
+            for extension in self.bot.extensions:
+                self.bot.unload_extension(extension)
+                self.bot.load_extension(extension)
+            await ctx.send(f"Reload success! :thumbsup:\n")
+        except Exception as e:
+            await ctx.send(f"Could not reload `{extension}` -> `{e}`.\n")
+
+    @commands.command()
+    async def unload(self, ctx, *, extension: str):
+        """Unload an extension."""
+        if ctx.author.id not in self.bot.ownerlist:
+            return await ctx.send('Only my creator can use me like this :blush:', delete_after=1)
+
+        extension = extension.lower()
+        try:
+            self.bot.unload_extension("src.cogs.{}".format(extension))
+
+        except Exception as e:
+            traceback.print_exc()
+            if ctx.message.author.id not in self.bot.owner_list:
+                await ctx.send(f'Could not unload `{extension}` -> `{e}`')
+
+        else:
+            await ctx.send(f'Unloaded `{extension}`.')
+
+    @commands.command()
+    async def load(self, ctx,  *, extension: str):
+        """Load an extension."""
+        if ctx.author.id not in self.bot.ownerlist:
+            return await ctx.send('Only my creator can use me like this :blush:', delete_after=1)
+
+        extension = extension.lower()
+        try:
+            self.bot.load_extension("src.cogs.{}".format(extension))
+
+        except Exception as e:
+            traceback.print_exc()
+            await ctx.send(f'Could not unload `{extension}` -> `{e}`')
+        else:
+            await ctx.send(f'Loaded `{extension}`.')
+
+def setup(bot):
+    bot.add_cog(Upload(bot))