Implemented manual subtitles translation using Google Translate

1 files changed, 136 insertions, 0 deletions

diff --git a/libs/deep_translator/pons.py b/libs/deep_translator/pons.py
new file mode 100644
index 000000000..e4383eb5b
--- /dev/null
+++ b/libs/deep_translator/pons.py
@@ -0,0 +1,136 @@
+"""
+pons translator API
+"""
+from bs4 import BeautifulSoup
+import requests
+from deep_translator.constants import BASE_URLS, PONS_LANGUAGES_TO_CODES, PONS_CODES_TO_LANGUAGES
+from deep_translator.exceptions import (LanguageNotSupportedException,
+                                        TranslationNotFound,
+                                        NotValidPayload,
+                                        ElementNotFoundInGetRequest,
+                                        RequestError,
+                                        TooManyRequests)
+from deep_translator.parent import BaseTranslator
+from requests.utils import requote_uri
+
+
+class PonsTranslator(BaseTranslator):
+    """
+    class that uses PONS translator to translate words
+    """
+    _languages = PONS_LANGUAGES_TO_CODES
+    supported_languages = list(_languages.keys())
+
+    def __init__(self, source, target="english"):
+        """
+        @param source: source language to translate from
+        @param target: target language to translate to
+        """
+        self.__base_url = BASE_URLS.get("PONS")
+
+        if self.is_language_supported(source, target):
+            self._source, self._target = self._map_language_to_code(source, target)
+
+        super().__init__(base_url=self.__base_url,
+                         source=self._source,
+                         target=self._target,
+                         payload_key=None,
+                         element_tag='div',
+                         element_query={"class": "target"}
+                         )
+
+    @staticmethod
+    def get_supported_languages(as_dict=False):
+        """
+          return the supported languages by the linguee translator
+          @param as_dict: if True, the languages will be returned as a dictionary mapping languages to their abbreviations
+          @return: list or dict
+          """
+        return PonsTranslator.supported_languages if not as_dict else PonsTranslator._languages
+
+    def _map_language_to_code(self, *languages, **kwargs):
+        """
+           map language to its corresponding code (abbreviation) if the language was passed by its full name by the user
+           @param languages: list of languages
+           @return: mapped value of the language or raise an exception if the language is not supported
+        """
+        for language in languages:
+            if language in self._languages.values():
+                yield PONS_CODES_TO_LANGUAGES[language]
+            elif language in self._languages.keys():
+                yield language
+            else:
+                raise LanguageNotSupportedException(language)
+
+    def is_language_supported(self, *languages, **kwargs):
+        """
+         check if the language is supported by the translator
+         @param languages: list of languages
+         @return: bool or raise an Exception
+         """
+        for lang in languages:
+            if lang not in self._languages.keys():
+                if lang not in self._languages.values():
+                    raise LanguageNotSupportedException(lang)
+        return True
+
+    def translate(self, word, return_all=False, **kwargs):
+        """
+        function that uses PONS to translate a word
+        @param word: word to translate
+        @type word: str
+        @param return_all: set to True to return all synonym of the translated word
+        @type return_all: bool
+        @return: str: translated word
+        """
+        if self._validate_payload(word, max_chars=50):
+            url = "{}{}-{}/{}".format(self.__base_url, self._source, self._target, word)
+            url = requote_uri(url)
+            response = requests.get(url)
+
+            if response.status_code == 429:
+                raise TooManyRequests()
+
+            if response.status_code != 200:
+                raise RequestError()
+
+            soup = BeautifulSoup(response.text, 'html.parser')
+            elements = soup.findAll(self._element_tag, self._element_query)
+
+            if not elements:
+                raise ElementNotFoundInGetRequest(word)
+
+            filtered_elements = []
+            for el in elements:
+                temp = ''
+                for e in el.findAll('a'):
+                    if e.parent.name == 'div':
+                        if e and "/translate/{}-{}/".format(self._target, self._source) in e.get('href'):
+                            temp += e.get_text() + ' '
+                filtered_elements.append(temp)
+
+            if not filtered_elements:
+                raise ElementNotFoundInGetRequest(word)
+
+            word_list = [word for word in filtered_elements if word and len(word) > 1]
+
+            if not word_list:
+                raise TranslationNotFound(word)
+
+            return word_list if return_all else word_list[0]
+
+    def translate_words(self, words, **kwargs):
+        """
+        translate a batch of words together by providing them in a list
+        @param words: list of words you want to translate
+        @param kwargs: additional args
+        @return: list of translated words
+        """
+        if not words:
+            raise NotValidPayload(words)
+
+        translated_words = []
+        for word in words:
+            translated_words.append(self.translate(payload=word))
+        return translated_words
+