diff --git a/comet/core/database.py b/comet/core/database.py index 9b72267..253e542 100644 --- a/comet/core/database.py +++ b/comet/core/database.py @@ -344,7 +344,7 @@ async def setup_database(): await database.execute( """ CREATE TABLE IF NOT EXISTS anime_mapping_cache ( - kitsu_id TEXT PRIMARY KEY, + kitsu_id INTEGER PRIMARY KEY, imdb_id TEXT, is_anime BOOLEAN, updated_at INTEGER diff --git a/comet/metadata/manager.py b/comet/metadata/manager.py index ed9f219..46c15a7 100644 --- a/comet/metadata/manager.py +++ b/comet/metadata/manager.py @@ -185,7 +185,7 @@ class MetadataScraper: # Check if this IMDB ID has a Kitsu equivalent for additional aliases kitsu_id = anime_mapper.get_kitsu_from_imdb(media_id) if kitsu_id: - kitsu_aliases = await get_kitsu_aliases(self.session, str(kitsu_id)) + kitsu_aliases = await get_kitsu_aliases(self.session, kitsu_id) # Combine the aliases from both sources return self.combine_aliases(kitsu_aliases, trakt_aliases) diff --git a/comet/services/anime.py b/comet/services/anime.py index 9fc3b35..82a3e4b 100644 --- a/comet/services/anime.py +++ b/comet/services/anime.py @@ -170,9 +170,8 @@ class AnimeMapper: imdb_id = self._entry_value(entry, "imdb_id") if kitsu_id and imdb_id: - kitsu_id_str = str(kitsu_id) - self.kitsu_to_imdb[kitsu_id_str] = imdb_id - self.imdb_to_kitsu[imdb_id] = kitsu_id_str + self.kitsu_to_imdb[kitsu_id] = imdb_id + self.imdb_to_kitsu[imdb_id] = kitsu_id self.anime_imdb_ids.add(imdb_id) self.loaded = True @@ -187,7 +186,7 @@ class AnimeMapper: params.append( { - "kitsu_id": str(kitsu_id), + "kitsu_id": kitsu_id, "imdb_id": self._entry_value(entry, "imdb_id"), "is_anime": True, "updated_at": timestamp,