From b676895da758bd3ab512d26d949e2a1283f8e409 Mon Sep 17 00:00:00 2001 From: Goldy <153996346+g0ldyy@users.noreply.github.com> Date: Thu, 27 Feb 2025 11:09:26 +0100 Subject: [PATCH] feat: better kitsu aliases --- comet/metadata/kitsu.py | 35 ++++++++++++++++++++++------------- 1 file changed, 22 insertions(+), 13 deletions(-) diff --git a/comet/metadata/kitsu.py b/comet/metadata/kitsu.py index cad2442..4d3c01b 100644 --- a/comet/metadata/kitsu.py +++ b/comet/metadata/kitsu.py @@ -21,17 +21,26 @@ async def get_kitsu_metadata(session: aiohttp.ClientSession, id: str): async def get_kitsu_aliases(session: aiohttp.ClientSession, id: str): aliases = {} try: - response = await session.get(f"https://kitsu.io/api/edge/anime/{id}") - response = await response.json() - titles = response["data"]["attributes"]["titles"] - + response = await session.get(f"https://find-my-anime.dtimur.de/api?id={id}&provider=Kitsu") + data = await response.json() + total_aliases = 0 + aliases["ez"] = [] - for country in titles: - aliases["ez"].append(titles[country]) - - for title in response["data"]["attributes"]["abbreviatedTitles"]: - aliases["ez"].append(title) - except Exception as e: - logger.warning(f"Exception while getting Kitsu aliases for {id}: {e}") - - return aliases + aliases["ez"].append(data[0]["title"]) + for synonym in data[0]["synonyms"]: + aliases["ez"].append(synonym) + total_aliases += 1 + + print(aliases) + if total_aliases > 0: + logger.log( + "SCRAPER", + f"📜 Found {total_aliases} Kitsu aliases for {id}", + ) + return aliases + except Exception: + pass + + logger.log("SCRAPER", f"📜 No Kitsu aliases found for {id}") + + return {}