mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
ce18d17f4b
- 将同步函数改为异步函数，提高性能- 引入 aiohttp 库，替换 requests 库以支持异步请求 - 优化了文件写入逻辑，使用 chunk 方式写入 - 更新了相关插件中的调用方式，使其支持异步
31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
from pathlib import Path
|
|
|
|
from lazy_object_proxy.utils import await_
|
|
from nonebot.rule import to_me
|
|
from nonebot.plugin import on_command
|
|
from nonebot.adapters.qq import Message, MessageEvent, MessageSegment
|
|
from src.clover_sqlite.models.touch import QrTouch, QrTouchLog
|
|
from src.clover_image.qq_image import download_qq_image_by_account
|
|
from src.clover_image.delete_file import delete_file
|
|
from src.clover_image.rua import rua
|
|
|
|
to = on_command("摸摸头", rule=to_me(), priority=10)
|
|
@to.handle()
|
|
async def handle_touch(message: MessageEvent):
|
|
member_openid = message.get_user_id()
|
|
num = await QrTouchLog.touch_count(member_openid)
|
|
if num > 10:
|
|
await to.finish("你今天已经摸了太多次了，请明天再吧！")
|
|
q = QrTouch()
|
|
q.touch_status = 1
|
|
if num > 5:
|
|
result = await QrTouch.touch(1)
|
|
else:
|
|
result = await QrTouch.touch(0)
|
|
q.reply_touch_content = result.reply_touch_content
|
|
await QrTouchLog.insert_touch_log(q, member_openid)
|
|
local_gif = rua(await download_qq_image_by_account(None)).add_gif()
|
|
msg = Message([MessageSegment.file_image(Path(local_gif)),
|
|
MessageSegment.text(result.reply_touch_content),])
|
|
await delete_file(local_gif)
|
|
await to.finish(msg)
|