From 4b3c7142613ac58054db8901b6e724041c8e6bbe Mon Sep 17 00:00:00 2001 From: rouggy Date: Wed, 8 Dec 2021 11:00:52 +0100 Subject: [PATCH] update --- app/plexAutoCollectionUpdater.py | 17 +++++++++++++---- models/movie_collection.py | 4 ++++ 2 files changed, 17 insertions(+), 4 deletions(-) create mode 100644 models/movie_collection.py diff --git a/app/plexAutoCollectionUpdater.py b/app/plexAutoCollectionUpdater.py index d56dd0d..a84640f 100644 --- a/app/plexAutoCollectionUpdater.py +++ b/app/plexAutoCollectionUpdater.py @@ -2,28 +2,37 @@ from config.config import Config from tmdbv3api import TMDb, Movie, Collection from plexapi.server import PlexServer from fuzzywuzzy import fuzz +from models.movie_collection import MovieCollection # Instantiate Tmdb object tmdb = TMDb() tmdb.api_key = Config.tmdbApiKey tmdb.language = Config.tmdbLanguage -# Instantiate movie object +# Instantiate TMDb movie & collection object tmdbMovie = Movie() tmdbCollection = Collection() +# Connect to Plex Server baseurl = Config.plexBaseUrl token = Config.plexToken plex = PlexServer(baseurl, token) +# Create empty lists +AvailablePlexCollectionsMovies = [] PlexMovieList = [] MatchingList = [] TMDbCollection = [] MovieCollectionFinal = [] -movies = plex.library.section('Movies') -for video in movies.search(): - PlexMovieList.append(video.title) +# Retrieve the library Movies from Plex +plexMoviesLibrary = plex.library.section('Movies') + +for movie in plexMoviesLibrary.all(): + for collection in movie.collections: + if collection.tag and collection.tag not in AvailablePlexCollectionsMovies: + AvailablePlexCollectionsMovies.append(collection.tag) + PlexMovieList.append(movie.title) i = 0 diff --git a/models/movie_collection.py b/models/movie_collection.py new file mode 100644 index 0000000..4311660 --- /dev/null +++ b/models/movie_collection.py @@ -0,0 +1,4 @@ +class MovieCollection: + def __init__(self, id, title): + self.collectionTitle = title + self.collectionId = id