Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-05-24 02:02:36 -04:00
Code Issues Packages Projects Releases Wiki Activity

Handle missing translations for series

Browse Source
This commit is contained in:
Zoe Roux 2024-05-14 01:52:09 +02:00
parent 710e3cf10b
commit ca7dd81452
No known key found for this signature in database
2 changed files with 19 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scanner/matcher/matcher.py
View File

@ -117,7 +117,7 @@ class Matcher:
)
episode.path = path
logger.debug("Got episode: %s", episode)
episode.show_id = await self.create_or_get_show(episode)
episode.show_id = await self.create_or_get_show(episode, title)
if episode.season_number is not None:
episode.season_id = await self.register_seasons(
@ -141,7 +141,7 @@ class Matcher:
provider_id = collection.external_id[self._provider.name].data_id
return await create_collection(provider_id)
async def create_or_get_show(self, episode: Episode) -> str:
async def create_or_get_show(self, episode: Episode, fallback_name: str) -> str:
@cache(ttl=timedelta(days=1), cache=self._show_cache)
async def create_show(_: str):
# TODO: Check if a show with the same metadata id exists already on kyoo.
@ -152,6 +152,7 @@ class Matcher:
if isinstance(episode.show, PartialShow)
else episode.show
)
show.file_title = fallback_name
# TODO: collections
logger.debug("Got show: %s", episode)
ret = await self._client.post("show", data=show.to_kyoo())

30
scanner/providers/types/show.py
View File

@ -1,9 +1,10 @@
import os
from dataclasses import asdict, dataclass, field
from datetime import date
from typing import Optional
from enum import Enum
from providers.utils import select_translation, select_image
from .genre import Genre
from .studio import Studio
from .season import Season
@ -20,14 +21,14 @@ class Status(str, Enum):
@dataclass
class ShowTranslation:
name: str
tagline: Optional[str]
tags: list[str]
overview: Optional[str]
tagline: Optional[str] = None
tags: list[str] = []
overview: Optional[str] = None
posters: list[str]
logos: list[str]
trailers: list[str]
thumbnails: list[str]
posters: list[str] = []
logos: list[str] = []
trailers: list[str] = []
thumbnails: list[str] = []
@dataclass
@ -46,20 +47,21 @@ class Show:
external_id: dict[str, MetadataID]
translations: dict[str, ShowTranslation] = field(default_factory=dict)
# The title of this show according to it's filename (None only for ease of use in providers)
file_title: Optional[str] = None
def to_kyoo(self):
from providers.utils import select_image
# For now, the API of kyoo only support one language so we remove the others.
default_language = os.environ["LIBRARY_LANGUAGES"].split(",")[0]
trans = select_translation(self) or ShowTranslation(name=self.file_title or "")
return {
**asdict(self),
**asdict(self.translations[default_language]),
**asdict(trans),
"rating": self.rating or 0,
"studio": next((x.to_kyoo() for x in self.studios), None),
"seasons": None,
"poster": select_image(self, "posters"),
"thumbnail": select_image(self, "thumbnails"),
"logo": select_image(self, "logos"),
"trailer": next(iter(self.translations[default_language].trailers), None),
"trailer": next(iter(trans.trailers), None),
"genres": [x.to_kyoo() for x in self.genres],
"file_title": None,
}