85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
 
|
import random
|
||
|
|
import re
|
||
|
|
|
||
|
|
from nonebot import on_regex, on_startswith, require
|
||
|
|
from nonebot.adapters.onebot.v11 import MessageEvent
|
||
|
|
from nonebot.plugin import PluginMetadata
|
||
|
|
from nonebot.typing import T_State
|
||
|
|
|
||
|
|
require("nonebot_plugin_alconna")
|
||
|
|
from nonebot_plugin_alconna import UniMessage
|
||
|
|
|
||
|
|
__plugin_meta__ = PluginMetadata(
|
||
|
|
name="掷骰子",
|
||
|
|
description="掷骰子 / 结婚判定",
|
||
|
|
usage=".r 掷骰子\n.r 3d12 掷3次12面骰子\n.r 2d3~6+1 掷2次3~6面骰子并加1\n.qj [名字] 与xxx结婚判定",
|
||
|
|
type="application",
|
||
|
|
)
|
||
|
|
|
||
|
|
dice = on_regex(
|
||
|
|
re.compile(
|
||
|
|
r"^\.r\s*(?:(?P<num>\d{0,2})d(?:(?P<min>\d{1,4})~)?(?P<max>\d{0,4})(?:(?P<opr>[+-])(?P<offset>\d{0,5}))?)?\b",
|
||
|
|
re.I,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
marriage = on_startswith(".qj")
|
||
|
|
|
||
|
|
|
||
|
|
async def do_dice(
|
||
|
|
ev: MessageEvent,
|
||
|
|
num: int,
|
||
|
|
min_: int,
|
||
|
|
max_: int,
|
||
|
|
opr: int,
|
||
|
|
offset: int,
|
||
|
|
tip: str = "的掷骰结果是：",
|
||
|
|
) -> None:
|
||
|
|
if num == 0:
|
||
|
|
await UniMessage.text("咦？我骰子呢？").send()
|
||
|
|
return
|
||
|
|
min_, max_ = min(min_, max_), max(min_, max_)
|
||
|
|
rolls = [random.randint(min_, max_) for _ in range(num)]
|
||
|
|
total = sum(rolls)
|
||
|
|
rolls_str = "+".join(map(str, rolls))
|
||
|
|
if len(rolls_str) > 100:
|
||
|
|
rolls_str = str(total)
|
||
|
|
res = total + opr * offset
|
||
|
|
sign = " +-"[opr]
|
||
|
|
msg = [
|
||
|
|
f"{tip}\n",
|
||
|
|
str(num) if num > 1 else "",
|
||
|
|
"D",
|
||
|
|
f"{min_}~" if min_ != 1 else "",
|
||
|
|
str(max_),
|
||
|
|
f"{sign}{offset}" if offset else "",
|
||
|
|
f"={rolls_str}",
|
||
|
|
f"{sign}{offset}" if offset else "",
|
||
|
|
f"={res}" if offset or num > 1 else "",
|
||
|
|
]
|
||
|
|
await UniMessage.text("".join(msg)).send(at_sender=True)
|
||
|
|
|
||
|
|
|
||
|
|
@dice.handle()
|
||
|
|
async def dice_handle(ev: MessageEvent, state: T_State):
|
||
|
|
num, min_, max_, opr, offset = 1, 1, 100, 1, 0
|
||
|
|
matched = state.get("_matched_dict") or {}
|
||
|
|
if s := matched.get("num"):
|
||
|
|
num = int(s)
|
||
|
|
if s := matched.get("min"):
|
||
|
|
min_ = int(s)
|
||
|
|
if s := matched.get("max"):
|
||
|
|
max_ = int(s)
|
||
|
|
if s := matched.get("opr"):
|
||
|
|
opr = -1 if s == "-" else 1
|
||
|
|
if s := matched.get("offset"):
|
||
|
|
offset = int(s)
|
||
|
|
await do_dice(ev, num, min_, max_, opr, offset)
|
||
|
|
|
||
|
|
|
||
|
|
@marriage.handle()
|
||
|
|
async def marriage_handle(ev: MessageEvent):
|
||
|
|
text = ev.message.extract_plain_text().strip()
|
||
|
|
wife = text[len(".qj") :].strip()
|
||
|
|
tip = f"与{wife}的ケッコンカッコカリ结果是：" if wife else "的ケッコンカッコカリ结果是："
|
||
|
|
await do_dice(ev, 1, 3, 6, 1, 0, tip)
|