80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
from __future__ import annotations
|
||
from typing import TYPE_CHECKING
|
||
import random
|
||
|
||
from src.classes.equipment_grade import EquipmentGrade
|
||
from src.classes.single_choice import make_decision
|
||
|
||
if TYPE_CHECKING:
|
||
from src.classes.avatar import Avatar
|
||
|
||
|
||
async def kill_and_grab(winner: Avatar, loser: Avatar) -> str:
|
||
"""
|
||
处理杀人夺宝逻辑
|
||
|
||
Args:
|
||
winner: 胜利者
|
||
loser: 失败者（已死亡）
|
||
|
||
Returns:
|
||
str: 夺宝结果描述文本（如"并夺取了..."），如果没有夺取则为空字符串
|
||
"""
|
||
loot_candidates = []
|
||
|
||
# 检查兵器
|
||
if loser.weapon and loser.weapon.grade != EquipmentGrade.COMMON:
|
||
loot_candidates.append(("weapon", loser.weapon))
|
||
|
||
# 检查辅助装备
|
||
if loser.auxiliary and loser.auxiliary.grade != EquipmentGrade.COMMON:
|
||
loot_candidates.append(("auxiliary", loser.auxiliary))
|
||
|
||
if not loot_candidates:
|
||
return ""
|
||
|
||
# 优先选法宝，其次宝物；如果有多个同级，随机选一个
|
||
loot_candidates.sort(key=lambda x: 1 if x[1].grade == EquipmentGrade.ARTIFACT else 0, reverse=True)
|
||
# 筛选出最高优先级的那些
|
||
best_grade = loot_candidates[0][1].grade
|
||
best_candidates = [c for c in loot_candidates if c[1].grade == best_grade]
|
||
loot_type, loot_item = random.choice(best_candidates)
|
||
|
||
should_loot = False
|
||
|
||
# 判定是否夺取
|
||
# 1. 如果winner当前部位为空或为凡品，直接夺取
|
||
winner_current = getattr(winner, loot_type)
|
||
if winner_current is None or winner_current.grade == EquipmentGrade.COMMON:
|
||
should_loot = True
|
||
else:
|
||
# 2. 否则让 AI 决策
|
||
context = f"战斗胜利，{loser.name} 身死道消，留下了一件{loot_item.grade.value}{'兵器' if loot_type == 'weapon' else '辅助装备'}『{loot_item.name}』（{loot_item.desc}）。"
|
||
options = [
|
||
{
|
||
"key": "A",
|
||
"desc": f"夺取{loot_item.grade.value}『{loot_item.name}』（{loot_item.desc}），替换掉身上的『{winner_current.name}』（{winner_current.grade.value}，{winner_current.desc}）。"
|
||
},
|
||
{
|
||
"key": "B",
|
||
"desc": f"放弃『{loot_item.name}』，保留身上的『{winner_current.name}』。"
|
||
}
|
||
]
|
||
choice = await make_decision(winner, context, options)
|
||
if choice == "A":
|
||
should_loot = True
|
||
|
||
if should_loot:
|
||
if loot_type == "weapon":
|
||
winner.change_weapon(loot_item)
|
||
from src.classes.weapon import get_common_weapon
|
||
loser.change_weapon(get_common_weapon(loot_item.weapon_type)) # 给死者塞个凡品防止空指针
|
||
else:
|
||
winner.change_auxiliary(loot_item)
|
||
loser.change_auxiliary(None)
|
||
|
||
return f"并夺取了对方的{loot_item.grade.value}『{loot_item.name}』！"
|
||
|
||
return ""
|
||
|