blob: a3d2d61a30c653d501339e26b7c07c15302d5b96 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
from random import choice
from nonebot.adapters.onebot.v11 import MessageEvent, MessageSegment, Message
from nonebot.adapters.onebot.v11.helpers import Cooldown
from ATRI.rule import to_bot
from .data_source import Polaroid, TEMP_PATH
_flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~"])
polaroid = Polaroid().on_command("拍立得", "获取一张以自己头像的拍立得图片！需at", rule=to_bot())
@polaroid.handle([Cooldown(15, prompt=_flmt_notice)])
async def _(event: MessageEvent):
user_id = event.get_user_id()
temp_p = TEMP_PATH / f"{user_id}.png"
if not temp_p.is_file():
pol = await Polaroid().generate(user_id)
result = MessageSegment.image(pol)
else:
result = MessageSegment.image(temp_p)
await polaroid.finish(Message(result))
|
