From c035b1b19ca858caae0d7d970eb8bea3076c5c61 Mon Sep 17 00:00:00 2001 From: AkashiCoin <55268546+AkashiCoin@users.noreply.github.com> Date: Tue, 10 May 2022 18:47:39 +0800 Subject: [PATCH] Update pid_search.py --- plugins/pid_search.py | 206 ++++++++++++++++++++++-------------------- 1 file changed, 106 insertions(+), 100 deletions(-) diff --git a/plugins/pid_search.py b/plugins/pid_search.py index 38e0a3f6..748833dc 100755 --- a/plugins/pid_search.py +++ b/plugins/pid_search.py @@ -1,100 +1,106 @@ -from asyncio.exceptions import TimeoutError -from nonebot import on_command -from nonebot.adapters.onebot.v11 import MessageEvent, Message, GroupMessageEvent -from nonebot.typing import T_State -from configs.path_config import TEMP_PATH -from services.log import logger -from utils.message_builder import image -from utils.utils import is_number -from utils.manager import withdraw_message_manager -from utils.http_utils import AsyncHttpx -from nonebot.params import CommandArg, Arg - -try: - import ujson as json -except ModuleNotFoundError: - import json - - -__zx_plugin_name__ = "pid搜索" -__plugin_usage__ = """ -usage： - 通过 pid 搜索图片 - 指令： - p搜 [pid] -""".strip() -__plugin_des__ = "通过 pid 搜索图片" -__plugin_cmd__ = ["p搜 [pid]"] -__plugin_version__ = 0.1 -__plugin_author__ = "HibiKier" -__plugin_settings__ = { - "level": 5, - "default_status": True, - "limit_superuser": False, - "cmd": ["p搜"], -} - -pid_search = on_command("p搜", aliases={"pixiv搜", "P搜"}, priority=5, block=True) - -url = "https://api.fantasyzone.cc/tu/search.php" - - -@pid_search.handle() -async def _(state: T_State, arg: Message = CommandArg()): - pid = arg.extract_plain_text().strip() - if pid: - state["pid"] = pid - - -@pid_search.got("pid", prompt="需要查询的图片PID是？") -async def _(event: MessageEvent, state: T_State, pid: str = Arg("pid")): - if pid in ["取消", "算了"]: - await pid_search.finish("已取消操作...") - if is_number(pid): - await pid_search.reject_arg("pid", "笨蛋，重新输入数！字！") - params = { - "id": pid, - "p": 1, - } - for _ in range(3): - try: - data = (await AsyncHttpx.get(url, params=params, timeout=5)).json() - except TimeoutError: - pass - except Exception as e: - await pid_search.finish(f"发生了一些错误..{type(e)}：{e}") - 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"] - if not await AsyncHttpx.download_file( - img_url, TEMP_PATH / f"pid_search_{event.user_id}.png" - ): - 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(TEMP_PATH / f"pid_search_{event.user_id}.png")}' - f"{tmp}" - ) - ) - logger.info( - f"(USER {event.user_id}, " - f"GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})" - f" 查询图片 PID：{pid}" - ) - if isinstance(event, GroupMessageEvent): - withdraw_message_manager.append((msg_id, 30)) - break - else: - await pid_search.finish("图片下载失败了....", at_sender=True) +from nonebot import on_command +from nonebot.adapters.onebot.v11 import Bot, MessageEvent, Message, GroupMessageEvent +from nonebot.typing import T_State +from utils.utils import is_number +from utils.message_builder import image +from services.log import logger +from asyncio.exceptions import TimeoutError +from configs.path_config import IMAGE_PATH +from utils.manager import withdraw_message_manager +from utils.http_utils import AsyncHttpx +from nonebot.params import CommandArg, Arg + + +__zx_plugin_name__ = "pid搜索" +__plugin_usage__ = """ +usage： + 通过 pid 搜索图片 + 指令： + p搜 [pid] +""".strip() +__plugin_des__ = "通过 pid 搜索图片" +__plugin_cmd__ = ["p搜 [pid]"] +__plugin_version__ = 0.1 +__plugin_author__ = "HibiKier" +__plugin_settings__ = { + "level": 5, + "default_status": True, + "limit_superuser": False, + "cmd": ["p搜"], +} + +pid_search = on_command("p搜", aliases={"pixiv搜", "P搜"}, priority=5, block=True) + +url = "https://api.obfs.dev/api/pixiv/" + + +@pid_search.handle() +async def _h(event: MessageEvent, state: T_State, arg: Message = CommandArg()): + pid = arg.extract_plain_text().strip() + if pid: + state["pid"] = pid + + +headers = { + "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6;" + " rv:2.0.1) Gecko/20100101 Firefox/4.0.1", + "Referer": "https://www.pixiv.net", +} + + +@pid_search.got("pid", prompt="需要查询的图片PID是？") +async def _g(event: MessageEvent, state: T_State, pid: str = Arg("pid")): + if pid in ["取消", "算了"]: + await pid_search.finish("已取消操作...") + # if is_number(pid): + # await pid_search.reject_arg("pid", "笨蛋，重新输入数！字！") + for _ in range(3): + try: + data = ( + await AsyncHttpx.get( + url, + params={"id": pid}, + timeout=5, + ) + ).json() + except TimeoutError: + pass + except Exception as e: + await pid_search.finish(f"发生了一些错误..{type(e)}：{e}") + else: + if not data["illust"]["width"] and not data["illust"]["height"]: + await pid_search.finish(f"没有搜索到 PID：{pid} 的图片", at_sender=True) + pid = data["illust"]["id"] + title = data["illust"]["title"] + author = data["illust"]["user"]["name"] + author_id = data["illust"]["user"]["id"] + img_url = data["illust"]["meta_single_page"]["original_image_url"] + if not await AsyncHttpx.download_file( + img_url, + IMAGE_PATH / "temp" / f"pid_search_{event.user_id}.png", + headers=headers, + ): + 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}, " + f"GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})" + f" 查询图片 PID：{pid}" + ) + if isinstance(event, GroupMessageEvent): + withdraw_message_manager.append((msg_id, 30)) + break + else: + await pid_search.finish("图片下载失败了....", at_sender=True)