Added the extended parameter to sync methods

#57
fuzeman · Oct 25, 2018 · ac5552f · ac5552f
1 parent 1f9469f
commit ac5552f
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 10 deletions.
diff --git a/trakt/interfaces/sync/collection.py b/trakt/interfaces/sync/collection.py
@@ -7,13 +7,20 @@ class SyncCollectionInterface(Get, Add, Remove):
     path = 'sync/collection'
     flags = {'is_collected': True}
 
-    def get(self, media=None, store=None, params=None, **kwargs):
+    def get(self, media=None, store=None, params=None, extended=None, **kwargs):
         if media is None:
             raise ValueError('Invalid value provided for the "media" parameter')
 
+        # Build query
+        query = {}
+
+        if extended:
+            query['extended'] = extended
+
+        # Request collection
         return super(SyncCollectionInterface, self).get(
-            media=media,
-            store=store,
+            media, store,
             params=params,
+            query=query,
             **kwargs
         )
diff --git a/trakt/interfaces/sync/history.py b/trakt/interfaces/sync/history.py
@@ -9,7 +9,7 @@ class SyncHistoryInterface(Get, Add, Remove):
     path = 'sync/history'
     flags = {'is_watched': True}
 
-    def get(self, media=None, id=None, page=1, per_page=10, start_at=None, end_at=None, store=None, **kwargs):
+    def get(self, media=None, id=None, page=1, per_page=10, start_at=None, end_at=None, store=None, extended=None, **kwargs):
         if not media and id:
             raise ValueError('The "id" parameter also requires the "media" parameter to be defined')
 
@@ -34,11 +34,14 @@ def get(self, media=None, id=None, page=1, per_page=10, start_at=None, end_at=No
         if end_at:
             query['end_at'] = to_iso8601_datetime(end_at)
 
+        if extended:
+            query['extended'] = extended
+
         # Request watched history
         return super(SyncHistoryInterface, self).get(
             media, store, params,
-            query=query,
             flat=True,
+            query=query,
             **kwargs
         )
 

diff --git a/trakt/interfaces/sync/ratings.py b/trakt/interfaces/sync/ratings.py
@@ -8,15 +8,24 @@ class SyncRatingsInterface(Get, Add, Remove):
     path = 'sync/ratings'
 
     @authenticated
-    def get(self, media=None, store=None, rating=None, **kwargs):
+    def get(self, media=None, store=None, rating=None, extended=None, **kwargs):
+        # Build parameters
         params = []
 
         if rating is not None:
             params.append(rating)
 
+        # Build query
+        query = {}
+
+        if extended:
+            query['extended'] = extended
+
+        # Request ratings
         return super(SyncRatingsInterface, self).get(
             media, store, params,
             flat=media is None,
+            query=query,
             **kwargs
         )
 

diff --git a/trakt/interfaces/sync/watched.py b/trakt/interfaces/sync/watched.py
@@ -7,13 +7,20 @@ class SyncWatchedInterface(Get):
     path = 'sync/watched'
     flags = {'is_watched': True}
 
-    def get(self, media=None, store=None, params=None, **kwargs):
+    def get(self, media=None, store=None, params=None, extended=None, **kwargs):
         if media is None:
             raise ValueError('Invalid value provided for the "media" parameter')
 
+        # Build query
+        query = {}
+
+        if extended:
+            query['extended'] = extended
+
+        # Request watched
         return super(SyncWatchedInterface, self).get(
-            media=media,
-            store=store,
+            media, store,
             params=params,
+            query=query,
             **kwargs
         )
diff --git a/trakt/interfaces/sync/watchlist.py b/trakt/interfaces/sync/watchlist.py
@@ -8,7 +8,7 @@ class SyncWatchlistInterface(Get, Add, Remove):
     path = 'sync/watchlist'
     flags = {'in_watchlist': True}
 
-    def get(self, media=None, page=1, per_page=10, start_at=None, end_at=None, store=None, **kwargs):
+    def get(self, media=None, page=1, per_page=10, start_at=None, end_at=None, store=None, extended=None, **kwargs):
         # Build query
         query = {}
 
@@ -18,6 +18,9 @@ def get(self, media=None, page=1, per_page=10, start_at=None, end_at=None, store
         if per_page:
             query['limit'] = per_page
 
+        if extended:
+            query['extended'] = extended
+
         # Request watched history
         return super(SyncWatchlistInterface, self).get(
             media, store,