aboutsummaryrefslogtreecommitdiff
path: root/code/cogs/owner/toggle.py
diff options
context:
space:
mode:
authorParker <contact@pkrm.dev>2025-03-25 23:17:22 -0500
committerParker <contact@pkrm.dev>2025-03-25 23:17:22 -0500
commit03e7111604a786e212e7b55f6d9ca9f84be9b432 (patch)
treed4c00b1e140e872bbde8addffa254a5fa8801d20 /code/cogs/owner/toggle.py
parentb343d0d45b0bafadaf2947aef4b2fe2a19d726f5 (diff)
Protect against errors
Diffstat (limited to 'code/cogs/owner/toggle.py')
-rw-r--r--code/cogs/owner/toggle.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/code/cogs/owner/toggle.py b/code/cogs/owner/toggle.py
index a276b01..bdb4b4a 100644
--- a/code/cogs/owner/toggle.py
+++ b/code/cogs/owner/toggle.py
@@ -19,7 +19,12 @@ class Toggle(commands.Cog):
self.bot.youtube_broken = True
return await ctx.send("YouTube has been marked as broken.")
- await ctx.send("Invalid action. Please use 'on' or 'off'.")
+ @toggle.error
+ async def toggle_error(self, ctx, error):
+ if isinstance(error, commands.BadLiteralArgument):
+ return await ctx.send("Invalid action. Use either 'on' or 'off'.")
+ else:
+ return await ctx.send("An unknown error occurred.")
async def setup(bot):