From 34137af81531859a18e937aa75a85cdb4dd1882b Mon Sep 17 00:00:00 2001
From: Colin Powell <colin@unbl.ink>
Date: Wed, 9 Oct 2024 21:08:04 -0400
Subject: [PATCH] [videos] Fix series lookup from IDMB too

---
 vrobbler/apps/videos/imdb.py   |  4 +++-
 vrobbler/apps/videos/models.py | 12 ++++++------
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/vrobbler/apps/videos/imdb.py b/vrobbler/apps/videos/imdb.py
index de5e74d..4d0a568 100644
--- a/vrobbler/apps/videos/imdb.py
+++ b/vrobbler/apps/videos/imdb.py
@@ -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"),
     }
diff --git a/vrobbler/apps/videos/models.py b/vrobbler/apps/videos/models.py
index 16a8196..4324bd4 100644
--- a/vrobbler/apps/videos/models.py
+++ b/vrobbler/apps/videos/models.py
@@ -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)