50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
"""掷骰子 —— 触发器(Trigger)层。"""
|
|||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import re
|
||
|
|
|
||
|
|
from nonebot import on_regex, on_startswith, require
|
||
|
|
from nonebot.adapters.onebot.v11 import MessageEvent
|
||
|
|
from nonebot.typing import T_State
|
||
|
|
from nonebot_plugin_alconna import UniMessage
|
||
|
|
|
||
|
|
from ..services.dice import build_dice_message
|
||
|
|
|
||
|
|
require("nonebot_plugin_alconna")
|
||
|
|
|
||
|
|
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")
|
||
|
|
|
||
|
|
|
||
|
|
@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 UniMessage.text(build_dice_message(num, min_, max_, opr, offset)).send(
|
||
|
|
at_sender=True
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@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 UniMessage.text(build_dice_message(1, 3, 6, 1, 0, tip)).send(at_sender=True)
|