Merge pull request #155 from AkashiCoin/main

fix: 更改p搜api，解决p搜无法使用的问题
2025-12-15 06:12:53 +08:00 · 2022-05-11 22:44:32 +08:00 · 2022-05-11 22:44:32 +08:00 · f0fd5bfb9e
commit f0fd5bfb9e
parent 33374df671 c035b1b19c
1 changed files with 106 additions and 100 deletions
--- 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)