350 lines
12 KiB
Python
350 lines
12 KiB
Python
"""
|
||
卡牌YAML加载器
|
||
从YAML文件加载卡牌定义并创建卡牌对象
|
||
"""
|
||
import yaml
|
||
from typing import List, Dict, Optional, Set, Any
|
||
from pathlib import Path
|
||
from ..core.enums import Nation, Rarity
|
||
from .card import UnitCard, CardStats
|
||
from .card_params import ParameterDefinition
|
||
|
||
|
||
class CardLoadError(Exception):
|
||
"""卡牌加载错误"""
|
||
pass
|
||
|
||
|
||
class CardLoader:
|
||
"""卡牌YAML加载器"""
|
||
|
||
def __init__(self, cards_dir: Optional[str] = None):
|
||
"""
|
||
初始化卡牌加载器
|
||
|
||
Args:
|
||
cards_dir: 卡牌文件目录，默认为 assets/cards/
|
||
"""
|
||
if cards_dir is None:
|
||
# 从当前文件位置推断assets目录
|
||
current_dir = Path(__file__).parent
|
||
project_root = current_dir.parent.parent # kards_battle -> kards
|
||
self.cards_dir = project_root / "assets" / "cards"
|
||
else:
|
||
self.cards_dir = Path(cards_dir)
|
||
|
||
# 缓存模板定义
|
||
self._templates = {}
|
||
self._templates_loaded = False
|
||
|
||
def _parse_nation(self, nation_str: str) -> Nation:
|
||
"""将字符串转换为Nation枚举"""
|
||
nation_map = {
|
||
'germany': Nation.GERMANY,
|
||
'italy': Nation.ITALY,
|
||
'finland': Nation.FINLAND,
|
||
'japan': Nation.JAPAN,
|
||
'usa': Nation.USA,
|
||
'uk': Nation.UK,
|
||
'soviet': Nation.SOVIET,
|
||
'france': Nation.FRANCE,
|
||
'poland': Nation.POLAND,
|
||
}
|
||
|
||
nation_lower = nation_str.lower()
|
||
if nation_lower not in nation_map:
|
||
raise CardLoadError(f"未知的国家: {nation_str}")
|
||
|
||
return nation_map[nation_lower]
|
||
|
||
def _parse_rarity(self, rarity: int) -> Rarity:
|
||
"""将整数转换为Rarity枚举"""
|
||
rarity_map = {
|
||
1: Rarity.ELITE,
|
||
2: Rarity.RARE,
|
||
3: Rarity.UNCOMMON,
|
||
4: Rarity.COMMON
|
||
}
|
||
|
||
if rarity not in rarity_map:
|
||
raise CardLoadError(f"无效的稀有度: {rarity}，必须是1-4")
|
||
|
||
return rarity_map[rarity]
|
||
|
||
def _parse_exile_nations(self, exile_nations: Optional[List[str]]) -> Optional[Set[Nation]]:
|
||
"""解析流亡国家列表"""
|
||
if not exile_nations:
|
||
return None
|
||
|
||
result = set()
|
||
for nation_str in exile_nations:
|
||
result.add(self._parse_nation(nation_str))
|
||
|
||
return result
|
||
|
||
def _load_yaml_file(self, file_path: Path) -> Dict[str, Any]:
|
||
"""加载YAML文件"""
|
||
try:
|
||
with open(file_path, 'r', encoding='utf-8') as f:
|
||
return yaml.safe_load(f)
|
||
except FileNotFoundError:
|
||
raise CardLoadError(f"卡牌文件不存在: {file_path}")
|
||
except yaml.YAMLError as e:
|
||
raise CardLoadError(f"YAML解析错误 ({file_path}): {e}")
|
||
|
||
def _parse_card_parameters(self, params_data: Optional[List[Dict[str, Any]]]) -> List[ParameterDefinition]:
|
||
"""解析卡牌参数定义"""
|
||
if not params_data:
|
||
return []
|
||
|
||
parameters = []
|
||
for param_data in params_data:
|
||
try:
|
||
param_def = ParameterDefinition.from_dict(param_data)
|
||
parameters.append(param_def)
|
||
except Exception as e:
|
||
raise CardLoadError(f"解析参数定义失败: {e}")
|
||
|
||
return parameters
|
||
|
||
def _load_templates(self):
|
||
"""加载卡牌模板定义"""
|
||
if self._templates_loaded:
|
||
return
|
||
|
||
spec_file = self.cards_dir / "cards_spec.yaml"
|
||
if not spec_file.exists():
|
||
self._templates_loaded = True
|
||
return
|
||
|
||
try:
|
||
spec_data = self._load_yaml_file(spec_file)
|
||
templates = spec_data.get('card_templates', [])
|
||
|
||
for template in templates:
|
||
if 'id' not in template:
|
||
continue
|
||
self._templates[template['id']] = template
|
||
|
||
except Exception as e:
|
||
raise CardLoadError(f"Failed to load templates: {e}")
|
||
|
||
self._templates_loaded = True
|
||
|
||
def _apply_template(self, card_data: Dict[str, Any]) -> Dict[str, Any]:
|
||
"""应用模板到卡牌数据"""
|
||
self._load_templates()
|
||
|
||
template_id = card_data.get('template')
|
||
if not template_id:
|
||
return card_data
|
||
|
||
if template_id not in self._templates:
|
||
raise CardLoadError(f"Template not found: {template_id}")
|
||
|
||
template = self._templates[template_id]
|
||
|
||
# 创建合并后的卡牌数据
|
||
merged_data = dict(card_data) # 复制原始数据
|
||
|
||
# 合并模板字段（卡牌数据优先）
|
||
for key, value in template.items():
|
||
if key == 'id': # 跳过模板ID
|
||
continue
|
||
if key not in merged_data:
|
||
merged_data[key] = value
|
||
elif key == 'params':
|
||
# 参数需要特殊合并逻辑
|
||
template_params = template.get('params', [])
|
||
card_params = card_data.get('params', [])
|
||
merged_data['params'] = template_params + card_params
|
||
|
||
return merged_data
|
||
|
||
def _create_unit_card(self, card_data: Dict[str, Any]) -> UnitCard:
|
||
"""从数据字典创建单位卡"""
|
||
# 应用模板
|
||
card_data = self._apply_template(card_data)
|
||
|
||
required_fields = ['id', 'name', 'type', 'cost', 'nation', 'rarity', 'unit_definition_id']
|
||
for field in required_fields:
|
||
if field not in card_data:
|
||
raise CardLoadError(f"单位卡缺少必需字段: {field}")
|
||
|
||
# 验证卡牌类型
|
||
if card_data['type'] != 'unit':
|
||
raise CardLoadError(f"期望单位卡，但得到: {card_data['type']}")
|
||
|
||
# 解析基础属性
|
||
nation = self._parse_nation(card_data['nation'])
|
||
rarity = self._parse_rarity(card_data['rarity'])
|
||
exile_nations = self._parse_exile_nations(card_data.get('exile_nations'))
|
||
|
||
# 解析参数定义
|
||
parameters = self._parse_card_parameters(card_data.get('params'))
|
||
|
||
# 创建单位卡
|
||
card = UnitCard(
|
||
name=card_data['name'],
|
||
cost=card_data['cost'],
|
||
nation=nation,
|
||
rarity=rarity,
|
||
unit_definition_id=card_data['unit_definition_id'],
|
||
exile_nations=exile_nations,
|
||
description=card_data.get('description', ''),
|
||
card_id=card_data['id'],
|
||
parameters=parameters
|
||
)
|
||
|
||
return card
|
||
|
||
def load_unit_cards_from_file(self, file_path: str) -> List[UnitCard]:
|
||
"""
|
||
从YAML文件加载单位卡
|
||
|
||
Args:
|
||
file_path: YAML文件路径（相对于cards_dir）
|
||
|
||
Returns:
|
||
单位卡列表
|
||
"""
|
||
full_path = self.cards_dir / file_path
|
||
data = self._load_yaml_file(full_path)
|
||
|
||
if 'unit_cards' not in data:
|
||
raise CardLoadError(f"YAML文件中找不到 'unit_cards' 节点: {file_path}")
|
||
|
||
cards = []
|
||
for card_data in data['unit_cards']:
|
||
try:
|
||
card = self._create_unit_card(card_data)
|
||
cards.append(card)
|
||
except Exception as e:
|
||
raise CardLoadError(f"创建卡牌失败 (文件: {file_path}, ID: {card_data.get('id', 'unknown')}): {e}")
|
||
|
||
return cards
|
||
|
||
def load_nation_cards(self, nation: Nation) -> List[UnitCard]:
|
||
"""
|
||
加载指定国家的所有卡牌
|
||
|
||
Args:
|
||
nation: 国家
|
||
|
||
Returns:
|
||
该国家的所有卡牌
|
||
"""
|
||
nation_file_map = {
|
||
Nation.GERMANY: 'germany.yaml',
|
||
Nation.ITALY: 'italy.yaml',
|
||
Nation.USA: 'usa.yaml',
|
||
Nation.UK: 'uk.yaml',
|
||
Nation.SOVIET: 'soviet.yaml',
|
||
Nation.JAPAN: 'japan.yaml',
|
||
Nation.FRANCE: 'france.yaml',
|
||
Nation.POLAND: 'poland.yaml',
|
||
Nation.FINLAND: 'finland.yaml',
|
||
}
|
||
|
||
if nation not in nation_file_map:
|
||
raise CardLoadError(f"不支持的国家: {nation}")
|
||
|
||
filename = nation_file_map[nation]
|
||
|
||
# 检查文件是否存在
|
||
file_path = self.cards_dir / filename
|
||
if not file_path.exists():
|
||
# 如果文件不存在，返回空列表（某些国家可能还没有卡牌）
|
||
return []
|
||
|
||
return self.load_unit_cards_from_file(filename)
|
||
|
||
def load_all_available_cards(self) -> Dict[Nation, List[UnitCard]]:
|
||
"""
|
||
加载所有可用的卡牌
|
||
|
||
Returns:
|
||
按国家分组的卡牌字典
|
||
"""
|
||
all_cards = {}
|
||
|
||
# 遍历所有支持的国家
|
||
for nation in Nation:
|
||
try:
|
||
cards = self.load_nation_cards(nation)
|
||
if cards: # 只添加有卡牌的国家
|
||
all_cards[nation] = cards
|
||
except CardLoadError:
|
||
# 如果某个国家的卡牌加载失败，跳过但记录
|
||
continue
|
||
|
||
return all_cards
|
||
|
||
def validate_card_data(self, card_data: Dict[str, Any]) -> List[str]:
|
||
"""
|
||
验证卡牌数据的完整性
|
||
|
||
Args:
|
||
card_data: 卡牌数据字典
|
||
|
||
Returns:
|
||
错误信息列表，空列表表示验证通过
|
||
"""
|
||
errors = []
|
||
|
||
# 检查必需字段
|
||
required_fields = ['id', 'name', 'type', 'cost', 'nation', 'rarity']
|
||
for field in required_fields:
|
||
if field not in card_data:
|
||
errors.append(f"缺少必需字段: {field}")
|
||
|
||
# 检查卡牌类型
|
||
if 'type' in card_data:
|
||
valid_types = ['unit', 'order', 'countermeasure']
|
||
if card_data['type'] not in valid_types:
|
||
errors.append(f"无效的卡牌类型: {card_data['type']}")
|
||
|
||
# 检查费用
|
||
if 'cost' in card_data:
|
||
if not isinstance(card_data['cost'], int) or card_data['cost'] < 0:
|
||
errors.append(f"费用必须是非负整数: {card_data['cost']}")
|
||
|
||
# 检查稀有度
|
||
if 'rarity' in card_data:
|
||
if not isinstance(card_data['rarity'], int) or card_data['rarity'] < 1 or card_data['rarity'] > 4:
|
||
errors.append(f"稀有度必须是1-4的整数: {card_data['rarity']}")
|
||
|
||
# 检查国家
|
||
if 'nation' in card_data:
|
||
try:
|
||
self._parse_nation(card_data['nation'])
|
||
except CardLoadError as e:
|
||
errors.append(str(e))
|
||
|
||
# 单位卡特殊验证
|
||
if card_data.get('type') == 'unit':
|
||
if 'unit_definition_id' not in card_data:
|
||
errors.append("单位卡必须包含 unit_definition_id 字段")
|
||
|
||
return errors
|
||
|
||
|
||
# 全局加载器实例
|
||
_default_loader = None
|
||
|
||
def get_card_loader() -> CardLoader:
|
||
"""获取默认的卡牌加载器实例"""
|
||
global _default_loader
|
||
if _default_loader is None:
|
||
_default_loader = CardLoader()
|
||
return _default_loader
|
||
|
||
|
||
def load_nation_cards(nation: Nation) -> List[UnitCard]:
|
||
"""便捷函数：加载指定国家的卡牌"""
|
||
return get_card_loader().load_nation_cards(nation)
|
||
|
||
|
||
def load_all_cards() -> Dict[Nation, List[UnitCard]]:
|
||
"""便捷函数：加载所有可用卡牌"""
|
||
return get_card_loader().load_all_available_cards() |