error handlers

bot/plugins/context/context.py

@@ -8,6 +8,10 @@
 plugin = Plugin()
 
 
+class NoSauceError(IndexError):
+    pass
+
+
 @plugin.include
 @crescent.message_command(name="Find Sauce")
 async def sauce(ctx: crescent.Context, message: hikari.Message):
@@ -16,16 +20,29 @@ async def sauce(ctx: crescent.Context, message: hikari.Message):
     if message.attachments:
         for att in message.attachments:
             sourced = await sauceclient.from_url(att.url)
+            if not sourced:
+                raise NoSauceError
             await ctx.respond(sourced[0].url)
     elif message.content:
         if "https://" not in message.content:
             return await ctx.respond("Nothing to be sauced")
         for word in message.content.split():
             if word.startswith("https://"):
-                try:
-                    sourced = await sauceclient.from_url(word)
-                    await ctx.respond(sourced[0].url)
-                except errors.InvalidImageException:
-                    await ctx.respond("Invalid url")
+                sourced = await sauceclient.from_url(word)
+                await ctx.respond(sourced[0].url)
     else:
         await ctx.respond("Nothing to be sauced")
+
+
+@plugin.include
+@crescent.catch_command(NoSauceError)
+async def no_sauce_handler(err: NoSauceError, ctx: crescent.Context):
+    await ctx.respond("Can't sauce message :<")
+
+
+@plugin.include
+@crescent.catch_command(errors.InvalidImageException)
+async def invalid_image_handler(
+    err: errors.InvalidImageException, ctx: crescent.Context
+):
+    await ctx.respond("Invalid url")

bot/plugins/main/events.py

@@ -37,3 +37,11 @@ async def startup(event: hikari.GuildAvailableEvent):
     for channel_id in guild.get_channels():
         channel = guild.get_channel(channel_id)
         logger.debug(f"L Connected to #{channel.name}({channel.id})")
+
+
+@plugin.include
+@crescent.catch_command(Exception)
+async def on_any_command_error(exc: Exception, ctx: crescent.Context):
+    await ctx.respond("Unknown error. Report to maintainer")
+    logger.error(f"{exc}")
+    raise