SanYeCao-Nonebot/src/qq_plugins/touch.py
2025-01-16 17:07:05 +08:00

37 lines
1.2 KiB
Python

from pathlib import Path
from nonebot.rule import to_me
from nonebot.plugin import on_command
from nonebot.adapters.qq import Message,MessageEvent,MessageSegment
from src.my_sqlite.touch_by_sqlite import touch_count, QrTouchLog, insert_touch_log, touch
from src.image.get_image import download_qq_image_by_account,qq_image_delete,rua
to = on_command("摸摸头",rule=to_me(),priority=10,block=True)
@to.handle()
async def handle_touch(event: MessageEvent):
member_openid = event.get_user_id()
num = touch_count(member_openid)
q = QrTouchLog()
q.user_id = member_openid
if num > 10 :
await to.finish("你今天已经摸了太多次了,请明天再吧!")
elif num > 5:
result = touch(1)
q.touch_status = 1
q.reply_touch_content = result.reply_touch_content
else:
result = touch(0)
q.touch_status = 1
q.reply_touch_content = result.reply_touch_content
insert_touch_log(q)
local_gif = rua(download_qq_image_by_account(None)).add_gif()
msg = Message([
MessageSegment.file_image(Path(local_gif)),
MessageSegment.text(result.reply_touch_content),
])
qq_image_delete()
await to.finish(msg)