mirror of
https://github.com/g0ldyy/comet.git
synced 2026-01-12 01:16:12 +01:00
refactor: update data types and streamline ID handling
- Changed the data type of 'kitsu_id' in the anime_mapping_cache table from TEXT to INTEGER for better consistency. - Simplified the handling of 'kitsu_id' in the MetadataScraper and AnimeMapper classes by removing unnecessary string conversions.
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user