105 lines
3.1 KiB
Python
105 lines
3.1 KiB
Python
from dataclasses import dataclass, field
|
|
from typing import Optional
|
|
|
|
from src.utils.df import game_configs, get_str, get_int, get_list_int
|
|
from src.classes.material import Material, materials_by_id
|
|
from src.classes.cultivation import Realm
|
|
|
|
@dataclass
|
|
class Lode:
|
|
"""
|
|
矿脉
|
|
"""
|
|
id: int
|
|
name: str
|
|
desc: str
|
|
realm: Realm
|
|
material_ids: list[int] = field(default_factory=list) # 该矿脉对应的物品IDs
|
|
|
|
# 这些字段将在__post_init__中设置
|
|
materials: list[Material] = field(init=False, default_factory=list) # 该矿脉对应的物品实例
|
|
|
|
def __post_init__(self):
|
|
"""初始化物品实例"""
|
|
for material_id in self.material_ids:
|
|
if material_id in materials_by_id:
|
|
self.materials.append(materials_by_id[material_id])
|
|
|
|
def __hash__(self) -> int:
|
|
return hash(self.id)
|
|
|
|
def __str__(self) -> str:
|
|
return self.name
|
|
|
|
def get_info(self) -> str:
|
|
"""
|
|
获取矿脉的详细信息
|
|
"""
|
|
from src.i18n import t
|
|
# 使用格式化字符串 msgid
|
|
base_info = t("[{name}] ({realm})", name=self.name, realm=str(self.realm))
|
|
info_parts = [base_info, self.desc]
|
|
|
|
if self.materials:
|
|
material_names = [material.name for material in self.materials]
|
|
materials_str = t("comma_separator").join(material_names)
|
|
info_parts.append(t("Drops: {materials}", materials=materials_str))
|
|
|
|
return " - ".join(info_parts)
|
|
|
|
def get_structured_info(self) -> dict:
|
|
materials_info = [material.get_structured_info() for material in self.materials]
|
|
return {
|
|
"id": str(self.id),
|
|
"name": self.name,
|
|
"desc": self.desc,
|
|
"grade": str(self.realm),
|
|
"drops": materials_info,
|
|
"type": "lode"
|
|
}
|
|
|
|
def _load_lodes() -> tuple[dict[int, Lode], dict[str, Lode]]:
|
|
"""从配表加载lode数据"""
|
|
lodes_by_id: dict[int, Lode] = {}
|
|
lodes_by_name: dict[str, Lode] = {}
|
|
|
|
# 检查配置是否存在，避免初始化错误
|
|
if "lode" not in game_configs:
|
|
return {}, {}
|
|
|
|
lode_df = game_configs["lode"]
|
|
for row in lode_df:
|
|
material_ids_list = get_list_int(row, "material_ids")
|
|
|
|
lode = Lode(
|
|
id=get_int(row, "id"),
|
|
name=get_str(row, "name"),
|
|
desc=get_str(row, "desc"),
|
|
realm=Realm.from_id(get_int(row, "stage_id")),
|
|
material_ids=material_ids_list
|
|
)
|
|
lodes_by_id[lode.id] = lode
|
|
lodes_by_name[lode.name] = lode
|
|
|
|
return lodes_by_id, lodes_by_name
|
|
|
|
lodes_by_id: dict[int, Lode] = {}
|
|
lodes_by_name: dict[str, Lode] = {}
|
|
|
|
def reload():
|
|
"""重新加载数据，保留全局字典引用"""
|
|
new_id, new_name = _load_lodes()
|
|
|
|
lodes_by_id.clear()
|
|
lodes_by_id.update(new_id)
|
|
|
|
lodes_by_name.clear()
|
|
lodes_by_name.update(new_name)
|
|
|
|
# 更新 ORE_MATERIAL_IDS
|
|
global ORE_MATERIAL_IDS
|
|
ORE_MATERIAL_IDS = {material_id for lode in lodes_by_id.values() for material_id in lode.material_ids}
|
|
|
|
# 模块初始化时执行一次
|
|
reload()
|