From 20389193b1acca625c186f94855ac28649df8b42 Mon Sep 17 00:00:00 2001 From: Goldy <153996346+g0ldyy@users.noreply.github.com> Date: Thu, 27 Feb 2025 23:54:33 +0100 Subject: [PATCH] fix: duplicated title aliases --- comet/metadata/kitsu.py | 3 +-- comet/metadata/trakt.py | 15 ++++++--------- 2 files changed, 7 insertions(+), 11 deletions(-) diff --git a/comet/metadata/kitsu.py b/comet/metadata/kitsu.py index 2c6ea1d..f0023f8 100644 --- a/comet/metadata/kitsu.py +++ b/comet/metadata/kitsu.py @@ -23,14 +23,13 @@ async def get_kitsu_aliases(session: aiohttp.ClientSession, id: str): try: 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"] = [] aliases["ez"].append(data[0]["title"]) for synonym in data[0]["synonyms"]: aliases["ez"].append(synonym) - total_aliases += 1 + total_aliases = len(aliases["ez"]) if total_aliases > 0: logger.log( "SCRAPER", diff --git a/comet/metadata/trakt.py b/comet/metadata/trakt.py index 68ad681..2306d37 100644 --- a/comet/metadata/trakt.py +++ b/comet/metadata/trakt.py @@ -6,26 +6,23 @@ async def get_trakt_aliases( session: aiohttp.ClientSession, media_type: str, media_id: str ): - aliases = {} + aliases = set() try: response = await session.get( f"https://api.trakt.tv/{'movies' if media_type == 'movie' else 'shows'}/{media_id}/aliases" ) data = await response.json() - total_aliases = 0 + for aliase in data: - country = aliase["country"] - if country not in aliases: - aliases[country] = [] - aliases[country].append(aliase["title"]) - total_aliases += 1 + aliases.add(aliase["title"]) + total_aliases = len(aliases) if total_aliases > 0: logger.log( "SCRAPER", - f"📜 Found {total_aliases} Trakt aliases for {media_id} across {len(aliases)} countries", + f"📜 Found {total_aliases} Trakt aliases for {media_id}", ) - return aliases + return {"ez": list(aliases)} except Exception: pass