103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
from __future__ import annotations
|
||
|
||
from src.classes.action import InstantAction
|
||
from src.classes.event import Event
|
||
from src.classes.weapon import get_random_weapon_by_realm
|
||
from src.classes.weapon_type import WeaponType
|
||
from src.classes.cultivation import Realm
|
||
from src.classes.normalize import normalize_weapon_type
|
||
|
||
|
||
class SwitchWeapon(InstantAction):
|
||
"""
|
||
切换兵器：将当前兵器切换为指定类型的练气兵器。
|
||
熟练度重置为0。
|
||
"""
|
||
|
||
ACTION_NAME = "切换兵器"
|
||
EMOJI = "🔄"
|
||
DESC = "切换到指定类型的练气兵器，或卸下兵器。当前兵器会丧失，熟练度会重置为0。适用于想要更换兵器类型或从头修炼新兵器的情况。"
|
||
DOABLES_REQUIREMENTS = "无前置条件"
|
||
PARAMS = {"weapon_type_name": "str"}
|
||
|
||
def _execute(self, weapon_type_name: str) -> None:
|
||
# 处理卸下兵器的情况
|
||
if weapon_type_name in ["无", "None", "none", ""]:
|
||
self.avatar.change_weapon(None)
|
||
return
|
||
|
||
# 规范化兵器类型名称
|
||
normalized_type = normalize_weapon_type(weapon_type_name)
|
||
|
||
# 匹配 WeaponType 枚举
|
||
target_weapon_type = None
|
||
for wt in WeaponType:
|
||
if wt.value == normalized_type:
|
||
target_weapon_type = wt
|
||
break
|
||
|
||
if target_weapon_type is None:
|
||
return
|
||
|
||
# 获取练气兵器（练气期）
|
||
common_weapon = get_random_weapon_by_realm(Realm.Qi_Refinement, target_weapon_type)
|
||
if common_weapon is None:
|
||
return
|
||
|
||
# 切换兵器（使用 Avatar 的 change_weapon 方法）
|
||
self.avatar.change_weapon(common_weapon)
|
||
|
||
def can_start(self, weapon_type_name: str | None = None) -> tuple[bool, str]:
|
||
if weapon_type_name is None:
|
||
# AI调用：总是可以切换兵器
|
||
return True, ""
|
||
|
||
# 处理卸下兵器的情况
|
||
if weapon_type_name in ["无", "None", "none", ""]:
|
||
if self.avatar.weapon is None:
|
||
return False, "当前已处于无兵器状态"
|
||
return True, ""
|
||
|
||
# 规范化并验证兵器类型
|
||
normalized_type = normalize_weapon_type(weapon_type_name)
|
||
target_weapon_type = None
|
||
for wt in WeaponType:
|
||
if wt.value == normalized_type:
|
||
target_weapon_type = wt
|
||
break
|
||
|
||
if target_weapon_type is None:
|
||
return False, f"未知兵器类型: {weapon_type_name}（支持的类型：剑/刀/枪/棍/扇/鞭/琴/笛/暗器/无）"
|
||
|
||
# 检查是否已经是该类型的练气兵器
|
||
if self.avatar.weapon is not None and \
|
||
self.avatar.weapon.weapon_type == target_weapon_type and \
|
||
self.avatar.weapon.realm == Realm.Qi_Refinement:
|
||
return False, f"已经装备了基础{target_weapon_type.value}"
|
||
|
||
# 检查练气兵器是否存在
|
||
common_weapon = get_random_weapon_by_realm(Realm.Qi_Refinement, target_weapon_type)
|
||
if common_weapon is None:
|
||
return False, f"系统中不存在练气{target_weapon_type.value}"
|
||
|
||
return True, ""
|
||
|
||
def start(self, weapon_type_name: str) -> Event:
|
||
if weapon_type_name in ["无", "None", "none", ""]:
|
||
return Event(
|
||
self.world.month_stamp,
|
||
f"{self.avatar.name} 卸下了兵器",
|
||
related_avatars=[self.avatar.id]
|
||
)
|
||
|
||
normalized_type = normalize_weapon_type(weapon_type_name)
|
||
return Event(
|
||
self.world.month_stamp,
|
||
f"{self.avatar.name} 切换兵器为练气{normalized_type}",
|
||
related_avatars=[self.avatar.id]
|
||
)
|
||
|
||
async def finish(self, weapon_type_name: str) -> list[Event]:
|
||
return []
|
||
|