153 lines
4.8 KiB
Python
153 lines
4.8 KiB
Python
from __future__ import annotations
|
||
|
||
import random
|
||
from enum import Enum
|
||
from typing import Optional
|
||
|
||
from src.utils.config import CONFIG
|
||
from src.classes.avatar import Avatar
|
||
from src.classes.event import Event
|
||
from src.classes.story_teller import StoryTeller
|
||
from src.classes.fortune import get_cultivation_exp_reward
|
||
|
||
class MisfortuneKind(Enum):
|
||
"""霉运类型"""
|
||
LOSS_SPIRIT_STONE = "loss_spirit_stone" # 破财
|
||
INJURY = "injury" # 受伤
|
||
CULTIVATION_BACKLASH = "backlash" # 修为倒退
|
||
|
||
|
||
MF_LOSS_SPIRIT_STONE_THEMES: list[str] = [
|
||
"遭遇扒手",
|
||
"误买假货",
|
||
"遭人勒索",
|
||
"洞府失窃",
|
||
"赌石惨败",
|
||
"投资失败",
|
||
]
|
||
|
||
MF_INJURY_THEMES: list[str] = [
|
||
"修炼岔气",
|
||
"出门摔伤",
|
||
"妖兽偷袭",
|
||
"仇家闷棍",
|
||
"误触机关",
|
||
"天降横祸",
|
||
]
|
||
|
||
MF_BACKLASH_THEMES: list[str] = [
|
||
"心魔滋生",
|
||
"灵气逆行",
|
||
"感悟错乱",
|
||
"急火攻心",
|
||
]
|
||
|
||
|
||
def _choose_misfortune_kind(avatar: Avatar) -> Optional[MisfortuneKind]:
|
||
"""选择霉运类型"""
|
||
candidates = []
|
||
|
||
# 破财：必须有灵石
|
||
if avatar.magic_stone.value > 0:
|
||
candidates.append(MisfortuneKind.LOSS_SPIRIT_STONE)
|
||
|
||
# 受伤：任何人都可以受伤
|
||
candidates.append(MisfortuneKind.INJURY)
|
||
|
||
# 修为倒退：只有修炼者（有修为且未满？）或者任何人都可以？
|
||
# 简单处理：任何人都可以走火入魔
|
||
candidates.append(MisfortuneKind.CULTIVATION_BACKLASH)
|
||
|
||
if not candidates:
|
||
return None
|
||
|
||
return random.choice(candidates)
|
||
|
||
|
||
def _pick_misfortune_theme(kind: MisfortuneKind) -> str:
|
||
if kind == MisfortuneKind.LOSS_SPIRIT_STONE:
|
||
return random.choice(MF_LOSS_SPIRIT_STONE_THEMES)
|
||
elif kind == MisfortuneKind.INJURY:
|
||
return random.choice(MF_INJURY_THEMES)
|
||
elif kind == MisfortuneKind.CULTIVATION_BACKLASH:
|
||
return random.choice(MF_BACKLASH_THEMES)
|
||
return ""
|
||
|
||
|
||
async def try_trigger_misfortune(avatar: Avatar) -> list[Event]:
|
||
"""
|
||
触发霉运
|
||
规则：
|
||
- 概率：config + effects
|
||
- 类型：破财、受伤、修为倒退
|
||
- 破财：随机数，不超过总量
|
||
- 受伤：扣减HP，可能致死（由simulator结算）
|
||
- 修为倒退：扣减经验，不降级（经验值可为负？）-> 此处逻辑：扣减当前经验，最小为0
|
||
"""
|
||
base_prob = float(getattr(CONFIG.game, "misfortune_probability", 0.0))
|
||
extra_prob = float(avatar.effects.get("extra_misfortune_probability", 0.0))
|
||
prob = base_prob + extra_prob
|
||
if prob <= 0.0:
|
||
return []
|
||
|
||
if random.random() >= prob:
|
||
return []
|
||
|
||
kind = _choose_misfortune_kind(avatar)
|
||
if kind is None:
|
||
return []
|
||
|
||
theme = _pick_misfortune_theme(kind)
|
||
res_text: str = ""
|
||
|
||
if kind == MisfortuneKind.LOSS_SPIRIT_STONE:
|
||
# 破财：随机数，不超过总量
|
||
max_loss = avatar.magic_stone.value
|
||
# 设定一个随机范围，例如 10~500，但受 max_loss 限制
|
||
# 或者完全随机
|
||
loss = random.randint(50, 300)
|
||
loss = min(loss, max_loss)
|
||
avatar.magic_stone.value -= loss
|
||
res_text = f"{avatar.name} 损失灵石 {loss} 枚"
|
||
|
||
elif kind == MisfortuneKind.INJURY:
|
||
# 受伤：扣减HP
|
||
# 扣减量：最大生命值的 10%~80% + 固定值
|
||
max_hp = avatar.hp.max
|
||
ratio = random.uniform(0.1, 0.8)
|
||
damage = int(max_hp * ratio) + random.randint(10, 50)
|
||
|
||
avatar.hp.cur -= damage
|
||
# 注意：这里可能扣成负数，simulator 会在 _phase_resolve_death 中处理
|
||
res_text = f"{avatar.name} 受到伤害 {damage} 点，剩余HP {avatar.hp.cur}/{max_hp}"
|
||
|
||
elif kind == MisfortuneKind.CULTIVATION_BACKLASH:
|
||
# 修为倒退
|
||
# 扣减量：100~500
|
||
loss = random.randint(100, 500)
|
||
|
||
# 确保不扣到负数（或者允许负数？通常经验不为负）
|
||
# 这里只扣减当前经验，不掉级
|
||
current_exp = avatar.cultivation_progress.exp
|
||
actual_loss = min(current_exp, loss)
|
||
avatar.cultivation_progress.exp -= actual_loss
|
||
|
||
res_text = f"{avatar.name} 修为倒退"
|
||
|
||
# 生成故事
|
||
event_text = f"遭遇霉运（{theme}），{res_text}"
|
||
story_prompt = "请据此写100~300字小故事。只描述倒霉事件本身，不要描述角色的心理活动或者愈挫愈勇。"
|
||
|
||
month_at_finish = avatar.world.month_stamp
|
||
base_event = Event(month_at_finish, event_text, related_avatars=[avatar.id], is_major=True)
|
||
|
||
story = await StoryTeller.tell_story(
|
||
event_text, res_text, avatar,
|
||
prompt=story_prompt,
|
||
allow_relation_changes=False
|
||
)
|
||
story_event = Event(month_at_finish, story, related_avatars=[avatar.id], is_story=True)
|
||
|
||
return [base_event, story_event]
|
||
|