Merge pull request #668 from glensc/rating-sync

plextraktsync/commands/inspect.py

@@ -72,6 +72,8 @@ def inspect_media(id):
         m.show = ms
 
     print(f"Trakt: {m.trakt_url}")
+    print(f"Plex Rating: {m.plex_rating}")
+    print(f"Trakt Rating: {m.trakt_rating}")
     print(f"Watched on Plex: {m.watched_on_plex}")
     print(f"Watched on Trakt: {m.watched_on_trakt}")
 

plextraktsync/plex_api.py

@@ -2,7 +2,7 @@
 
 import datetime
 import re
-from typing import List, Optional, Union
+from typing import List, Optional, Union, Dict
 
 from plexapi import X_PLEX_CONTAINER_SIZE
 from plexapi.exceptions import BadRequest, NotFound, Unauthorized
@@ -121,9 +121,29 @@ def __str__(self):
         return self.guid
 
 
+class PlexRatingCollection(dict):
+    def __init__(self, plex: PlexApi):
+        super(dict, self).__init__()
+        self.plex = plex
+
+    def __missing__(self, section_id: int):
+        section = self.plex.library_sections[section_id]
+        ratings = self.ratings(section)
+        self[section_id] = ratings
+
+        return ratings
+
+    @flatten_dict
+    def ratings(self, section: PlexLibrarySection):
+        ratings = section.find_with_rating()
+        for item in ratings:
+            yield item.ratingKey, item.userRating
+
+
 class PlexLibraryItem:
-    def __init__(self, item: Union[Movie, Show, Episode]):
+    def __init__(self, item: Union[Movie, Show, Episode], plex: PlexApi = None):
         self.item = item
+        self.plex = plex
 
     @property
     def is_legacy_agent(self):
@@ -169,13 +189,20 @@ def type(self):
         return self.item.type
 
     @property
+    @memoize
     @nocache
     @rate_limit(retries=1)
     def rating(self):
-        if self.item.userRating is None:
+        if self.plex is not None:
+            ratings = self.plex.ratings[self.item.librarySectionID]
+            user_rating = ratings[self.item.ratingKey] if self.item.ratingKey in ratings else None
+        else:
+            user_rating = self.item.userRating
+
+        if user_rating is None:
             return None
 
-        return int(self.item.userRating)
+        return int(user_rating)
 
     @property
     def seen_date(self):
@@ -362,8 +389,9 @@ def to_json(self):
 
 
 class PlexLibrarySection:
-    def __init__(self, section: LibrarySection):
+    def __init__(self, section: LibrarySection, plex=None):
         self.section = section
+        self.plex = plex
 
     @nocache
     def __len__(self):
@@ -384,6 +412,15 @@ def find_by_title(self, name: str):
         except NotFound:
             return None
 
+    def find_with_rating(self):
+        filters = {
+            'and': [
+                {'userRating>>': -1},
+            ]
+        }
+
+        return self.section.search(filters=filters)
+
     @nocache
     def find_by_id(self, id: Union[str, int]) -> Optional[Union[Movie, Show, Episode]]:
         try:
@@ -414,7 +451,7 @@ def fetch_items(self, key: str, size: int, start: int):
 
     def items(self, max_items: int):
         for item in self.all(max_items):
-            yield PlexLibraryItem(item)
+            yield PlexLibraryItem(item, plex=self.plex)
 
     def __repr__(self):
         return f"<PlexLibrarySection:{self.type}:{self.title}>"
@@ -489,12 +526,12 @@ def updated_at(self):
     @memoize
     @flatten_dict
     @nocache
-    def library_sections(self) -> dict[PlexLibrarySection[MovieSection, ShowSection]]:
+    def library_sections(self) -> Dict[int, PlexLibrarySection]:
         CONFIG = factory.config()
         for section in self.plex.library.sections():
             if section.title in CONFIG["excluded-libraries"]:
                 continue
-            yield section.key, PlexLibrarySection(section)
+            yield section.key, PlexLibrarySection(section, plex=self)
 
     @property
     def library_section_names(self):
@@ -510,6 +547,11 @@ def system_device(self, device_id: int) -> SystemDevice:
     def system_account(self, account_id: int) -> SystemAccount:
         return self.plex.systemAccount(account_id)
 
+    @property
+    @memoize
+    def ratings(self):
+        return PlexRatingCollection(self)
+
     @nocache
     def rate(self, m, rating):
         m.rate(rating)