32 lines
977 B
Python
32 lines
977 B
Python
|
'''
|
||
|
|
This file provides the functionality for
|
||
|
|
the translation function, by first converting the
|
||
|
|
country/language name to an ISO369-1 language identifier
|
||
|
|
and then translating it using the Yandex Translation API
|
||
|
|
'''
|
||
|
|
import requests, urllib.parse
|
||
|
|
from util import logging, config
|
||
|
|
import pycountry
|
||
|
|
|
||
|
|
api_key = config.settings["api_keys"]["yandex"]
|
||
|
|
langs = pycountry.languages
|
||
|
|
|
||
|
|
def name_to_ISO(name: str) -> str:
|
||
|
|
alpha = ""
|
||
|
|
for language in langs:
|
||
|
|
if language.name.lower() == name.lower():
|
||
|
|
alpha = language.alpha_2
|
||
|
|
break
|
||
|
|
|
||
|
|
return alpha
|
||
|
|
|
||
|
|
def translate(text: str, lang: str) -> str:
|
||
|
|
url_encoded_text = urllib.parse.quote(text)
|
||
|
|
url = f"https://translate.yandex.net/api/v1.5/tr.json/translate?key={api_key}&text={url_encoded_text}&lang={lang}"
|
||
|
|
|
||
|
|
response = requests.get(url)
|
||
|
|
if not response.ok:
|
||
|
|
logging.error(f"Failed to contact Yandex API: {response.status_code}")
|
||
|
|
return ""
|
||
|
|
|
||
|
|
return response.json()["text"][0]
|