translate: catch requests exceptions

sopel-irc · Jul 9, 2021 · 12af23d · 12af23d
1 parent 677cdbf
commit 12af23d
Showing 1 changed file with 35 additions and 2 deletions.
diff --git a/sopel/modules/translate.py b/sopel/modules/translate.py
@@ -9,6 +9,7 @@
 from __future__ import generator_stop
 
 import json
+import logging
 import random
 
 import requests
@@ -17,6 +18,7 @@
 from sopel.tools import web
 
 
+LOGGER = logging.getLogger(__name__)
 PLUGIN_OUTPUT_PREFIX = '[translate] '
 
 
@@ -64,6 +66,9 @@ def translate(text, in_lang='auto', out_lang='en'):
     try:
         data = json.loads(result)
     except ValueError:
+        LOGGER.error(
+            'Error parsing JSON response from translate API (%s to %s: "%s")',
+            in_lang, out_lang, text)
         return None, None
 
     if raw:
@@ -100,7 +105,21 @@ def tr(bot, trigger):
         bot.reply('Language guessing failed, so try suggesting one!')
         return
 
-    msg, in_lang = translate(phrase, in_lang, out_lang)
+    try:
+        msg, in_lang = translate(phrase, in_lang, out_lang)
+    except requests.Timeout:
+        bot.reply("Translation service unavailable (timeout).")
+        LOGGER.error(
+            'Translate API error (%s to %s: "%s"): timeout.',
+            in_lang, out_lang, phrase)
+        return
+    except requests.RequestException as http_error:
+        bot.reply("Failed to request the translation service.")
+        LOGGER.exception(
+            'Translate API error (%s to %s: "%s"): %s.',
+            in_lang, out_lang, phrase, http_error)
+        return
+
     if not in_lang:
         bot.reply("Translation failed, probably because of a rate-limit.")
         return
@@ -164,7 +183,21 @@ def langcode(p):
         bot.reply('Language guessing failed, so try suggesting one!')
         return
 
-    msg, src = translate(phrase, src, dest)
+    try:
+        msg, src = translate(phrase, src, dest)
+    except requests.Timeout:
+        bot.reply("Translation service unavailable (timeout).")
+        LOGGER.error(
+            'Translate API error (%s to %s: "%s"): timeout.',
+            src, dest, phrase)
+        return
+    except requests.RequestException as http_error:
+        bot.reply("Failed to request the translation service.")
+        LOGGER.exception(
+            'Translate API error (%s to %s: "%s"): %s.',
+            src, dest, phrase, http_error)
+        return
+
     if not src:
         return bot.say("Translation failed, probably because of a rate-limit.")