import requests as r
from conf.settings.base import YANDEX_DICT_API_KEY
from itertools import chain
from typing import List


def translate_ru_en(word: str) -> List[str]:
    res = r.get(f"https://dictionary.yandex.net/api/v1/dicservice.json/lookup?key={YANDEX_DICT_API_KEY}&lang=ru-en&text={word}")
    return [i['text'] for i in chain(*[j['tr']for j in res.json()['def']])]

def translate_en_ru(word: str) -> List[str]:
    res = r.get(f"https://dictionary.yandex.net/api/v1/dicservice.json/lookup?key={YANDEX_DICT_API_KEY}&lang=en-ru&text={word}")
    return [i['text'] for i in chain(*[j['tr']for j in res.json()['def']])]