108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
from __future__ import annotations
|
||
|
||
from dataclasses import dataclass, field
|
||
from typing import Optional, Dict
|
||
|
||
from src.utils.df import game_configs, get_str, get_int
|
||
from src.classes.effect import load_effect_from_str
|
||
from src.classes.equipment_grade import EquipmentGrade
|
||
from src.classes.sect import Sect, sects_by_id
|
||
|
||
|
||
@dataclass
|
||
class Auxiliary:
|
||
"""
|
||
辅助装备类：提供各种辅助功能的装备
|
||
字段与 static/game_configs/auxiliary.csv 对应：
|
||
- grade: 装备等级（普通、宝物、法宝）
|
||
- sect_id: 对应宗门ID（见 sect.csv）；允许为空表示无特定宗门归属
|
||
- effects: 解析为 dict，用于与 Avatar.effects 合并
|
||
"""
|
||
id: int
|
||
name: str
|
||
grade: EquipmentGrade
|
||
sect_id: Optional[int]
|
||
desc: str
|
||
effects: dict[str, object] = field(default_factory=dict)
|
||
effect_desc: str = ""
|
||
sect: Optional[Sect] = None
|
||
# 特殊属性（用于存储实例特定数据）
|
||
special_data: dict = field(default_factory=dict)
|
||
|
||
def get_info(self) -> str:
|
||
"""获取简略信息"""
|
||
return f"{self.name}"
|
||
|
||
def get_detailed_info(self) -> str:
|
||
"""获取详细信息"""
|
||
effect_part = f" 效果：{self.effect_desc}" if self.effect_desc else ""
|
||
return f"{self.name}（{self.grade}，{self.desc}）{effect_part}"
|
||
|
||
def get_colored_info(self) -> str:
|
||
"""获取带颜色标记的信息，供前端渲染使用"""
|
||
r, g, b = self.grade.color_rgb
|
||
return f"<color:{r},{g},{b}>{self.get_info()}</color>"
|
||
|
||
def get_structured_info(self) -> dict:
|
||
return {
|
||
"name": self.name,
|
||
"desc": self.desc,
|
||
"grade": self.grade.value,
|
||
"color": self.grade.color_rgb,
|
||
"effect_desc": self.effect_desc,
|
||
}
|
||
|
||
|
||
def _load_auxiliaries() -> tuple[Dict[int, Auxiliary], Dict[str, Auxiliary], Dict[int, Auxiliary]]:
|
||
"""从配表加载 auxiliary 数据。
|
||
返回：(按ID、按名称、按宗门ID 的映射)。
|
||
若同一宗门配置多个辅助装备，按首次出现保留（每门至多一个法宝级）。
|
||
"""
|
||
auxiliaries_by_id: Dict[int, Auxiliary] = {}
|
||
auxiliaries_by_name: Dict[str, Auxiliary] = {}
|
||
auxiliaries_by_sect_id: Dict[int, Auxiliary] = {}
|
||
|
||
df = game_configs.get("auxiliary")
|
||
if df is None:
|
||
return auxiliaries_by_id, auxiliaries_by_name, auxiliaries_by_sect_id
|
||
|
||
for row in df:
|
||
sect_id = get_int(row, "sect_id", -1)
|
||
if sect_id == -1:
|
||
sect_id = None
|
||
|
||
effects = load_effect_from_str(get_str(row, "effects"))
|
||
from src.utils.effect_desc import format_effects_to_text
|
||
effect_desc = format_effects_to_text(effects)
|
||
|
||
sect_obj: Optional[Sect] = sects_by_id.get(sect_id) if sect_id is not None else None
|
||
|
||
# 解析grade
|
||
grade_str = get_str(row, "grade", "普通")
|
||
grade = EquipmentGrade.COMMON
|
||
for g in EquipmentGrade:
|
||
if g.value == grade_str:
|
||
grade = g
|
||
break
|
||
|
||
a = Auxiliary(
|
||
id=get_int(row, "id"),
|
||
name=get_str(row, "name"),
|
||
grade=grade,
|
||
sect_id=sect_id,
|
||
desc=get_str(row, "desc"),
|
||
effects=effects,
|
||
effect_desc=effect_desc,
|
||
sect=sect_obj,
|
||
)
|
||
|
||
auxiliaries_by_id[a.id] = a
|
||
auxiliaries_by_name[a.name] = a
|
||
if a.sect_id is not None and a.sect_id not in auxiliaries_by_sect_id:
|
||
auxiliaries_by_sect_id[a.sect_id] = a
|
||
|
||
return auxiliaries_by_id, auxiliaries_by_name, auxiliaries_by_sect_id
|
||
|
||
|
||
auxiliaries_by_id, auxiliaries_by_name, auxiliaries_by_sect_id = _load_auxiliaries()
|