59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
 
|
"""打卡记鹿 —— 触发器(Trigger + View)层。"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import base64
|
||
|
|
import io
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
from nonebot import on_command
|
||
|
|
from nonebot.adapters.onebot.v11 import MessageEvent, Bot, MessageSegment
|
||
|
|
from nonebot.rule import to_me
|
||
|
|
from PIL import Image
|
||
|
|
|
||
|
|
from ..services.checkin import checkin, review
|
||
|
|
|
||
|
|
record = on_command("打卡记鹿", aliases={"打卡"}, rule=to_me())
|
||
|
|
query = on_command("查询记录", aliases={"查卡"})
|
||
|
|
|
||
|
|
|
||
|
|
def pic2b64(pic: Image) -> str:
|
||
|
|
buf = io.BytesIO()
|
||
|
|
pic.save(buf, format="PNG")
|
||
|
|
base64_str = base64.b64encode(buf.getvalue()).decode()
|
||
|
|
return "base64://" + base64_str
|
||
|
|
|
||
|
|
|
||
|
|
@record.handle()
|
||
|
|
async def do_job(bot: Bot, ev: MessageEvent):
|
||
|
|
now = datetime.now()
|
||
|
|
year = now.year
|
||
|
|
month = now.month
|
||
|
|
uid = str(ev.user_id)
|
||
|
|
message_id = ev.message_id
|
||
|
|
img = await checkin(uid, year, month)
|
||
|
|
text_msg = MessageSegment.text("这是您的打卡记鹿：\n")
|
||
|
|
image_msg = MessageSegment.image(pic2b64(img))
|
||
|
|
mix_msg = (text_msg,)
|
||
|
|
mix_msg += image_msg
|
||
|
|
reply_msg = (MessageSegment.reply(message_id),)
|
||
|
|
reply_msg += mix_msg
|
||
|
|
await record.finish(reply_msg)
|
||
|
|
|
||
|
|
|
||
|
|
@query.handle()
|
||
|
|
async def query_job(bot: Bot, ev: MessageEvent):
|
||
|
|
now = datetime.now()
|
||
|
|
year = now.year
|
||
|
|
month = now.month
|
||
|
|
uid = str(ev.user_id)
|
||
|
|
message_id = ev.message_id
|
||
|
|
img_file = await review(uid, year, month)
|
||
|
|
text_msg = MessageSegment.text("这是您的记鹿：\n")
|
||
|
|
image_msg = MessageSegment.image(pic2b64(img_file))
|
||
|
|
mix_msg = (text_msg,)
|
||
|
|
mix_msg += image_msg
|
||
|
|
reply_msg = (MessageSegment.reply(message_id),)
|
||
|
|
reply_msg += mix_msg
|
||
|
|
await record.finish(reply_msg)
|