diff --git a/vrobbler/apps/videos/sources/tmdb.py b/vrobbler/apps/videos/sources/tmdb.py index 8bbb389..35e0eda 100644 --- a/vrobbler/apps/videos/sources/tmdb.py +++ b/vrobbler/apps/videos/sources/tmdb.py @@ -13,7 +13,7 @@ tmdb_direct.api_key = "33de8d24785931068ae356510dcfbac8" tmdb = TMDb(key=key, language="en-US", region="US") -TMDB_STILL_URL = "https://image.tmdb.org/t/p/original" +TMDB_IMAGE_URL = "https://image.tmdb.org/t/p/original" logger = logging.getLogger(__name__) @@ -46,6 +46,7 @@ def lookup_video_from_tmdb( if len(tmdb_result.movie_results) > 0: media = tmdb_result.movie_results[0] video_metadata.video_type = VideoType.MOVIE.value + video_metadata.cover_url = TMDB_IMAGE_URL + media.poster_path # TODO: enrich this with TMDB url if len(tmdb_result.tv_episode_results) > 0: video_metadata.video_type = VideoType.TV_EPISODE.value media = tmdb_result.tv_episode_results[0] @@ -57,13 +58,13 @@ def lookup_video_from_tmdb( series.name = show_data.name series.save() video_metadata.tv_series_id = series.id + video_metadata.cover_url = TMDB_IMAGE_URL + media.still_path # TODO: enrich this with TMDB url if not media: logger.warning("Video not found on TMDB", extra={"imdb_id":imdb_id}) return video_metadata video_metadata.tmdb_id = media.id - video_metadata.cover_url = TMDB_STILL_URL + media.still_path # TODO: enrich this with TMDB url video_metadata.run_time_seconds = media.runtime * 60 video_metadata.title = media.name video_metadata.episode_number = media.episode_number