From 78f69b932547f1c245981e29ea18307c321460d8 Mon Sep 17 00:00:00 2001
From: HibiKier <45528451+HibiKier@users.noreply.github.com>
Date: Tue, 6 Jul 2021 17:27:59 +0800
Subject: [PATCH] Update data_source.py

---
 plugins/what_anime/data_source.py | 49 ++++++++++++++++++++-----------
 1 file changed, 32 insertions(+), 17 deletions(-)

diff --git a/plugins/what_anime/data_source.py b/plugins/what_anime/data_source.py
index 6d968781..e417544c 100644
--- a/plugins/what_anime/data_source.py
+++ b/plugins/what_anime/data_source.py
@@ -8,28 +8,43 @@ from utils.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)
+    url = 'https://api.trace.moe/search?anilistInfo&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]
+                    if not anime_json['error']:
+                        if anime_json == 'Error reading imagenull':
+                            return "图像源错误，注意必须是静态图片哦"
+                        repass = ""
+                        # 拿到动漫 中文名
+                        for anime in anime_json["result"][:5]:
+                            synonyms = anime['anilist']['synonyms']
+                            for x in synonyms:
+                                _count_ch = 0
+                                for word in x:
+                                    if '\u4e00' <= word <= '\u9fff':
+                                        _count_ch += 1
+                                if _count_ch > 3:
+                                    anime_name = x
+                                    break
+                            else:
+                                anime_name = anime['anilist']['title']['native']
+                            episode = anime["episode"]
+                            from_ = int(anime["from"])
+                            m, s = divmod(from_, 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'访问错误 error：{anime_json["error"]}'
                 else:
                     return f'访问失败，请再试一次吧, status: {response.status}'
-    except Exception:
-        return '直接超时，那就没办法了，再试一次？'
+    except Exception as e:
+        logger.error(f'识番发生错误 e：{e}')
+        return '发生了奇怪的错误，那就没办法了，再试一次？'