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())
|