secstate/vrobbler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[videos] Fix series lookup from IDMB too

Browse Source
This commit is contained in:
Colin Powell
2024-10-09 21:08:04 -04:00
parent 0f2570e51b
commit 34137af815
2 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
vrobbler/apps/videos/imdb.py
View File

@ -40,7 +40,8 @@ def lookup_video_from_imdb(
if len(imdb_results) == 1: if len(imdb_results) == 1:
video_metadata = imdb_results[0] video_metadata = imdb_results[0]
imdb_client.update( imdb_client.update(
video_metadata, info=["plot", "synopsis", "taglines"] video_metadata,
info=["plot", "synopsis", "taglines", "next_episode", "genres"],
) )
if not video_metadata: if not video_metadata:
@ -83,4 +84,5 @@ def lookup_video_from_imdb(
"plot": video_metadata.get("plot outline"), "plot": video_metadata.get("plot outline"),
"imdb_rating": video_metadata.get("rating"), "imdb_rating": video_metadata.get("rating"),
"cover_url": cover_url, "cover_url": cover_url,
"genres": video_metadata.get("genres"),
} }

12
vrobbler/apps/videos/models.py
View File

@ -95,15 +95,15 @@ class Series(TimeStampedModel):
name_or_id = self.imdb_id name_or_id = self.imdb_id
imdb_dict = lookup_video_from_imdb(name_or_id) imdb_dict = lookup_video_from_imdb(name_or_id)
if not imdb_dict: if not imdb_dict:
logger.warn(f"No imdb data for {self}") logger.warning(f"No imdb data for {self}")
return return
self.imdb_id = imdb_dict.data.get("imdbID") self.imdb_id = imdb_dict.get("imdb_id")
self.imdb_rating = imdb_dict.data.get("arithmetic mean") self.imdb_rating = imdb_dict.get("imdb_rating")
self.plot = imdb_dict.data.get("plot outline") self.plot = imdb_dict.get("plot")
self.save(update_fields=["imdb_id", "imdb_rating", "plot"]) self.save(update_fields=["imdb_id", "imdb_rating", "plot"])
cover_url = imdb_dict.get("cover url") cover_url = imdb_dict.get("cover_url")
if (not self.cover_image or force_update) and cover_url: if (not self.cover_image or force_update) and cover_url:
r = requests.get(cover_url) r = requests.get(cover_url)
@ -111,7 +111,7 @@ class Series(TimeStampedModel):
fname = f"{self.name}_{self.uuid}.jpg" fname = f"{self.name}_{self.uuid}.jpg"
self.cover_image.save(fname, ContentFile(r.content), save=True) self.cover_image.save(fname, ContentFile(r.content), save=True)
if genres := imdb_dict.data.get("genres"): if genres := imdb_dict.get("genres"):
self.genre.add(*genres) self.genre.add(*genres)