subclass CommandTree

this subclass will be calling the error manager upon error

1 files changed, 19 insertions, 0 deletions

diff --git a/pydis_core/_bot.py b/pydis_core/_bot.py
index 0427c659..08dca6a4 100644
--- a/pydis_core/_bot.py
+++ b/pydis_core/_bot.py
@@ -6,12 +6,14 @@ from contextlib import suppress
 
 import aiohttp
 import discord
+from discord import app_commands
 from discord.ext import commands
 
 from pydis_core.async_stats import AsyncStatsClient
 from pydis_core.site_api import APIClient
 from pydis_core.utils import scheduling
 from pydis_core.utils._extensions import walk_extensions
+from pydis_core.utils.error_handling.commands import CommandErrorManager
 from pydis_core.utils.logging import get_logger
 
 try:
@@ -32,6 +34,23 @@ class StartupError(Exception):
         self.exception = base
 
 
+class CommandTreeBase(app_commands.CommandTree):
+    """A sub-class of the Command tree that implements common features that Python Discord bots use."""
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        # Instance is None since discordpy only passes an instance of the client to the command tree in its constructor.
+        self.command_error_manager: CommandErrorManager | None = None
+
+    async def on_error(self, interaction: discord.Interaction, error: app_commands.AppCommandError) -> None:
+        """A callback that is called when any command raises an :exc:`AppCommandError`."""
+        if not self.command_error_manager:
+            log.warning("Command error manager hasn't been loaded in the command tree.")
+            await super().on_error(interaction, error)
+            return
+        await self.command_error_manager.handle_error(error, interaction)
+
+
 class BotBase(commands.Bot):
     """
     A sub-class that implements many common features that Python Discord bots use.