diff options
author | Parker <contact@pkrm.dev> | 2025-03-25 23:17:22 -0500 |
---|---|---|
committer | Parker <contact@pkrm.dev> | 2025-03-25 23:17:22 -0500 |
commit | 03e7111604a786e212e7b55f6d9ca9f84be9b432 (patch) | |
tree | d4c00b1e140e872bbde8addffa254a5fa8801d20 /code/cogs/owner/toggle.py | |
parent | b343d0d45b0bafadaf2947aef4b2fe2a19d726f5 (diff) |
Protect against errors
Diffstat (limited to 'code/cogs/owner/toggle.py')
-rw-r--r-- | code/cogs/owner/toggle.py | 7 |
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): |