43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
from typing import Optional
|
|
from nonebot.adapters import Event as BaseEvent
|
|
from nonebot.adapters.console.event import MessageEvent as ConsoleMessageEvent
|
|
from nonebot.adapters.discord.event import MessageEvent as DiscordMessageEvent
|
|
from nonebot_plugin_alconna import Alconna, Args, UniMessage, on_alconna
|
|
|
|
from konabot.plugins.roll_dice.roll_dice import generate_dice_image
|
|
from konabot.plugins.roll_dice.roll_number import get_random_number, roll_number
|
|
|
|
evt = on_alconna(Alconna(
|
|
"摇数字"
|
|
), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
|
|
|
|
@evt.handle()
|
|
async def _(event: BaseEvent):
|
|
if isinstance(event, DiscordMessageEvent):
|
|
await evt.send(await UniMessage().text("```\n" + roll_number() + "\n```").export())
|
|
elif isinstance(event, ConsoleMessageEvent):
|
|
await evt.send(await UniMessage().text(roll_number()).export())
|
|
else:
|
|
await evt.send(await UniMessage().text(roll_number(wide=True)).export())
|
|
|
|
evt = on_alconna(Alconna(
|
|
"摇骰子",
|
|
Args["f1?", int]["f2?", int]
|
|
), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
|
|
|
|
@evt.handle()
|
|
async def _(event: BaseEvent, f1: Optional[int] = None, f2: Optional[int] = None):
|
|
# if isinstance(event, DiscordMessageEvent):
|
|
# await evt.send(await UniMessage().text("```\n" + roll_dice() + "\n```").export())
|
|
# elif isinstance(event, ConsoleMessageEvent):
|
|
number = 0
|
|
if(f1 is not None and f2 is not None):
|
|
number = get_random_number(f1, f2)
|
|
elif f1 is not None:
|
|
number = get_random_number(1, f1)
|
|
else:
|
|
number = get_random_number()
|
|
await evt.send(await UniMessage().image(raw=await generate_dice_image(number)).export())
|
|
# else:
|
|
# await evt.send(await UniMessage().text(roll_dice(wide=True)).export())
|