mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
ce18d17f4b
- 将同步函数改为异步函数，提高性能- 引入 aiohttp 库，替换 requests 库以支持异步请求 - 优化了文件写入逻辑，使用 chunk 方式写入 - 更新了相关插件中的调用方式，使其支持异步
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from pathlib import Path
|
|
from nonebot.adapters.qq import Message, MessageEvent
|
|
from nonebot.adapters.qq import MessageSegment
|
|
from nonebot.plugin import on_command
|
|
from nonebot.rule import to_me
|
|
from src.clover_image.qq_image import download_qq_image
|
|
from src.clover_image.delete_file import delete_file
|
|
from src.clover_sqlite.models.user import UserList
|
|
|
|
today_group_wife = on_command("群老婆", rule=to_me(), priority=10)
|
|
@today_group_wife.handle()
|
|
async def handle_function(message: MessageEvent):
|
|
member_openid = message.get_user_id()
|
|
|
|
user_id = await UserList.get_user_id(member_openid,message.group_id)
|
|
if user_id is None:
|
|
await today_group_wife.finish("潜在老婆太少了，快请群友多多使用吧")
|
|
local_image_path = await download_qq_image(user_id)
|
|
msg = Message([
|
|
MessageSegment.text("您的今日群老婆"),
|
|
MessageSegment.file_image(Path(local_image_path)),
|
|
])
|
|
await delete_file(local_image_path)
|
|
await today_group_wife.finish(msg)
|
|
|
|
|
|
today_wife = on_command("今日老婆", rule=to_me(), priority=10)
|
|
@today_wife.handle()
|
|
async def handle_function(message: MessageEvent):
|
|
member_openid = message.get_user_id()
|
|
|
|
local_image_path = await download_qq_image(member_openid)
|
|
msg = Message([
|
|
MessageSegment.file_image(Path(local_image_path)),
|
|
])
|
|
|
|
await delete_file(local_image_path)
|
|
await today_wife.finish(msg)
|
|
|
|
|