feat✨: 识图和 buff皮肤查询

2026-10-08 13:20:01 +08:00 · 2024-05-23 16:26:06 +08:00
parent 1aed19035e
commit bd4106190e
5 changed files with 312 additions and 2 deletions
@@ -0,0 +1,89 @@
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_alconna import Alconna, Args, Arparma
+from nonebot_plugin_alconna import Image as alcImg
+from nonebot_plugin_alconna import Match, on_alconna
+from nonebot_plugin_saa import Image, Text
+from nonebot_plugin_session import EventSession
+
+from zhenxun.configs.utils import PluginExtraData, RegisterConfig
+from zhenxun.services.log import logger
+
+from .saucenao import get_saucenao_image
+
+__plugin_meta__ = PluginMetadata(
+    name="识图",
+    description="以图搜图，看破本源",
+    usage="""
+    识别图片 [二次元图片]
+    指令：
+        识图 [图片]
+    """.strip(),
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1",
+        menu_type="一些工具",
+        configs=[
+            RegisterConfig(
+                key="MAX_FIND_IMAGE_COUNT",
+                value=3,
+                help="搜索动漫返回的最大数量",
+                default_value=3,
+                type=int,
+            ),
+            RegisterConfig(
+                key="API_KEY",
+                value=None,
+                help="Saucenao的API_KEY，通过 https://saucenao.com/user.php?page=search-api 注册获取",
+            ),
+        ],
+    ).dict(),
+)
+
+
+_matcher = on_alconna(
+    Alconna("识图", Args["mode?", str]["image?", alcImg]), block=True, priority=5
+)
+
+
+async def get_image_info(mod: str, url: str) -> str | list[Image | Text] | None:
+    if mod == "saucenao":
+        return await get_saucenao_image(url)
+
+
+# def parse_image(key: str):
+#     async def _key_parser(state: T_State, img: Message = Arg(key)):
+#         if not get_message_img(img):
+#             await search_image.reject_arg(key, "请发送要识别的图片！")
+#         state[key] = img
+
+#     return _key_parser
+
+
+@_matcher.handle()
+async def _(mode: Match[str], img: Match[alcImg]):
+    if mode.available:
+        _matcher.set_path_arg("mode", mode.result)
+    else:
+        _matcher.set_path_arg("mode", "saucenao")
+    if img.available:
+        _matcher.set_path_arg("image", img.result)
+
+
+@_matcher.got_path("image", prompt="图来！")
+async def _(
+    session: EventSession,
+    arparma: Arparma,
+    mode: str,
+    image: alcImg,
+):
+    if not image.url:
+        await Text("图片url为空...").finish()
+    await Text("开始处理图片...").send()
+    info_list = await get_image_info(mode, image.url)
+    if isinstance(info_list, str):
+        await Text(info_list).finish(at_sender=True)
+    if not info_list:
+        await Text("未查询到...").finish()
+    for info in info_list[1:]:
+        await info.send()
+    logger.info(f" 识图: {image.url}", arparma.header_result, session=session)