diff --git a/models/enums.py b/models/enums.py index 8b63e97..b828326 100644 --- a/models/enums.py +++ b/models/enums.py @@ -142,3 +142,15 @@ class RelicPosition(str, Enum): """ 位面球 """ OBJECT: str = "OBJECT" """ 连结绳 """ + + @property + def num(self): + index_map = { + RelicPosition.HEAD: 0, + RelicPosition.HAND: 1, + RelicPosition.BODY: 2, + RelicPosition.FOOT: 3, + RelicPosition.NECK: 0, + RelicPosition.OBJECT: 1, + } + return index_map.get(self) diff --git a/models/relic_affix.py b/models/relic_affix.py index c9b1c98..ce83575 100644 --- a/models/relic_affix.py +++ b/models/relic_affix.py @@ -37,11 +37,6 @@ class RelicAffixAll(BaseModel): sub_affix: Dict[str, SingleRelicAffix] """ 副词条 """ - @property - def rarity(self) -> int: - rarity_map = {9: 3, 12: 4, 16: 5} - return rarity_map.get(self.max_level, 2) - @root_validator(pre=True) def transform_dicts(cls, values): for data in ["main_affix", "sub_affix"]: