Added test cases and support for translate_batch and translate_file

deep_translator/libre.py

@@ -3,8 +3,6 @@ LibreTranslate API
 """
 
 import requests
-from bs4 import BeautifulSoup
-from requests.models import Response
 from .parent import BaseTranslator
 from .constants import BASE_URLS,LIBRE_LANGUAGES_TO_CODES, LIBRE_CODES_TO_LANGUAGES
 from .exceptions import (ServerException,
@@ -21,7 +19,7 @@ class LibreTranslator(BaseTranslator):
     _languages = LIBRE_LANGUAGES_TO_CODES
     supported_languages = list(_languages.keys())
 
-    def __init__(self, base_url = BASE_URLS.get("LIBRE_FREE"), api_key="", source="auto", target="en", **kwargs):
+    def __init__(self,source="auto", target="en", base_url = BASE_URLS.get("LIBRE_FREE"), api_key=None, **kwargs):
         """
         @param source: source language to translate from
         List of LibreTranslate nedpoints can be found at : https://github.com/LibreTranslate/LibreTranslate#mirrors
@@ -86,8 +84,11 @@ class LibreTranslator(BaseTranslator):
             "q": text,
             "source": self.source,
             "target": self.target,
-            "api_key": self.api_key
+            "format": 'text'
         }
+        # Add API Key if required
+        if self.api_key: 
+            params["api_key"] = self.api_key
         # Do the request and check the connection.
         try:
             response = requests.post(self.__base_url + translate_endpoint, params=params)
@@ -105,3 +106,32 @@ class LibreTranslator(BaseTranslator):
             raise TranslationNotFound(text)
         # Process and return the response.
         return res['translatedText']
+
+    def translate_file(self, path, **kwargs):
+        """
+        translate directly from file
+        @param path: path to the target file
+        @type path: str
+        @param kwargs: additional args
+        @return: str
+        """
+        try:
+            with open(path) as f:
+                text = f.read().strip()
+            return self.translate(text)
+        except Exception as e:
+            raise e
+
+    def translate_batch(self, batch=None, **kwargs):
+        """
+        translate a list of texts
+        @param batch: list of texts you want to translate
+        @return: list of translations
+        """
+        if not batch:
+            raise Exception("Enter your text list that you want to translate")
+        arr = []
+        for i, text in enumerate(batch):
+            translated = self.translate(text, **kwargs)
+            arr.append(translated)
+        return arr

deep_translator/main.py

@@ -10,6 +10,7 @@ from .pons import PonsTranslator
 from .yandex import YandexTranslator
 from .microsoft import MicrosoftTranslator
 from .papago import PapagoTranslator
+from .libre import LibreTranslator
 
 CONTEXT_SETTINGS = dict(help_option_names=["-h", "--help"])
 @click.group()
@@ -64,6 +65,11 @@ def translate(translator, source, target, text, api_key):
             source=source,
             target=target,
             api_key=api_key)
+    elif translator == "libre":
+        translator= LibreTranslator(
+            source=source,
+            target=target
+        )
     else:
         raise AttributeError("The given translator is not supported.")
 
@@ -105,6 +111,8 @@ def languages(translator, api_key):
         translator = MicrosoftTranslator(api_key=api_key)
     elif translator == "papago":
         translator = PapagoTranslator(api_key=api_key)
+    elif translator == "libre":
+        translator = LibreTranslator
     else:
         raise AttributeError("The given translator is not supported.")
 
@@ -117,7 +125,7 @@ def languages(translator, api_key):
 @cli.command()
 def list():
     """Lists available translators."""
-    click.echo("Available translators include: Google, MyMemory, QCRI, Linguee, Pons, Yandex, Microsoft (Bing), and Papago.")
+    click.echo("Available translators include: Google, MyMemory, QCRI, Linguee, Pons, Yandex, Microsoft (Bing), Papago and LibreTranslate.")
     return 0
 
 if __name__ == "__main__":

examples/libre.py

