File size: 3,434 Bytes
23ed530 58568a7 23ed530 58568a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
from __future__ import absolute_import
from abc import ABC
from constants import BASE_URLS, GOOGLE_LANGUAGES_TO_CODES
from exceptions import LanguageNotSupportedException, ElementNotFoundInGetRequest, NotValidPayload, NotValidLength
from parent import BaseTranslator
from bs4 import BeautifulSoup
import requests
class GoogleTranslator(BaseTranslator, ABC):
"""
class that uses google translate to translate texts
"""
def __init__(self, source="auto", target="en"):
"""
@param source: source language to translate from
@param target: target language to translate to
"""
self.__base_url = BASE_URLS.get("GOOGLE_TRANSLATE")
if self.is_language_supported(source, target):
self._source, self._target = self._map_language_to_code(source.lower(), target.lower())
super(GoogleTranslator, self).__init__(base_url=self.__base_url,
source=self._source,
target=self._target,
element_tag='div',
element_query={"class": "t0"},
payload_key='q', # key of payload in the url
hl=self._target,
sl=self._source)
def _map_language_to_code(self, *languages, **kwargs):
"""
@param language: type of language
@return: mapped value of the language or raise an exception if the language is not supported
"""
for language in languages:
if language in GOOGLE_LANGUAGES_TO_CODES.values() or language == 'auto':
yield language
elif language in GOOGLE_LANGUAGES_TO_CODES.keys():
yield GOOGLE_LANGUAGES_TO_CODES[language]
else:
raise LanguageNotSupportedException(language)
def is_language_supported(self, *languages, **kwargs):
for lang in languages:
if lang != 'auto' and lang not in GOOGLE_LANGUAGES_TO_CODES.keys():
if lang != 'auto' and lang not in GOOGLE_LANGUAGES_TO_CODES.values():
raise LanguageNotSupportedException(lang)
return True
def translate(self, payload, **kwargs):
"""
main function that uses google translate to translate a text
@param payload: desired text to translate
@return: str: translated text
"""
if not self.validate_payload(payload):
raise NotValidPayload(payload)
if not self._check_length(payload):
raise NotValidLength(payload)
try:
payload = payload.strip()
if self.payload_key:
self._url_params[self.payload_key] = payload
res = requests.get(self.__base_url, params=self._url_params)
soup = BeautifulSoup(res.text, 'html.parser')
element = soup.find(self._element_tag, self._element_query)
if not element:
raise ElementNotFoundInGetRequest(element)
return element.get_text(strip=True)
except Exception as e:
raise Exception(str(e.args))
if __name__ == '__main__':
res = GoogleTranslator(source="auto", target="de").translate(payload='this is a good day')
print(res)
|