fix: duplicated title aliases

g0ldyy · Feb 27, 2025 · 2038919 · 2038919
1 parent 8d118c9
commit 2038919
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 11 deletions.
diff --git 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
@@ -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