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

Improve image finder

Browse Source
This commit is contained in:
Zoe Roux 2023-09-06 21:06:56 +02:00
parent 328b378978
commit 6f3eefb611
No known key found for this signature in database
1 changed files with 36 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
scanner/providers/implementations/themoviedatabase.py
View File

@ -169,9 +169,23 @@ class TheMovieDatabase(Provider):
tagline=movie["tagline"] if movie["tagline"] else None,
tags=list(map(lambda x: x["name"], movie["keywords"]["keywords"])),
overview=movie["overview"],
posters=self.get_image(movie["images"]["posters"]),
posters=self.get_image(
movie["images"]["posters"]
+ (
[{"file_path": movie["poster_path"]}]
if lng == search["original_language"]
else []
)
),
logos=self.get_image(movie["images"]["logos"]),
thumbnails=self.get_image(movie["images"]["backdrops"]),
thumbnails=self.get_image(
movie["images"]["backdrops"]
+ (
[{"file_path": movie["backdrop_path"]}]
if lng == search["original_language"]
else []
)
),
trailers=[
f"https://www.youtube.com/watch?v={x['key']}"
for x in movie["videos"]["results"]
@ -185,13 +199,13 @@ class TheMovieDatabase(Provider):
async def identify_show(
self,
show: PartialShow,
pshow: PartialShow,
*,
language: list[str],
) -> Show:
show_id = show.external_id[self.name].data_id
if show.original_language not in language:
language.append(show.original_language)
show_id = pshow.external_id[self.name].data_id
if pshow.original_language not in language:
language.append(pshow.original_language)
async def for_language(lng: str) -> Show:
show = await self.get(
@ -199,7 +213,6 @@ class TheMovieDatabase(Provider):
params={
"language": lng,
"append_to_response": "alternative_titles,videos,credits,keywords,images,external_ids",
"include_image_language": f"{lng},null",
},
)
logging.debug("TMDb responded: %s", show)
@ -250,9 +263,23 @@ class TheMovieDatabase(Provider):
tagline=show["tagline"] if show["tagline"] else None,
tags=list(map(lambda x: x["name"], show["keywords"]["results"])),
overview=show["overview"],
posters=self.get_image(show["images"]["posters"]),
posters=self.get_image(
show["images"]["posters"]
+ (
[{"file_path": show["poster_path"]}]
if lng == pshow.original_language
else []
)
),
logos=self.get_image(show["images"]["logos"]),
thumbnails=self.get_image(show["images"]["backdrops"]),
thumbnails=self.get_image(
show["images"]["backdrops"]
+ (
[{"file_path": show["backdrop_path"]}]
if lng == pshow.original_language
else []
)
),
trailers=[
f"https://www.youtube.com/watch?v={x['key']}"
for x in show["videos"]["results"]