added new translation provider - LibreTranslate (#55)

* added new translation provider - LibreTranslate

* fix format

README.md

@@ -69,6 +69,7 @@ Currently supported services are:
 - `Google`
 - `Bing`
 - `Deepl`
+- 'LibreTranslate'
 
 ### Configuring with a Configuration File
 

md_translate/translators/__init__.py

@@ -4,6 +4,7 @@
 from .bing import BingTranslateProvider
 from .deepl import DeeplTranslateProvider
 from .google import GoogleTranslateProvider
+from .libretranslate import LibreTranslateTranslateProvider
 from .yandex import YandexTranslateProvider
 
 
@@ -12,3 +13,4 @@ class Translator(enum.Enum):
     google = GoogleTranslateProvider
     bing = BingTranslateProvider
     deepl = DeeplTranslateProvider
+    libretranslate = LibreTranslateTranslateProvider

md_translate/translators/libretranslate.py

@@ -0,0 +1,52 @@
+from typing import Any
+
+from selenium.common.exceptions import NoSuchElementException
+from selenium.webdriver.remote.webelement import WebElement
+
+from md_translate.exceptions import safe_run
+
+from ._selenium_base import SeleniumBaseTranslator
+
+
+class LibreTranslateTranslateProvider(SeleniumBaseTranslator):
+    HOST = 'https://libretranslate.com/'
+
+    def get_url(self) -> str:
+        params = {
+            'source': self.from_language,
+            'target': self.to_language,
+        }
+        return f'{self.HOST}?{self.build_params(params)}'
+
+    def get_input_element(self) -> WebElement:
+        return self._driver.find_element(by=self.WEBDRIVER_BY.ID, value='textarea1')
+
+    def get_output_element(self) -> WebElement:
+        return self._driver.find_element(by=self.WEBDRIVER_BY.ID, value='textarea2')
+
+    @safe_run(NoSuchElementException, default_return_value=False)
+    def check_for_translation(self) -> bool:
+        element = self.get_output_element()
+        return element.get_attribute('value') != ''
+
+    def accept_cookies(self) -> None:
+        return
+
+    @safe_run(NoSuchElementException, default_return_value=False)
+    def check_for_antispam(self) -> bool:
+        container = self._driver.find_element(by=self.WEBDRIVER_BY.ID, value='t_enter_captcha')
+        return container.is_displayed()
+
+    @staticmethod
+    def get_translated_data(output_element: WebElement) -> str:
+        return output_element.get_attribute('value')
+
+    def wait_for_page_load(self) -> None:
+        def wait_for(driver: Any) -> bool:
+            document_ready = driver.execute_script('return document.readyState') == 'complete'
+            controls_loaded = (
+                len(driver.find_elements(by=self.WEBDRIVER_BY.ID, value='textarea1')) > 0
+            )
+            return document_ready and controls_loaded
+
+        self.WEBDRIVER_WAIT(self._driver, self.PAGE_LOAD_TIMEOUT).until(wait_for)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "md_translate"
-version = "3.0.5"
+version = "3.1.0"
 description = "CLI tool to translate markdown files"
 authors = ["Ilya Chichak <[email protected]>"]
 license = "MIT License"

tests/test_translator.py

@@ -5,6 +5,7 @@
     DeeplTranslateProvider,
     GoogleTranslateProvider,
     YandexTranslateProvider,
+    LibreTranslateTranslateProvider,
 )
 
 
@@ -22,6 +23,7 @@ class TestTranslator:
             (GoogleTranslateProvider, 'Hello world', 'Привет, мир'),
             (BingTranslateProvider, 'Hello world', 'Всем привет'),
             (DeeplTranslateProvider, 'Hello world', 'Здравствуй мир'),
+            (LibreTranslateTranslateProvider, 'Hello world', 'Привет мир'),
         ],
     )
     def test_translate(self, translator, source_text, expected):