Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Update data_source.py

Browse Source
This commit is contained in:
HibiKier 2021-07-06 17:27:59 +08:00 committed by GitHub
parent 6071c83c9f
commit 78f69b9325
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
plugins/what_anime/data_source.py
View File

@ -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 '发生了奇怪的错误，那就没办法了，再试一次？'