72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
from __future__ import annotations
|
|
from typing import Dict, List, TYPE_CHECKING
|
|
import copy
|
|
|
|
if TYPE_CHECKING:
|
|
from src.classes.weapon import Weapon
|
|
from src.classes.auxiliary import Auxiliary
|
|
|
|
|
|
class CirculationManager:
|
|
"""
|
|
出世物品流通管理器
|
|
记录所有从角色身上流出的贵重物品（出售、死亡掉落且未被夺取等）
|
|
用于后续拍卖会等玩法的物品池
|
|
"""
|
|
|
|
def __init__(self):
|
|
# 存储被卖出的法宝
|
|
self.sold_weapons: List[Weapon] = []
|
|
# 存储被卖出的宝物
|
|
self.sold_auxiliaries: List[Auxiliary] = []
|
|
|
|
def add_weapon(self, weapon: "Weapon") -> None:
|
|
"""记录一件流出的兵器"""
|
|
if weapon is None:
|
|
return
|
|
# 使用深拷贝存储，防止外部修改影响记录
|
|
# 注意：这里假设 weapon 对象是可以被 copy 的
|
|
self.sold_weapons.append(weapon.instantiate())
|
|
|
|
def add_auxiliary(self, auxiliary: "Auxiliary") -> None:
|
|
"""记录一件流出的辅助装备"""
|
|
if auxiliary is None:
|
|
return
|
|
self.sold_auxiliaries.append(auxiliary.instantiate())
|
|
|
|
def to_save_dict(self) -> dict:
|
|
"""序列化为字典以便存档"""
|
|
return {
|
|
"weapons": [self._item_to_dict(w) for w in self.sold_weapons],
|
|
"auxiliaries": [self._item_to_dict(a) for a in self.sold_auxiliaries]
|
|
}
|
|
|
|
def load_from_dict(self, data: dict) -> None:
|
|
"""从字典恢复数据"""
|
|
from src.classes.weapon import weapons_by_id
|
|
from src.classes.auxiliary import auxiliaries_by_id
|
|
|
|
self.sold_weapons = []
|
|
for w_data in data.get("weapons", []):
|
|
w_id = w_data.get("id")
|
|
if w_id in weapons_by_id:
|
|
weapon = copy.copy(weapons_by_id[w_id])
|
|
weapon.special_data = w_data.get("special_data", {})
|
|
self.sold_weapons.append(weapon)
|
|
|
|
self.sold_auxiliaries = []
|
|
for a_data in data.get("auxiliaries", []):
|
|
a_id = a_data.get("id")
|
|
if a_id in auxiliaries_by_id:
|
|
auxiliary = copy.copy(auxiliaries_by_id[a_id])
|
|
auxiliary.special_data = a_data.get("special_data", {})
|
|
self.sold_auxiliaries.append(auxiliary)
|
|
|
|
def _item_to_dict(self, item) -> dict:
|
|
"""将物品对象转换为简略的存储格式"""
|
|
return {
|
|
"id": item.id,
|
|
"special_data": getattr(item, "special_data", {})
|
|
}
|
|
|