update v0.0.5.0

plugins/nonebot_plugin_picsearcher/__init__.py

@@ -19,16 +19,31 @@ from .trace import get_des as get_des_trace
 from .yandex import get_des as get_des_yandex
 
 
-__plugin_name__ = "识图"
-
-__plugin_usage__ = "用法：识图 [参数](默认nao) [图片]\n" "参数列表：\n" "\t1.nao\n" "\t2.asc"
+__zx_plugin_name__ = "识图"
+__plugin_usage__ = """
+usage：
+    识别图片 [二次元图片]
+    指令：
+        识图 [图片]
+""".strip()
+__plugin_des__ = "以图搜图，看破本源"
+__plugin_cmd__ = ["识图"]
+__plugin_type__ = ("一些工具",)
+__plugin_version__ = 0.1
+__plugin_author__ = "synodriver"
+__plugin_settings__ = {
+    "level": 5,
+    "default_status": True,
+    "limit_superuser": False,
+    "cmd": ["识图"],
+}
 
 
 async def get_des(url: str, mode: str, user_id: int):
     """
     :param url: 图片链接
     :param mode: 图源
-    :return:
+    :param user_id: 用户 id
     """
     if mode == "iqdb":
         async for msg in get_des_iqdb(url):
@@ -57,8 +72,6 @@ setu = on_command("识图", aliases={"search"}, block=True, priority=5)
 async def handle_first_receive(bot: Bot, event: MessageEvent, state: T_State):
     msg = get_message_text(event.json())
     imgs = get_message_imgs(event.json())
-    if msg in ["帮助"]:
-        await setu.finish("示例：\n\t识图 (图片)\n\t识图asc (图片)")
     if imgs:
         state["setu"] = imgs[0]
     if msg:
@@ -94,13 +107,16 @@ async def get_setu(bot: Bot, event: MessageEvent, state: T_State):
         await bot.send(event=event, message="正在处理图片")
         idx = 1
         async for msg in get_des(url, mod, event.user_id):
-            await bot.send(event=event, message=msg)
-            if idx == MAX_FIND_IMG_COUNT:
-                break
-            idx += 1
+            if msg:
+                await bot.send(event=event, message=msg)
+                if idx == MAX_FIND_IMG_COUNT:
+                    break
+                idx += 1
+        if id == 1:
+            await bot.send(event=event, message='没找着.')
         logger.info(
             f"(USER {event.user_id}, GROUP "
-            f"{event.group_id if event.message_type != 'private' else 'private'}) 识图:{url}"
+            f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'}) 识图:{url}"
         )
         # image_data: List[Tuple] = await get_pic_from_url(url)
         # await setu.finish("hso")

plugins/nonebot_plugin_picsearcher/saucenao.py

@@ -93,12 +93,13 @@ async def get_des(url: str, user_id: int):
         return
     for pic in image_data:
         # print(pic)
-        msg = await download_img(pic[0], user_id) \
-              + f"\n相似度:{pic[1]}" \
-                f"\n标题:{pic[2] if (str(pic[2]).strip() != 'Creator:' and len(str(pic[2]).split('-')) < 3) else '未知'}" \
-                f"\nPID:{pic[3]}" \
-                f"\nmember:{pic[4]}\n"
-        yield msg
+        if int(str(pic[1]).split('.')[0]) > 80:
+            msg = await download_img(pic[0], user_id) \
+                  + f"\n相似度:{pic[1]}" \
+                    f"\n标题:{pic[2] if (str(pic[2]).strip() != 'Creator:' and len(str(pic[2]).split('-')) < 3) else '未知'}" \
+                    f"\nPID:{pic[3]}" \
+                    f"\nmember:{pic[4]}\n"
+            yield msg
     pass