extractors/media/youtube: improve thumbnail selection

6 months ago · 32ddc9f03d
parent b6275b6b20
commit 32ddc9f03d
1 changed files with 5 additions and 4 deletions
--- a/server/entertainment_decider/extractors/media/youtube.py
+++ b/server/entertainment_decider/extractors/media/youtube.py
@ -140,13 +140,14 @@ class YoutubeMediaExtractor(MediaExtractor[YoutubeVideoData]):
            # TODO keep as int
            "duration": {"secondsText": str(ytdlp_info["duration"])},
            "viewCount": {"text": str(ytdlp_info["view_count"])},
-            # only supply best thumbnail available
            "thumbnails": [
                {
-                    "url": ytdlp_info["thumbnail"],
-                    "height": 1,
-                    "width": 1,
+                    "url": thumb["url"],
+                    "height": thumb["height"],
+                    "width": thumb["width"],
                }
+                for thumb in ytdlp_info["thumbnails"]
+                if "width" in thumb and "height" in thumb
            ],
            "description": ytdlp_info["description"],
            "channel": {