Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-05-31 04:04:21 -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, tagline=movie["tagline"] if movie["tagline"] else None,
tags=list(map(lambda x: x["name"], movie["keywords"]["keywords"])), tags=list(map(lambda x: x["name"], movie["keywords"]["keywords"])),
overview=movie["overview"], 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"]), 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=[ trailers=[
f"https://www.youtube.com/watch?v={x['key']}" f"https://www.youtube.com/watch?v={x['key']}"
for x in movie["videos"]["results"] for x in movie["videos"]["results"]
@ -185,13 +199,13 @@ class TheMovieDatabase(Provider):
async def identify_show( async def identify_show(
self, self,
show: PartialShow, pshow: PartialShow,
*, *,
language: list[str], language: list[str],
) -> Show: ) -> Show:
show_id = show.external_id[self.name].data_id show_id = pshow.external_id[self.name].data_id
if show.original_language not in language: if pshow.original_language not in language:
language.append(show.original_language) language.append(pshow.original_language)
async def for_language(lng: str) -> Show: async def for_language(lng: str) -> Show:
show = await self.get( show = await self.get(
@ -199,7 +213,6 @@ class TheMovieDatabase(Provider):
params={ params={
"language": lng, "language": lng,
"append_to_response": "alternative_titles,videos,credits,keywords,images,external_ids", "append_to_response": "alternative_titles,videos,credits,keywords,images,external_ids",
"include_image_language": f"{lng},null",
}, },
) )
logging.debug("TMDb responded: %s", show) logging.debug("TMDb responded: %s", show)
@ -250,9 +263,23 @@ class TheMovieDatabase(Provider):
tagline=show["tagline"] if show["tagline"] else None, tagline=show["tagline"] if show["tagline"] else None,
tags=list(map(lambda x: x["name"], show["keywords"]["results"])), tags=list(map(lambda x: x["name"], show["keywords"]["results"])),
overview=show["overview"], 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"]), 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=[ trailers=[
f"https://www.youtube.com/watch?v={x['key']}" f"https://www.youtube.com/watch?v={x['key']}"
for x in show["videos"]["results"] for x in show["videos"]["results"]