first_commit .gitignore

plugins/what_anime/__init__.py

@@ -0,0 +1,76 @@
+
+from .data_source import get_anime
+from nonebot import on_command
+from nonebot.typing import T_State
+from nonebot.adapters import Bot, Event
+from util.utils import get_message_imgs
+from services.log import logger
+from util.utils import UserExistLimiter
+
+
+__plugin_name__ = '识番'
+__plugin_usage__ = r"""
+以图识番
+识番 [图片]
+""".strip()
+
+_ulmt = UserExistLimiter()
+
+what_anime = on_command('识番', priority=5, block=True)
+
+
+@what_anime.args_parser
+async def _(bot: Bot, event: Event, state: T_State):
+    if str(event.get_message()) in ['取消', '算了']:
+        await what_anime.finish("已取消操作..", at_sender=True)
+    img_url = get_message_imgs(event.json())
+    if not img_url:
+        await what_anime.reject(prompt='图呢图呢图呢图呢GKD', at_sender=True)
+    state['img_url'] = img_url
+
+
+@what_anime.handle()
+async def _(bot: Bot, event: Event, state: T_State):
+    if str(event.get_message()) in ['帮助']:
+        await what_anime.finish(__plugin_usage__)
+    if _ulmt.check(event.user_id):
+        await what_anime.finish('您有识番任务正在进行，请稍等...', at_sender=True)
+    img_url = get_message_imgs(event.json())
+    if img_url:
+        state['img_url'] = img_url
+
+
+@what_anime.got('img_url', prompt='虚空识番？来图来图GKD')
+async def _(bot: Bot, event: Event, state: T_State):
+    img_url = state['img_url'][0]
+    _ulmt.set_True(event.user_id)
+    await what_anime.send('开始识别.....')
+    anime_data_report = await get_anime(img_url)
+    if anime_data_report:
+        await what_anime.send(anime_data_report, at_sender=True)
+        logger.info(f"USER {event.user_id} GROUP "f"{event.group_id if event.message_type != 'private' else 'private'}"
+                    f" 识番 {img_url} --> {anime_data_report}")
+    else:
+        logger.info(f"USER {event.user_id} GROUP "
+                    f"{event.group_id if event.message_type != 'private' else 'private'} 识番 {img_url} 未找到！！")
+        await what_anime.send(f"没有寻找到该番剧，果咩..", at_sender=True)
+    _ulmt.set_False(event.user_id)
+
+# @whatanime.args_parser
+# async def _(session: CommandSession):
+#     image_arg = session.current_arg_images
+#
+#     if session.is_first_run:
+#         if image_arg:
+#             session.state['whatanime'] = image_arg[0]
+#         return
+#
+#     if not image_arg:
+#         session.pause('没图说个J*，GKD!')
+#
+#     session.state[session.current_key] = image_arg
+#
+# @on_natural_language(keywords={'whatanime', '识番', '識番'}, permission=get_bot().level)
+# async def _(session: NLPSession):
+#     msg = session.msg
+#     return IntentCommand(90.0, 'whatanime', current_arg=msg or '')

plugins/what_anime/data_source.py

@@ -0,0 +1,35 @@
+
+import time
+from services.log import logger
+from util.langconv import *
+import aiohttp
+from util.user_agent import get_user_agent
+
+
+async def get_anime(anime: str) -> str:
+    s_time = time.time()
+    url = 'https://trace.moe/api/search?url={}'.format(anime)
+    logger.debug("[info]Now starting get the {}".format(url))
+    try:
+        async with aiohttp.ClientSession(headers=get_user_agent()) as session:
+            async with session.get(url, timeout=45) as response:
+                if response.status == 200:
+                    anime_json = await response.json()
+                    if anime_json == 'Error reading imagenull':
+                        return "图像源错误，注意必须是静态图片哦"
+                    repass = ""
+                    for anime in anime_json["docs"][:5]:
+                        anime_name = anime["anime"]
+                        episode = anime["episode"]
+                        at = int(anime["at"])
+                        m, s = divmod(at, 60)
+                        similarity = anime["similarity"]
+                        putline = "[ {} ][{}][{}:{}] 相似度:{:.2%}". \
+                            format(Converter("zh-hans").convert(anime_name),
+                                   episode if episode else '?', m, s, similarity)
+                        repass += putline + '\n'
+                    return f'耗时 {int(time.time() - s_time)} 秒\n' + repass[:-1]
+                else:
+                    return f'访问失败，请再试一次吧, status: {response.status}'
+    except Exception:
+        return '直接超时，那就没办法了，再试一次？'