1 files changed, 27 insertions, 0 deletions
diff --git a/ATRI/plugins/polaroid/__init__.py b/ATRI/plugins/polaroid/__init__.py
new file mode 100644
index 0000000..a3d2d61
--- /dev/null
+++ b/ATRI/plugins/polaroid/__init__.py
@@ -0,0 +1,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())
+
+
+[email protected]([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))