Add Translator Error

Author: EvieePy

docs/exts/commands/exceptions.rst

@@ -57,6 +57,8 @@ Exceptions
 
 .. autoexception:: twitchio.ext.commands.NoEntryPointError
 
+.. autoexception:: twitchio.ext.commands.TranslatorError
+
 
 Exception Hierarchy
 ~~~~~~~~~~~~~~~~~~~
@@ -80,6 +82,7 @@ Exception Hierarchy
                 - :exc:`ExpectedClosingQuoteError`
         - :exc:`GuardFailure`
             - :exc:`CommandOnCooldown`
+        - :exc:`TranslatorError`
     - :exc:`ModuleError`
         - :exc:`ModuleLoadFailure`
         - :exc:`ModuleAlreadyLoadedError`

twitchio/ext/commands/context.py

@@ -591,6 +591,8 @@ async def send_translated(self, content: str, *, me: bool = False, langcode: str
             Twitch failed to process the message, could be ``400``, ``401``, ``403``, ``422`` or any ``5xx`` status code.
         MessageRejectedError
             Twitch rejected the message from various checks.
+        TranslatorError
+            An error occurred during translation.
         """
         translator: Translator | None = getattr(self.command, "translator", None)
         new = (f"/me {content}" if me else content).strip()
@@ -599,14 +601,21 @@ async def send_translated(self, content: str, *, me: bool = False, langcode: str
             return await self.channel.send_message(sender=self.bot.bot_id, message=new)
 
         invoked = self.invoked_with
-        code = langcode or translator.get_langcode(self, invoked.lower()) if invoked else None
+
+        try:
+            code = langcode or translator.get_langcode(self, invoked.lower()) if invoked else None
+        except Exception as e:
+            raise TranslatorError(f"An exception occurred fetching a language code for '{invoked}'.", original=e) from e
 
         if not code:
             return await self.channel.send_message(sender=self.bot.bot_id, message=new)
 
-        translated = await translator.translate(self, content, code)
-        new_translated = (f"/me {translated}" if me else translated).strip()
+        try:
+            translated = await translator.translate(self, content, code)
+        except Exception as e:
+            raise TranslatorError(f"An exception occurred translating content for '{invoked}'.", original=e) from e
 
+        new_translated = (f"/me {translated}" if me else translated).strip()
         return await self.channel.send_message(sender=self.bot.bot_id, message=new_translated)
 
     async def reply(self, content: str, *, me: bool = False) -> SentMessage:

twitchio/ext/commands/exceptions.py

@@ -51,6 +51,7 @@
     "ModuleNotLoadedError",
     "NoEntryPointError",
     "PrefixError",
+    "TranslatorError",
     "UnexpectedQuoteError",
 )
 
@@ -72,7 +73,7 @@ class CommandInvokeError(CommandError):
     Attributes
     ----------
     original: :class:`Exception` | None
-        The original exception that caused this error. Could be None.
+        The original exception that caused this error. Could be ``None``.
     """
 
     def __init__(self, msg: str | None = None, original: Exception | None = None) -> None:
@@ -204,3 +205,18 @@ def __init__(self, msg: str | None = None, *, cooldown: BaseCooldown, remaining:
         self.cooldown: BaseCooldown = cooldown
         self.remaining: float = remaining
         super().__init__(msg or f"Cooldown is ratelimited. Try again in {remaining} seconds.")
+
+
+class TranslatorError(CommandError):
+    """Exception raised when a :class:`.commands.Translator` raises an error while attempting to get a language code
+    or translate content provided to :meth:`~.commands.Context.send_translated`.
+
+    Attributes
+    ----------
+    original: :class:`Exception` | None
+        The original exception that caused this error. Could be None.
+    """
+
+    def __init__(self, msg: str | None = None, original: Exception | None = None) -> None:
+        self.original: Exception | None = original
+        super().__init__(msg)