Selaa lähdekoodia

[videos] Fix lookup for mevie posters

Colin Powell 2 kuukautta sitten
vanhempi
commit
1f9713312b
1 muutettua tiedostoa jossa 3 lisäystä ja 2 poistoa
  1. 3 2
      vrobbler/apps/videos/sources/tmdb.py

+ 3 - 2
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