diff --git a/techsupport_bot/commands/extension.py b/techsupport_bot/commands/extension.py index 9056524d3..f47f127dd 100644 --- a/techsupport_bot/commands/extension.py +++ b/techsupport_bot/commands/extension.py @@ -94,7 +94,10 @@ async def load_extension(self, ctx, *, extension_name: str): ctx (discord.ext.Context): the context object for the message extension_name (str): the name of the extension """ - await ctx.bot.load_extension(f"extensions.{extension_name}") + try: + await ctx.bot.load_extension(f"functions.{extension_name}") + except (ModuleNotFoundError, commands.errors.ExtensionNotFound): + await ctx.bot.load_extension(f"commands.{extension_name}") await auxiliary.send_confirm_embed( message="I've loaded that extension", channel=ctx.channel ) @@ -114,7 +117,10 @@ async def unload_extension(self, ctx, *, extension_name: str): ctx (discord.ext.Context): the context object for the message extension_name (str): the name of the extension """ - await ctx.bot.unload_extension(f"extensions.{extension_name}") + try: + await ctx.bot.unload_extension(f"functions.{extension_name}") + except commands.errors.ExtensionNotLoaded: + await ctx.bot.unload_extension(f"commands.{extension_name}") await auxiliary.send_confirm_embed( message="I've unloaded that extension", channel=ctx.channel )