New option --skip-correctbot for translateodt

Author: holybiber

pywikitools/test/test_translateodt.py

@@ -262,6 +262,28 @@ def test_translate_worksheet_aborts(self, mock_correctbot, mock_fortraininglib):
 
         # TODO test all the main functionality of translate_worksheet()
 
+    @patch("pywikitools.translateodt.ForTrainingLib", autospec=True)
+    @patch("pywikitools.translateodt.CorrectBot", autospec=True)
+    def test_translate_worksheet_skips_correctbot(
+        self, mock_correctbot, mock_fortraininglib
+    ):
+        translateodt = TranslateODT(
+            keep_english_file=True,
+            skip_correctbot=True,
+            config={"translateodt": {"site": "4training"}},
+        )
+        translateodt._loffice = Mock(spec=LibreOffice)
+        self.addCleanup(translateodt.fortraininglib.session.close)
+
+        mock_fortraininglib.return_value.get_translation_units.return_value = None
+        with self.assertLogs("pywikitools.translateodt", level="ERROR"):
+            self.assertIsNone(translateodt.translate_worksheet("Prayer", "de"))
+
+        mock_correctbot.assert_not_called()
+        mock_fortraininglib.return_value.get_translation_units.assert_called_with(
+            "Prayer", "de"
+        )
+
 
 if __name__ == "__main__":
     unittest.main()

pywikitools/translateodt/__init__.py

@@ -30,10 +30,15 @@ def _missing_spaces_exceptions(self) -> List[str]:
 
 class TranslateODT:
     def __init__(
-        self, *, keep_english_file: bool = False, config: Dict[str, Dict[str, str]] = {}
+        self,
+        *,
+        keep_english_file: bool = False,
+        skip_correctbot: bool = False,
+        config: Dict[str, Dict[str, str]] = {},
     ):
         """Variable initializations (no connection to LibreOffice here)
         @param keep_english_file: Don't delete English ODT file afterwards (command line option)
+        @param skip_correctbot: Skip the pre-flight CorrectBot check (command line option)
         @param config: can be used to overwrite config settings
         """
         # Read configuration from config.ini in this folder; set default values in case it doesn't exist
@@ -47,7 +52,10 @@ def __init__(
         self.config.read(join(dirname(abspath(__file__)), "..", "..", "config.ini"))
         self.config.read_dict(config)
 
-        self.correctbot = CorrectBot(self.config, simulate=True)
+        self.skip_correctbot: bool = skip_correctbot
+        self.correctbot = (
+            None if skip_correctbot else CorrectBot(self.config, simulate=True)
+        )
 
         self.logger = logging.getLogger("pywikitools.translateodt")
         self.keep_english_file: bool = keep_english_file
@@ -363,13 +371,14 @@ def translate_worksheet(self, worksheet: str, language_code: str) -> Optional[st
             file name of the created ODT file (or None in case of error)
         """
         self.logger.debug(f"Worksheet: {worksheet}, language code: {language_code}")
-        self.correctbot.check_page(worksheet, language_code)
-        if (
-            self.correctbot.get_correction_counter() > 0
-            or self.correctbot.get_warning_counter() > 0
-        ):
-            self.logger.error("Please run CorrectBot first and then try again.")
-            return None
+        if not self.skip_correctbot:
+            self.correctbot.check_page(worksheet, language_code)
+            if (
+                self.correctbot.get_correction_counter() > 0
+                or self.correctbot.get_warning_counter() > 0
+            ):
+                self.logger.error("Please run CorrectBot first and then try again.")
+                return None
         translated_page: Optional[TranslatedPage] = (
             self.fortraininglib.get_translation_units(worksheet, language_code)
         )