|
import re |
|
import random |
|
import itertools |
|
import traceback |
|
from typing import List, Tuple, Optional |
|
|
|
from nonebot import on_command |
|
from nonebot.rule import to_me |
|
from nonebot.typing import T_State |
|
from nonebot.plugin import PluginMetadata |
|
from nonebot.params import ArgPlainText |
|
from nonebot.adapters.onebot.v11 import ( |
|
Bot, |
|
MessageEvent, |
|
GroupMessageEvent, |
|
) |
|
from nonebot.log import logger |
|
|
|
from .life import Life |
|
from .talent import Talent |
|
|
|
__plugin_meta__ = PluginMetadata( |
|
name="人生重开", |
|
description="人生重开模拟器", |
|
usage="@我 remake/liferestart/人生重开", |
|
extra={ |
|
"unique_name": "remake", |
|
"example": "@小Q remake", |
|
"author": "meetwq <meetwq@gmail.com>", |
|
"version": "0.2.7", |
|
}, |
|
) |
|
|
|
|
|
remake = on_command( |
|
"remake", |
|
aliases={"liferestart", "人生重开", "人生重来"}, |
|
block=True, |
|
rule=to_me(), |
|
priority=12, |
|
) |
|
|
|
|
|
@remake.handle() |
|
async def _(state: T_State): |
|
life_ = Life() |
|
life_.load() |
|
talents = life_.rand_talents(10) |
|
state["life"] = life_ |
|
state["talents"] = talents |
|
msg = "请发送编号选择3个天赋，如“0 1 2”，或发送“随机”随机选择" |
|
des = "\n".join([f"{i}.{t}" for i, t in enumerate(talents)]) |
|
await remake.send(f"{msg}\n\n{des}") |
|
|
|
|
|
@remake.got("nums") |
|
async def _(state: T_State, reply: str = ArgPlainText("nums")): |
|
def conflict_talents(talents: List[Talent]) -> Optional[Tuple[Talent, Talent]]: |
|
for (t1, t2) in itertools.combinations(talents, 2): |
|
if t1.exclusive_with(t2): |
|
return t1, t2 |
|
return None |
|
|
|
life_: Life = state["life"] |
|
talents: List[Talent] = state["talents"] |
|
|
|
match = re.fullmatch(r"\s*(\d)\s*(\d)\s*(\d)\s*", reply) |
|
if match: |
|
nums = list(match.groups()) |
|
nums = [int(n) for n in nums] |
|
nums.sort() |
|
if nums[-1] >= 10: |
|
await remake.reject("请发送正确的编号") |
|
|
|
talents_selected = [talents[n] for n in nums] |
|
ts = conflict_talents(talents_selected) |
|
if ts: |
|
await remake.reject(f"你选择的天赋“{ts[0].name}”和“{ts[1].name}”不能同时拥有，请重新选择") |
|
elif reply == "随机": |
|
while True: |
|
nums = random.sample(range(10), 3) |
|
nums.sort() |
|
talents_selected = [talents[n] for n in nums] |
|
if not conflict_talents(talents_selected): |
|
break |
|
elif re.fullmatch(r"[\d\s]+", reply): |
|
await remake.reject("请发送正确的编号，如“0 1 2”") |
|
else: |
|
await remake.finish("人生重开已取消") |
|
|
|
life_.set_talents(talents_selected) |
|
state["talents_selected"] = talents_selected |
|
|
|
msg = ( |
|
"请发送4个数字分配“颜值、智力、体质、家境”4个属性，如“5 5 5 5”，或发送“随机”随机选择；" |
|
f"可用属性点为{life_.total_property()}，每个属性不能超过10" |
|
) |
|
await remake.send(msg) |
|
|
|
|
|
@remake.got("prop") |
|
async def _( |
|
bot: Bot, |
|
event: MessageEvent, |
|
state: T_State, |
|
reply: str = ArgPlainText("prop"), |
|
): |
|
life_: Life = state["life"] |
|
talents: List[Talent] = state["talents_selected"] |
|
total_prop = life_.total_property() |
|
|
|
match = re.fullmatch(r"\s*(\d{1,2})\s+(\d{1,2})\s+(\d{1,2})\s+(\d{1,2})\s*", reply) |
|
if match: |
|
nums = list(match.groups()) |
|
nums = [int(n) for n in nums] |
|
if sum(nums) != total_prop: |
|
await remake.reject(f"属性之和需为{total_prop}，请重新发送") |
|
elif max(nums) > 10: |
|
await remake.reject("每个属性不能超过10，请重新发送") |
|
elif reply == "随机": |
|
half_prop1 = int(total_prop / 2) |
|
half_prop2 = total_prop - half_prop1 |
|
num1 = random.randint(0, half_prop1) |
|
num2 = random.randint(0, half_prop2) |
|
nums = [num1, num2, half_prop1 - num1, half_prop2 - num2] |
|
random.shuffle(nums) |
|
elif re.fullmatch(r"[\d\s]+", reply): |
|
await remake.reject("请发送正确的数字，如“5 5 5 5”") |
|
else: |
|
await remake.finish("人生重开已取消") |
|
|
|
prop = {"CHR": nums[0], "INT": nums[1], "STR": nums[2], "MNY": nums[3]} |
|
life_.apply_property(prop) |
|
|
|
await remake.send("你的人生正在重开...") |
|
|
|
msgs = [ |
|
"已选择以下天赋：\n" + "\n".join([str(t) for t in talents]), |
|
"已设置如下属性：\n" + f"颜值{nums[0]} 智力{nums[1]} 体质{nums[2]} 家境{nums[3]}", |
|
] |
|
try: |
|
life_msgs = [] |
|
for s in life_.run(): |
|
life_msgs.append("\n".join(s)) |
|
n = 5 |
|
life_msgs = [ |
|
"\n\n".join(life_msgs[i : i + n]) for i in range(0, len(life_msgs), n) |
|
] |
|
msgs.extend(life_msgs) |
|
msgs.append(life_.gen_summary()) |
|
await send_forward_msg(bot, event, "人生重开模拟器", bot.self_id, msgs) |
|
except: |
|
logger.warning(traceback.format_exc()) |
|
await remake.finish("你的人生重开失败（") |
|
|
|
|
|
async def send_forward_msg( |
|
bot: Bot, |
|
event: MessageEvent, |
|
name: str, |
|
uin: str, |
|
msgs: List[str], |
|
): |
|
def to_json(msg): |
|
return {"type": "node", "data": {"name": name, "uin": uin, "content": msg}} |
|
|
|
messages = [to_json(msg) for msg in msgs] |
|
if isinstance(event, GroupMessageEvent): |
|
await bot.call_api( |
|
"send_group_forward_msg", group_id=event.group_id, messages=messages |
|
) |
|
else: |
|
await bot.call_api( |
|
"send_private_forward_msg", user_id=event.user_id, messages=messages |
|
) |
|
|
