Feature: can update reviews (#32)

metatube-community · Dec 17, 2023 · af1e821 · af1e821
1 parent 8e80187
commit af1e821
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/MetaTube.bundle/Contents/Code/__init__.py b/MetaTube.bundle/Contents/Code/__init__.py
@@ -143,7 +143,18 @@ def translate_reviews(reviews, lang):
                     review.comment = translate_text(review.comment, lang=lang,
                                                     fallback=review.comment)
 
-    # noinspection PyMethodMayBeStatic
+    @staticmethod
+    def try_update_reviews(pid):
+        if not pid.update:
+            return
+
+        try:
+            return api.get_movie_reviews(provider=pid.provider, id=pid.id,
+                                         lazy=(pid.update is not True))
+        except Exception as e:
+            Log.Warn('Try to update reviews for {pid:s} failed: {error}'
+                     .format(pid=pid, error=e))
+
     def search(self, results, media, lang, manual=False):
 
         position = None
@@ -165,6 +176,7 @@ def search(self, results, media, lang, manual=False):
                 position = pid.position  # update position
                 search_results.append(api.get_movie_info(
                     provider=pid.provider, id=pid.id, lazy=(pid.update is not True)))
+                self.try_update_reviews(pid=pid)
             except ValueError:  # fallback to name based search
                 search_results = api.search_movie(q=media.name)
 

diff --git a/MetaTube.bundle/Contents/Code/api_client.py b/MetaTube.bundle/Contents/Code/api_client.py
@@ -179,11 +179,11 @@ def get_movie_info(self, provider, id, lazy=None):
                 lazy=lazy),
             require_auth=True))
 
-    def get_movie_reviews(self, provider, id, homepage=None):
+    def get_movie_reviews(self, provider, id, homepage=None, lazy=None):
         return [MovieReviewObject(**data) for data in self.get_json(
             url=self.prepare_url(
                 self.MOVIE_REVIEW_API, provider, id,
-                homepage=homepage),
+                homepage=homepage, lazy=lazy),
             require_auth=True)]
 
     def get_primary_image_url(self, provider, id,