from deep_translator import GoogleTranslator
from typing import Dict, Any, List
import time


class ProductTranslator:
    def __init__(self):
        self.translator = GoogleTranslator(source="pl", target="uk")

    def translate_text(self, text: str) -> str:
        """Переводит текст с обработкой ошибок и задержкой"""
        if not text or not isinstance(text, str):
            return text

        try:
            translated = self.translator.translate(text)
            time.sleep(0.5)  # Задержка чтобы избежать блокировки
            return translated
        except Exception as e:
            print(f"Ошибка перевода: {e}")
            return text

    def translate_list(self, items: List[str]) -> List[str]:
        """Переводит список строк"""
        return [self.translate_text(item) for item in items]

    def is_translated(self, original: str, translated: str) -> bool:
        """Проверяет, был ли текст действительно переведен"""
        return original.strip() != translated.strip()

    def translate_product(self, product: Dict[str, Any]) -> Dict[str, Any] | None:
        translated = product.copy()
        any_changes = False

        # Название
        translated_name = self.translate_text(product["name"])
        if self.is_translated(product["name"], translated_name):
            translated["name"] = translated_name
            any_changes = True
        else:
            print(f"[SKIP] Название не переведено: {product['name']}")

        # Атрибути
        if "attributes" in translated:
            for attr in translated["attributes"]:
                name_trans = self.translate_text(attr["name"])
                if self.is_translated(attr["name"], name_trans):
                    attr["name"] = name_trans
                    any_changes = True

                values_trans = self.translate_list(attr["value"])
                if values_trans != attr["value"]:
                    attr["value"] = values_trans
                    any_changes = True

        # Опис
        if "description" in translated:
            for section in translated["description"]:
                title_trans = self.translate_text(section["title"])
                text_trans = self.translate_text(section["text"])

                if self.is_translated(section["title"], title_trans):
                    section["title"] = title_trans
                    any_changes = True
                if self.is_translated(section["text"], text_trans):
                    section["text"] = text_trans
                    any_changes = True

        if any_changes:
            return translated
        else:
            print(f"[SKIP] Товар не содержит переведенных полей: {product['name']}")
            return None