From 299b3e2328a997b1b5927f492e0dc1b812a9e73c Mon Sep 17 00:00:00 2001 From: Zoe Roux Date: Wed, 15 May 2024 00:28:13 +0200 Subject: [PATCH] Use image selection logic for trailers (#497) --- scanner/providers/types/movie.py | 2 +- scanner/providers/types/show.py | 2 +- scanner/providers/utils.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/scanner/providers/types/movie.py b/scanner/providers/types/movie.py index b9f3c63c..0a984e39 100644 --- a/scanner/providers/types/movie.py +++ b/scanner/providers/types/movie.py @@ -58,7 +58,7 @@ class Movie: "poster": select_image(self, "posters"), "thumbnail": select_image(self, "thumbnails"), "logo": select_image(self, "logos"), - "trailer": next(iter(trans.trailers), None), + "trailer": select_image(self, "trailers"), "studio": next((x.to_kyoo() for x in self.studios), None), "genres": [x.to_kyoo() for x in self.genres], "collections": None, diff --git a/scanner/providers/types/show.py b/scanner/providers/types/show.py index 4d82a0e6..df9abba4 100644 --- a/scanner/providers/types/show.py +++ b/scanner/providers/types/show.py @@ -61,7 +61,7 @@ class Show: "poster": select_image(self, "posters"), "thumbnail": select_image(self, "thumbnails"), "logo": select_image(self, "logos"), - "trailer": next(iter(trans.trailers), None), + "trailer": select_image(self, "trailers"), "genres": [x.to_kyoo() for x in self.genres], "file_title": None, } diff --git a/scanner/providers/utils.py b/scanner/providers/utils.py index 67938722..11171ec6 100644 --- a/scanner/providers/utils.py +++ b/scanner/providers/utils.py @@ -58,7 +58,7 @@ def select_translation( def select_image( value: Movie | Show | Season | Collection, - kind: Literal["posters"] | Literal["thumbnails"] | Literal["logos"], + kind: Literal["posters", "thumbnails", "logos", "trailers"], ) -> str | None: return next( chain(