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:
video_metadata = imdb_results[0]
imdb_client.update(
video_metadata, info=["plot", "synopsis", "taglines"]
video_metadata,
info=["plot", "synopsis", "taglines", "next_episode", "genres"],
)
if not video_metadata:
@ -83,4 +84,5 @@ def lookup_video_from_imdb(
"plot": video_metadata.get("plot outline"),
"imdb_rating": video_metadata.get("rating"),
"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
imdb_dict = lookup_video_from_imdb(name_or_id)
if not imdb_dict:
logger.warn(f"No imdb data for {self}")
logger.warning(f"No imdb data for {self}")
return
self.imdb_id = imdb_dict.data.get("imdbID")
self.imdb_rating = imdb_dict.data.get("arithmetic mean")
self.plot = imdb_dict.data.get("plot outline")
self.imdb_id = imdb_dict.get("imdb_id")
self.imdb_rating = imdb_dict.get("imdb_rating")
self.plot = imdb_dict.get("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:
r = requests.get(cover_url)
@ -111,7 +111,7 @@ class Series(TimeStampedModel):
fname = f"{self.name}_{self.uuid}.jpg"
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)