Implement find_movies using sections
+from plex_trakt_sync.media import MediaFactory
+from plex_trakt_sync.plex_api import PlexApi
+
+
class Walker:
"""
Class dealing with finding and walking library, movies/shows, episodes
"""
- def __init__(self):
+ def __init__(self, plex: PlexApi, mf: MediaFactory):
+ self.plex = plex
+ self.mf = mf
self.library = []
self.show = []
self.movie = []
@@ -18,7 +24,25 @@ def add_movie(self, movie):
self.movie.append(movie)
def find_movies(self):
- pass
+ """
+ Iterate over movie sections unless specific movie is requested
+ """
+ for section in self.movie_sections():
+ for pm in section.items():
+ m = self.mf.resolve(pm)
+ if not m:
+ continue
+ yield m
+
+ def movie_sections(self):
+ """
+ Return movie sections, optionally filter only matching section names
+ """
+ sections = self.plex.movie_sections()
+ if self.library:
+ # Filter by matching section names
+ sections = [x for x in sections if x.title in self.library]
+ return sections
def find_episodes(self):
pass
