From 8f02ab0f893b13da905c582815cb868873158426 Mon Sep 17 00:00:00 2001
From: twolaw <twolaw@free.fr>
Date: Tue, 21 Sep 2021 16:22:47 +0200
Subject: [PATCH] use LazyUserList

---
 plex_trakt_sync/trakt_list_util.py | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/plex_trakt_sync/trakt_list_util.py b/plex_trakt_sync/trakt_list_util.py
index da8c56d206..1b85ec6f42 100644
--- a/plex_trakt_sync/trakt_list_util.py
+++ b/plex_trakt_sync/trakt_list_util.py
@@ -5,17 +5,45 @@
 from trakt.movies import Movie
 from trakt.tv import TVEpisode
 from trakt.users import UserList
+from trakt.core import get
+from trakt.utils import slugify, extract_ids
 
 from plex_trakt_sync.logging import logger
 from plex_trakt_sync.plex_api import PlexApi
 
 
+class LazyUserList(UserList):
+    @get
+    def get_items(self):
+        data = yield 'users/{user}/lists/{id}/items'.format(
+            user=slugify(self.creator), id=self.slug)
+        for item in data:
+            if 'type' not in item:
+                continue
+            item_type = item['type']
+            item_data = item.pop(item_type)
+            extract_ids(item_data)
+            self._items.append((item_type + 's', item_data['trakt']))
+        yield self._items
+
+    @classmethod
+    @get
+    def _get(cls, title, creator):
+        data = yield 'users/{user}/lists/{id}'.format(user=slugify(creator),
+                                                      id=slugify(title))
+        extract_ids(data)
+        ulist = LazyUserList(creator=creator, **data)
+        ulist.get_items()
+        yield ulist
+
+
 class TraktList():
     def __init__(self, username, listname):
         self.name = listname
         self.plex_items = []
         if username is not None:
-            self.trakt_items = dict(zip([(elem.media_type, elem.trakt) for elem in UserList._get(listname, username)._items if isinstance(elem, (Movie, TVEpisode))], count(1)))
+            prelist = [(elem[0], elem[1]) for elem in LazyUserList._get(listname, username)._items if elem[0] in ["movies", "episodes"]]
+            self.trakt_items = dict(zip(prelist, count(1)))
 
     @staticmethod
     def from_trakt_list(listname, trakt_list):