mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
from nonebot import on_command
|
||
from nonebot.adapters.cqhttp import Bot, MessageEvent, Message, GroupMessageEvent
|
||
from nonebot.typing import T_State
|
||
from utils.utils import get_message_text, is_number
|
||
from utils.init_result import image
|
||
import aiohttp
|
||
from services.log import logger
|
||
from asyncio.exceptions import TimeoutError
|
||
import asyncio
|
||
import aiofiles
|
||
from io import BytesIO
|
||
from configs.path_config import IMAGE_PATH
|
||
|
||
try:
|
||
import ujson as json
|
||
except ModuleNotFoundError:
|
||
import json
|
||
|
||
__plugin_name__ = 'p搜'
|
||
__plugin_usage__ = '用法： 通过pid在Pixiv上搜索图片\n格式：p搜 [pid]\n\t示例：p搜 79520120'
|
||
|
||
pid_search = on_command("p搜", aliases={'pixiv搜', 'P搜'}, priority=5, block=True)
|
||
|
||
url = 'https://api.fantasyzone.cc/tu/search.php'
|
||
|
||
|
||
@pid_search.args_parser
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
pid = get_message_text(event.json())
|
||
if pid:
|
||
if pid in ['取消', '算了']:
|
||
await pid_search.finish('已取消操作...')
|
||
if not is_number(pid):
|
||
await pid_search.reject('笨蛋，重新输入数！字！', at_sender=True)
|
||
state['pid'] = pid
|
||
|
||
|
||
@pid_search.handle()
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
pid = get_message_text(event.json())
|
||
if pid:
|
||
state['pid'] = pid
|
||
|
||
|
||
@pid_search.got('pid', prompt='需要查询的图片PID是？')
|
||
async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||
pid = state['pid']
|
||
params = {
|
||
'id': pid,
|
||
'p': 1,
|
||
}
|
||
async with aiohttp.ClientSession() as session:
|
||
for _ in range(10):
|
||
try:
|
||
async with session.get(url, timeout=2, params=params) as response:
|
||
data = json.loads(await response.text())
|
||
except TimeoutError:
|
||
pass
|
||
else:
|
||
if not data['width'] and not data['height']:
|
||
await pid_search.finish(f'没有搜索到 PID：{pid} 的图片', at_sender=True)
|
||
pid = data['id']
|
||
title = data['title']
|
||
author = data['userName']
|
||
author_id = data['userId']
|
||
img_url = data['url']
|
||
for _ in range(5):
|
||
try:
|
||
await download_pic(img_url, event.user_id)
|
||
except TimeoutError:
|
||
pass
|
||
else:
|
||
break
|
||
else:
|
||
await pid_search.finish('图片下载失败了....', at_sender=True)
|
||
tmp = ''
|
||
if isinstance(event, GroupMessageEvent):
|
||
tmp = '\n【注】将在30后撤回......'
|
||
msg_id = await pid_search.send(Message(f'title：{title}\n'
|
||
f'pid：{pid}\n'
|
||
f'author：{author}\n'
|
||
f'author_id：{author_id}\n'
|
||
f'{image(f"pid_search_{event.user_id}.png", "temp")}'
|
||
f'{tmp}'))
|
||
logger.info(
|
||
f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'})"
|
||
f" 查询图片 PID：{pid}")
|
||
if isinstance(event, GroupMessageEvent):
|
||
await asyncio.sleep(30)
|
||
await bot.delete_msg(message_id=msg_id['message_id'], self_id=int(bot.self_id))
|
||
break
|
||
else:
|
||
await pid_search.finish('图片下载失败了....', at_sender=True)
|
||
|
||
|
||
async def download_pic(img_url: str, user_id: int):
|
||
async with aiohttp.ClientSession() as session:
|
||
async with session.get(img_url, timeout=2) as res:
|
||
async with aiofiles.open(f'{IMAGE_PATH}/temp/pid_search_{user_id}.png', 'wb') as f:
|
||
await f.write(await res.read())
|