diff --git a/media_manager/metadataProvider/schemas.py b/media_manager/metadataProvider/schemas.py index 80fe1b5..a71f57b 100644 --- a/media_manager/metadataProvider/schemas.py +++ b/media_manager/metadataProvider/schemas.py @@ -1,4 +1,6 @@ from pydantic import BaseModel +from media_manager.movies.schemas import MovieId +from media_manager.tv.schemas import ShowId class MetaDataProviderSearchResult(BaseModel): @@ -10,4 +12,4 @@ class MetaDataProviderSearchResult(BaseModel): metadata_provider: str added: bool vote_average: float | None = None - id: str | None = None # Internal ID if already added + id: MovieId | ShowId | None = None # Internal ID if already added diff --git a/media_manager/movies/service.py b/media_manager/movies/service.py index fe4faf1..0931a5c 100644 --- a/media_manager/movies/service.py +++ b/media_manager/movies/service.py @@ -234,7 +234,7 @@ class MovieService: movie = self.movie_repository.get_movie_by_external_id( external_id=result.external_id, metadata_provider=metadata_provider.name ) - result.id = str(movie.id) + result.id = movie.id except Exception: log.error(f"Unable to find internal movie ID for {result.external_id} on {metadata_provider.name}") return results diff --git a/media_manager/tv/service.py b/media_manager/tv/service.py index 9329b78..2640194 100644 --- a/media_manager/tv/service.py +++ b/media_manager/tv/service.py @@ -247,7 +247,7 @@ class TvService: show = self.tv_repository.get_show_by_external_id( external_id=result.external_id, metadata_provider=metadata_provider.name ) - result.id = str(show.id) + result.id = show.id except Exception: log.error(f"Unable to find internal show ID for {result.external_id} on {metadata_provider.name}") return results