Merge pull request #1274 from yajiwa/main

update coser
This commit is contained in:
HibiKier 2023-02-25 16:51:28 +08:00 committed by GitHub
commit cbc32a79d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,8 +2,10 @@ from typing import Tuple, Any
from nonebot import on_regex
from nonebot.params import RegexGroup
from nonebot.typing import T_State
from configs.path_config import TEMP_PATH
from nonebot.adapters.onebot.v11 import Bot, MessageEvent
import time
from utils.http_utils import AsyncHttpx
from utils.message_builder import image
from services.log import logger
from utils.manager import withdraw_message_manager
@ -40,15 +42,17 @@ coser = on_regex(r"^(\d)?连?(cos|COS|coser|括丝)$", priority=5, block=True)
# 纯cos较慢:https://picture.yinux.workers.dev
# 比较杂,有福利姬,较快:https://api.jrsgslb.cn/cos/url.php?return=img
url = "https://picture.yinux.workers.dev/"
url = "https://picture.yinux.workers.dev"
@coser.handle()
async def _(event: MessageEvent, reg_group: Tuple[Any, ...] = RegexGroup()):
num = reg_group[0] or 1
for _ in range(int(num)):
path = TEMP_PATH / f'cos_cc{int(time.time())}.jpeg'
try:
msg_id = await coser.send(image(url))
await AsyncHttpx.download_file(url, path)
msg_id = await coser.send(image(path))
withdraw_message_manager.withdraw_message(
event,
msg_id["message_id"],