Merge pull request #1912 from glensc/trakt_lists-plugin

Refactor: Move trakt lists logic to TraktListsPlugin plugin

plextraktsync/sync/Sync.py

@@ -2,7 +2,6 @@
 
 from typing import TYPE_CHECKING
 
-from plextraktsync.decorators.measure_time import measure_time
 from plextraktsync.factory import logging
 from plextraktsync.trakt.TraktUserListCollection import TraktUserListCollection
 
@@ -33,22 +32,11 @@ def sync(self, walker: Walker, dry_run=False):
 
         pm.hook.init(sync=self, trakt_lists=trakt_lists, is_partial=is_partial, dry_run=dry_run)
 
-        # Skip updating lists if it's empty
-        add_to_lists = not trakt_lists.is_empty
-
         if self.config.need_library_walk:
             for movie in walker.find_movies():
                 pm.hook.walk_movie(movie=movie, dry_run=dry_run)
-                if add_to_lists:
-                    trakt_lists.add_to_lists(movie)
 
             for episode in walker.find_episodes():
                 pm.hook.walk_episode(episode=episode, dry_run=dry_run)
-                if add_to_lists:
-                    trakt_lists.add_to_lists(episode)
-
-        if not dry_run and not trakt_lists.is_empty:
-            with measure_time("Updated liked list"):
-                trakt_lists.sync()
 
         pm.hook.fini(walker=walker, trakt_lists=trakt_lists, dry_run=dry_run)

plextraktsync/sync/TraktListsPlugin.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from plextraktsync.decorators.measure_time import measure_time
+from plextraktsync.factory import logging
+from plextraktsync.plugin import hookimpl
+
+if TYPE_CHECKING:
+    from plextraktsync.config.SyncConfig import SyncConfig
+    from plextraktsync.media.Media import Media
+    from plextraktsync.sync.Sync import Sync
+    from plextraktsync.trakt.TraktUserListCollection import \
+        TraktUserListCollection
+
+
+class TraktListsPlugin:
+    """
+    Plugin handling syncing of Trakt lists.
+    """
+    logger = logging.getLogger(__name__)
+
+    def __init__(self):
+        self.trakt_lists = None
+        self.add_to_lists = None
+
+    @staticmethod
+    def enabled(config: SyncConfig):
+        # Use True for now, would need to keep in sync with other plugins
+        return True
+
+    @classmethod
+    def factory(cls, sync: Sync):
+        return cls()
+
+    @hookimpl(trylast=True)
+    def init(self, trakt_lists: TraktUserListCollection):
+        self.trakt_lists = trakt_lists
+        # Skip updating lists if it's empty
+        self.add_to_lists = not trakt_lists.is_empty
+
+    @hookimpl
+    def fini(self, dry_run: bool):
+        if dry_run or self.trakt_lists.is_empty:
+            return
+
+        with measure_time("Updated liked list"):
+            self.trakt_lists.sync()
+
+    @hookimpl
+    def walk_movie(self, movie: Media):
+        if not self.add_to_lists:
+            return
+        self.trakt_lists.add_to_lists(movie)
+
+    @hookimpl
+    def walk_episode(self, episode: Media):
+        if not self.add_to_lists:
+            return
+        self.trakt_lists.add_to_lists(episode)

plextraktsync/sync/plugin/SyncPluginManager.py

@@ -34,13 +34,15 @@ def plugins(self):
         from ..LikedListsPlugin import LikedListsPlugin
         from ..SyncRatingsPlugin import SyncRatingsPlugin
         from ..SyncWatchedPlugin import SyncWatchedPlugin
+        from ..TraktListsPlugin import TraktListsPlugin
         from ..WatchListPlugin import WatchListPlugin
         from ..WatchProgressPlugin import WatchProgressPlugin
         yield AddCollectionPlugin
         yield ClearCollectedPlugin
         yield LikedListsPlugin
         yield SyncRatingsPlugin
         yield SyncWatchedPlugin
+        yield TraktListsPlugin
         yield WatchListPlugin
         yield WatchProgressPlugin