blob: 18795e7a4b62006d82a2acececf5f0669e56024b (
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
28
29
30
31
32
33
|
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 ATRI.service import Service
from .data_source import Polaroid, TEMP_PATH
plugin = Service("拍立得").document("根据头像生成拍立得风格照片！")
pol = Polaroid()
_flmt_notice = choice(["慢...慢一..点❤", "冷静1下", "歇会歇会~~"])
polaroid = plugin.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))
|
