Extract find_by_episode_guid method

Taxel · Jan 8, 2024 · 80696af · 80696af
1 parent eee4151
commit 80696af
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/plextraktsync/trakt/TraktApi.py b/plextraktsync/trakt/TraktApi.py
@@ -211,15 +211,18 @@ def remove_from_watchlist(self, m):
 
         self.queue.remove_from_watchlist((m.media_type, item))
 
+    def find_by_episode_guid(self, guid: PlexGuid):
+        ts: TVShow = self.search_by_id(guid.show_id, id_type=guid.provider, media_type="show")
+        if not ts:
+            return None
+
+        lookup = TraktLookup(ts)
+
+        return self.find_episode_guid(guid, lookup)
+
     def find_by_guid(self, guid: PlexGuid):
         if guid.type == "episode" and guid.is_episode:
-            ts: TVShow = self.search_by_id(
-                guid.show_id, id_type=guid.provider, media_type="show"
-            )
-            if ts:
-                lookup = TraktLookup(ts)
-
-                return self.find_episode_guid(guid, lookup)
+            return self.find_by_episode_guid(guid)
         else:
             tm = self.search_by_id(guid.id, id_type=guid.provider, media_type=guid.type)
             if tm is None and guid.type == "movie":