Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-06-22 23:10:35 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix images chain

Browse Source
This commit is contained in:
Zoe Roux 2024-05-14 03:02:29 +02:00
parent a346800724
commit 9919f69720
No known key found for this signature in database
1 changed files with 24 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
scanner/providers/utils.py
View File

@ -2,8 +2,8 @@ from __future__ import annotations
import os import os
from datetime import date from datetime import date
from itertools import chain
from typing import TYPE_CHECKING, Literal, Any from typing import TYPE_CHECKING, Literal, Any, Optional
if TYPE_CHECKING: if TYPE_CHECKING:
from providers.types.movie import Movie from providers.types.movie import Movie
@ -12,8 +12,6 @@ if TYPE_CHECKING:
from providers.types.episode import Episode from providers.types.episode import Episode
from providers.types.collection import Collection from providers.types.collection import Collection
type Resource = Movie | Show | Season | Episode | Collection
def format_date(date: date | int | None) -> str | None: def format_date(date: date | int | None) -> str | None:
if date is None: if date is None:
@ -27,7 +25,11 @@ def format_date(date: date | int | None) -> str | None:
default_languages = os.environ["LIBRARY_LANGUAGES"].split(",") default_languages = os.environ["LIBRARY_LANGUAGES"].split(",")
def select_translation(value: Resource, *, prefer_orginal=False) -> Any: def sort_translations(
value: Movie | Show | Season | Episode | Collection,
*,
prefer_orginal=False,
):
from providers.types.movie import Movie from providers.types.movie import Movie
from providers.types.show import Show from providers.types.show import Show
@ -37,23 +39,31 @@ def select_translation(value: Resource, *, prefer_orginal=False) -> Any:
and value.original_language and value.original_language
and value.original_language in value.translations and value.original_language in value.translations
): ):
return value.translations[value.original_language] yield value.translations[value.original_language]
for lang in default_languages: for lang in default_languages:
if lang in value.translations: if lang in value.translations:
return value.translations[lang] yield value.translations[lang]
return None
def select_translation(
value: Movie | Show | Season | Episode | Collection, *, prefer_orginal=False
) -> Optional[Any]:
return next(sort_translations(value, prefer_orginal=prefer_orginal), None)
def select_image( def select_image(
value: Resource, value: Movie | Show | Season | Collection,
kind: Literal["posters"] | Literal["thumbnails"] | Literal["logos"], kind: Literal["posters"] | Literal["thumbnails"] | Literal["logos"],
) -> str | None: ) -> str | None:
trans = select_translation(value, prefer_orginal=True) or next( return next(
iter(value.translations.values()), None chain(
*(
getattr(trans, kind)
for trans in sort_translations(value, prefer_orginal=True)
)
),
None,
) )
if trans is None:
return None
return getattr(trans, kind)
class ProviderError(RuntimeError): class ProviderError(RuntimeError):