Files
cultivation-world-simulator/src/classes/action/catch.py
2026-01-01 15:08:09 +08:00

99 lines
3.8 KiB
Python

from __future__ import annotations
import random
from typing import TYPE_CHECKING
from src.classes.action import TimedAction
from src.classes.event import Event
from src.classes.region import NormalRegion
from src.classes.spirit_animal import SpiritAnimal
from src.classes.cultivation import Realm
if TYPE_CHECKING:
from src.classes.animal import Animal
class Catch(TimedAction):
"""
御兽:仅百兽宗弟子可用。
条件:
- 当前处于普通区域,且该区域有动物分布
- 目标动物境界 <= Avatar 境界
结果:
- 按动物境界映射成功率尝试捕捉,成功则成为灵兽(覆盖旧灵兽)。
"""
ACTION_NAME = "御兽"
EMOJI = "🕸️"
DESC = "尝试驯服一只灵兽,成为自身灵兽。只能有一只灵兽,但是可以高级替换低级。"
DOABLES_REQUIREMENTS = "仅百兽宗;在有动物的普通区域;目标动物境界不高于角色"
PARAMS = {}
duration_months = 4
def __init__(self, avatar, world):
super().__init__(avatar, world)
self._caught_result: tuple[str, Realm] | None = None
def _calc_success_rate_by_realm(self, animal_realm: Realm) -> float:
mapping: dict[Realm, float] = {
Realm.Qi_Refinement: 0.8,
Realm.Foundation_Establishment: 0.6,
Realm.Core_Formation: 0.4,
Realm.Nascent_Soul: 0.2,
}
return mapping.get(animal_realm, 0.1)
def _execute(self) -> None:
region = self.avatar.tile.region
animals = region.animals
if not animals:
return
# 若已成功捕捉过一次,本次动作内不再重复尝试
if self._caught_result is not None:
return
target = random.choice(animals)
base = self._calc_success_rate_by_realm(target.realm)
extra = float(self.avatar.effects.get("extra_catch_success_rate", 0) or 0)
rate = max(0.0, min(1.0, base + extra))
if random.random() < rate:
# 覆盖为新的灵兽
self.avatar.spirit_animal = SpiritAnimal(name=target.name, realm=target.realm)
# 记录结果供 finish 生成事件
self._caught_result = (str(target.name), target.realm, "success")
else:
self._caught_result = (None, None, "fail")
def can_start(self) -> tuple[bool, str]:
region = self.avatar.tile.region
if not isinstance(region, NormalRegion):
return False, "当前不在普通区域"
animals = region.animals
if len(animals) == 0:
return False, f"当前区域{region.name}没有动物"
# 动物境界是否可御
available_animals = [animal for animal in animals if self.avatar.cultivation_progress.realm >= animal.realm]
if len(available_animals) == 0:
return False, "当前区域的动物境界于角色境界"
return True, ""
def start(self) -> Event:
# 清理状态
self._caught_result = None
region = self.avatar.tile.region
return Event(self.world.month_stamp, f"{self.avatar.name}{self.avatar.tile.location_name} 尝试御兽", related_avatars=[self.avatar.id])
async def finish(self) -> list[Event]:
res = self._caught_result
if not (isinstance(res, tuple) and len(res) == 3):
return []
target_name, target_realm, result = res[0], res[1], res[2]
if result == "fail":
return [Event(self.world.month_stamp, f"{self.avatar.name} 御兽失败", related_avatars=[self.avatar.id])]
else:
realm_label = target_realm.value
text = f"{self.avatar.name} 御兽成功,{realm_label}境的{target_name}成为其灵兽"
return [Event(self.world.month_stamp, text, related_avatars=[self.avatar.id])]