@@ -1,7 +1,5 @@
-
 from deep_translator import LibreTranslator
 
+res = LibreTranslator(source='de', target='en').translate('laufen')
 
-# res = LibreTranslator(source='de', target='en').translate('laufen', return_all=False)
-
-# print(res)
+print(res)

tests/test_libre.py

@@ -4,7 +4,29 @@
 
 import pytest
 from deep_translator import exceptions, LibreTranslator
-
+from deep_translator.constants import LIBRE_CODES_TO_LANGUAGES
+import random
+
+test_text_standard = 'Hello world.'
+TRANSLATED_RESULTS = {
+    'English': 'Hello world.', 
+    'Arabic': 'مرحبا العالم.', 
+    'Chinese': '霍洛美世界。', 
+    'French': 'Bonjour.', 
+    'German': 'Hallo Welt.',
+    'Hindi': 'नमस्ते दुनिया।', 
+    'Indonesian': 'Halo dunia.', 
+    'Irish': 'Dia duit domhan.', 
+    'Italian': 'Ciao mondo.', 
+    'Japanese': 'こんにちは。',
+    'Korean': '안녕하세요.', 
+    'Polish': 'Hello world (ang.).', 
+    'Portuguese': 'Olá, mundo.', 
+    'Russian': 'Привет мир.', 
+    'Spanish': 'Hola mundo.', 
+    'Turkish': 'Merhaba dünya.', 
+    'Vietnamese': 'Xin chào.'
+}
 
 @pytest.fixture
 def libre():
@@ -17,6 +39,13 @@ def test_content(libre):
     # assert 'GitHub' in BeautifulSoup(response.content).title.string
     assert libre.translate(text='good') is not None
 
+def test_random_tranlations_cases_multiple_names():
+    random_sample_size = 2
+    d = dict.fromkeys(list(TRANSLATED_RESULTS.keys()))
+    random_lang_names = random.sample(list(d.keys()), random_sample_size)
+    random_subset_dict = {k: TRANSLATED_RESULTS[k] for k in random_lang_names}
+    for lang, translation in random_subset_dict.items():
+        assert LibreTranslator(source='en', target=lang).translate(test_text_standard) == translation
 
 def test_inputs():
     with pytest.raises(exceptions.LanguageNotSupportedException):
@@ -25,11 +54,12 @@ def test_inputs():
     with pytest.raises(exceptions.LanguageNotSupportedException):
         LibreTranslator(source="auto", target="nothing")
 
-    l1 = LibreTranslator("en", "fr")
-    l2 = LibreTranslator("english", "french")
-    assert l1.source == l2.source
-    assert l1.target == l2.target
-
+def test_abbreviations_and_languages_mapping():
+    for abb, lang in LIBRE_CODES_TO_LANGUAGES.items():
+        if abb != 'en':
+            l1 = LibreTranslator(abb)
+            l2 = LibreTranslator(lang)
+            assert l1.source == l2.source
 
 def test_payload(libre):
     with pytest.raises(exceptions.NotValidPayload):
@@ -47,3 +77,13 @@ def test_payload(libre):
 
 def test_one_character_words():
     assert LibreTranslator(source='es', target='en').translate('y') == 'and'
+
+def test_translate_batch():
+    words_to_translate = ['How are you','Good','Thank You']
+    translated_words = ['¿Cómo estás?','Bien.','Gracias.']
+    assert LibreTranslator(source='en',target='es').translate_batch((words_to_translate)) == translated_words
+
+def test_translate_file():
+    filePath = 'examples/test.txt'
+    translatedText='Un párrafo es una serie de frases relacionadas que desarrollan una idea central, llamada el tema. Trate de pensar en los párrafos en términos de unidad temática: un párrafo es una frase o un grupo de oraciones que apoya una idea central y unificada. Los párrafos añaden una idea a la vez a su argumento más amplio.'
+    assert LibreTranslator(source='en',target='es').translate_file(filePath) == translatedText