zhenxun_bot/plugins/what_anime/data_source.py

48 lines
1.9 KiB
Python
Raw Normal View History

2021-05-20 19:25:51 +08:00
import time
from services.log import logger
2021-06-30 19:50:55 +08:00
from utils.langconv import *
2021-11-23 21:44:59 +08:00
from utils.http_utils import AsyncHttpx
2021-05-20 19:25:51 +08:00
async def get_anime(anime: str) -> str:
s_time = time.time()
2021-07-30 21:21:51 +08:00
url = "https://api.trace.moe/search?anilistInfo&url={}".format(anime)
2021-05-20 19:25:51 +08:00
logger.debug("[info]Now starting get the {}".format(url))
try:
2021-11-23 21:44:59 +08:00
anime_json = (await AsyncHttpx.get(url)).json()
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
2021-05-20 19:25:51 +08:00
else:
2021-11-23 21:44:59 +08:00
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"]}'
2021-07-06 17:27:59 +08:00
except Exception as e:
2021-11-23 21:44:59 +08:00
logger.error(f"识番发生错误 {type(e)}{e}")
2021-07-30 21:21:51 +08:00
return "发生了奇怪的错误,那就没办法了,再试一次?"