Add error handler helper method

Author: Paillat-dev

src/extensions/nice-errors/handler.py

@@ -136,9 +136,9 @@ def __init__(self, error_handlers: ErrorHandlersType | None = None) -> None:
             self.error_handlers = DEFAULT_ERROR_HANDLERS
 
     def _get_handler(self, error: Exception) -> ErrorHandlerType | None:
-        if handler := self.error_handlers.get(type(error)): # faster but might miss subclasses
+        if handler := self.error_handlers.get(type(error)):  # faster but might miss subclasses
             return handler
-        for error_type, handler in self.error_handlers.items(): # slower but catches subclasses
+        for error_type, handler in self.error_handlers.items():  # slower but catches subclasses
             if issubclass(type(error), error_type):
                 return handler
         return None
@@ -184,4 +184,5 @@ async def handle_error(
     def add_error_handler(self, error: type[Exception], handler: ErrorHandlerType) -> None:
         self.error_handlers[error] = handler
 
+
 error_handler = ErrorHandler()

src/extensions/nice-errors/main.py

@@ -51,6 +51,9 @@ async def on_command_error(self, ctx: custom.ExtContext, error: commands.Command
             use_sentry_sdk=self.sentry_sdk,
         )
 
+    def add_error_handler(self, *args: Any, **kwargs: Any) -> None:
+        error_handler.add_error_handler(*args, **kwargs)
+
 
 def setup(bot: custom.Bot, config: dict[str, Any]) -> None:
     bot.add_cog(NiceErrors(bot, bool(config.get("sentry", {}).get("dsn")), config))