Translation.py

from typing import Self, Set


class Translation:

    def __init__(self, việt: str, eng: str, category: str) -> None:
        if not (việt and eng and category):
            raise ValueError(f'args:[{việt}, {eng}, {category}] cannot be have falsy.')
        struct_pattern: str = '...'

        self.việt: str = việt
        self.engTranslations: Set[str] = {eng}
        self.categories: Set[str] = {category}
        self.isStructure: bool = struct_pattern in việt

    def add(self, other: Self) -> None:
        if self != other:
            raise ValueError(f"Cannot add translations: ${self.việt} + {other.việt}")

        self.engTranslations |= other.engTranslations
        self.categories |= other.categories

    def __str__(self) -> str:
        return f"\n{self.việt} {self.categories}: {self.engTranslations}"

    def __repr__(self) -> str:
        return self.__str__()

    def __eq__(self, other: object):
        if isinstance(other, Translation):
            return self.việt == other.việt
        return False

    def add_eng(self, eng: str) -> None:
        self.engTranslations.add(eng)

    def add_category(self, category: str) -> None:
        self.categories.add(category)