1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import logging
from aiohttp import ClientResponseError
from discord.ext.commands import Bot, Context
from discord.ext.commands import (
BadArgument,
BotMissingPermissions,
CommandError,
CommandInvokeError,
CommandNotFound,
NoPrivateMessage,
UserInputError,
)
log = logging.getLogger(__name__)
class ErrorHandler:
"""Handles errors emttted from commands."""
def __init__(self, bot: Bot):
self.bot = bot
async def on_command_error(self, ctx: Context, e: CommandError):
command = ctx.command
parent = None
if command is not None:
parent = command.parent
if parent and command:
help_command = (self.bot.get_command("help"), parent.name, command.name)
elif command:
help_command = (self.bot.get_command("help"), command.name)
else:
help_command = (self.bot.get_command("help"),)
if hasattr(command, "on_error"):
log.debug(f"Command {command} has a local error handler, ignoring.")
return
if isinstance(e, CommandNotFound) and not hasattr(ctx, "invoked_from_error_handler"):
tags_get_command = self.bot.get_command("tags get")
ctx.invoked_from_error_handler = True
# Return to not raise the exception
return await ctx.invoke(tags_get_command, tag_name=ctx.invoked_with)
elif isinstance(e, BadArgument):
await ctx.send(f"Bad argument: {e}\n")
await ctx.invoke(*help_command)
elif isinstance(e, UserInputError):
await ctx.invoke(*help_command)
elif isinstance(e, NoPrivateMessage):
await ctx.send("Sorry, this command can't be used in a private message!")
elif isinstance(e, BotMissingPermissions):
await ctx.send(
f"Sorry, it looks like I don't have the permissions I need to do that.\n\n"
f"Here's what I'm missing: **{e.missing_perms}**"
)
elif isinstance(e, CommandInvokeError):
if isinstance(e.original, ClientResponseError):
if e.original.code == 404:
await ctx.send("There does not seem to be anything matching your query.")
else:
await ctx.send("BEEP BEEP UNKNOWN API ERROR!=?!??!?!?!?")
else:
await ctx.send(
f"Sorry, an unexpected error occurred. Please let us know!\n\n```{e}```"
)
raise e.original
raise e
def setup(bot: Bot):
bot.add_cog(ErrorHandler(bot))
log.info("Cog loaded: Events")
|
