Merge pull request #233 from glensc/episode_collected_at

Taxel · Apr 12, 2021 · fb60b12 · fb60b12
2 parents b969b1c + 6345f5b
commit fb60b12
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/plex_trakt_sync/trakt_api.py b/plex_trakt_sync/trakt_api.py
@@ -166,8 +166,6 @@ def mark_watched(self, m, time):
     @nocache
     @rate_limit(delay=TRAKT_POST_DELAY)
     def add_to_collection(self, m, pm: PlexLibraryItem):
-        # support is missing, compose custom json ourselves
-        # https://github.com/moogar0880/PyTrakt/issues/143
         if m.media_type == "movies":
             json = {
                 m.media_type: [dict(
@@ -177,9 +175,17 @@ def add_to_collection(self, m, pm: PlexLibraryItem):
                     **pm.to_json(),
                 )],
             }
-            return trakt.sync.add_to_collection(json)
+        elif m.media_type == "episodes":
+            json = {
+                m.media_type: [dict(
+                    **m.ids,
+                    **pm.to_json(),
+                )],
+            }
         else:
-            return m.add_to_library()
+            raise ValueError(f"Unsupported media type: {m.media_type}")
+
+        return trakt.sync.add_to_collection(json)
 
     @memoize
     @nocache