StarRailCopilot/tasks/rogue/keywords/classes.py

from dataclasses import dataclass
from typing import ClassVar

from dev_tools.keyword_extract import UI_LANGUAGES
from module.exception import ScriptError
from module.ocr.keyword import Keyword


@dataclass(repr=False)
class RogueBlessing(Keyword):
    instances: ClassVar = {}
    path_id: int
    rarity: int
    enhancement: str

    @property
    def path_name(self):
        path = RoguePath.instances[self.path_id]
        return path.path_name

    @property
    def blessing_name(self):
        return [self.__getattribute__(f"{server}_parsed")
                for server in UI_LANGUAGES if hasattr(self, f"{server}_parsed")]


@dataclass(repr=False)
class RoguePath(Keyword):
    instances: ClassVar = {}

    def __hash__(self) -> int:
        return super().__hash__()

    @property
    def path_name(self):
        return [self.__getattribute__(f"{server}_parsed").replace("the", '')
                for server in UI_LANGUAGES if hasattr(self, f"{server}_parsed")]

    @classmethod
    def find_path(cls, name):
        if isinstance(name, RoguePath):
            return name
        for instance in cls.instances.values():
            if name == instance.name:
                return instance
        # Not found
        raise ScriptError(f'Cannot find a {cls.__name__} instance that matches "{name}"')


@dataclass(repr=False)
class RogueResonance(Keyword):
    instances: ClassVar = {}
    path_id: int
    rarity: int

    @property
    def resonance_name(self):
        return [self.__getattribute__(f"{server}_parsed")
                for server in UI_LANGUAGES if hasattr(self, f"{server}_parsed")]


@dataclass(repr=False)
class RogueCurio(Keyword):
    instances: ClassVar = {}

    @property
    def curio_name(self):
        return [self.__getattribute__(f"{server}_parsed")
                for server in UI_LANGUAGES if hasattr(self, f"{server}_parsed")]


@dataclass(repr=False)
class RogueBonus(Keyword):
    instances: ClassVar = {}


@dataclass(repr=False)
class RogueEnhancement(Keyword):
    instances: ClassVar = {}

    @property
    def enhancement_keyword(self):
        return [self.__getattribute__(f"{server}_parsed")
                for server in UI_LANGUAGES if hasattr(self, f"{server}_parsed")]


@dataclass(repr=False)
class RogueEventTitle(Keyword):
    instances: ClassVar = {}
    option_ids: list[int]

    def __hash__(self):
        return super().__hash__()


@dataclass(repr=False)
class RogueEventOption(Keyword):
    instances: ClassVar = {}