update

PlexAutoCollectionUpdater.py

@@ -32,6 +32,7 @@ logger.info(f"Retrieving {len(plexMoviesLibrary)} Movies from Plex and Updating
 for item in plexMoviesLibrary:
     movie = UpdatedMovie()
     tmdbMovie = Movie()
+    tmdbMovieDetails = object()
 
     movie.PlexTitle = item.title
     for guid in item.guids:
@@ -58,7 +59,7 @@ for item in plexMoviesLibrary:
     if movie.MovieTMDbCollectionTMDbId:
         logger.info(f'Movie {movie.PlexTitle} has been added to Collection List')
         MoviesList.append(movie)
-logger.info("Retrieved Movies from Plex and Updated TMDb information for {len(MoviesList)} movies...")
+logger.info(f"Retrieved Movies from Plex and Updated TMDb information for {len(MoviesList)} movies...")
 
 i = 0
 count = 0
@@ -79,36 +80,44 @@ while i < len(MoviesList):
 
 logger.info(f"Found a total of {len(MovieListFinal)} collections, updating config file now !")
 
-currentYaml = object()
+def UpdateYAML(configFile):
+    currentYaml = object()
 
-try:
+    try:
-    with open(Config.PlexAutoCollectionConfigFilePath, 'r') as f:
+        with open(configFile, 'r') as f:
-        try:
+            try:
-            currentYaml = yaml.safe_load(f)
+                currentYaml = yaml.safe_load(f)
-        except yaml.YAMLError as exc:
+            except yaml.YAMLError as exc:
-            print(exc)
+                print(exc)
-except:
+    except:
-    logger.warning(f"Could not find the config file with path: {Config.PlexAutoCollectionConfigFilePath}")
+        logger.warning(f"Could not find the config file with path: {configFile}")
 
-currentYamlCollections = []
+    currentYamlCollections = []
 
-for movies in currentYaml['collections']:
+    for movies in currentYaml['collections']:
-    currentYamlCollections.append(movies)
+        currentYamlCollections.append(movies)
 
-for collection in MovieListFinal:
+    for collection in MovieListFinal:
-    if collection.MovieTMDbCollectionName not in currentYamlCollections:
+        if collection.MovieTMDbCollectionName not in currentYamlCollections:
-        with open(Config.PlexAutoCollectionConfigFilePath, 'r') as f:
+            with open(configFile, 'r') as f:
-            new_coll = {f'{collection.MovieTMDbCollectionName}': {'sync_mode': 'sync', 'tmdb_collection': collection.MovieTMDbCollectionTMDbId, 'tmdb_summary': collection.MovieTMDbCollectionTMDbId}}
+                new_coll = {f'{collection.MovieTMDbCollectionName}': {'sync_mode': 'sync',
-            currentYaml['collections'].update(new_coll)
+                                                                      'tmdb_collection': collection.MovieTMDbCollectionTMDbId,
+                                                                      'tmdb_summary': collection.MovieTMDbCollectionTMDbId}}
+                currentYaml['collections'].update(new_coll)
 
-            if currentYaml:
+                if currentYaml:
-                logger.info(f"Adding new collection: {collection.MovieTMDbCollectionName} to Yaml file...")
+                    logger.info(f"Adding new collection: {collection.MovieTMDbCollectionName} to {configFile} file...")
-                with open(Config.PlexAutoCollectionConfigFilePath, 'w') as f:
+                    with open(configFile, 'w') as f:
-                    yaml.safe_dump(currentYaml, f)
+                        yaml.safe_dump(currentYaml, f)
 
-        if not os.path.exists(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}"):
+            if not os.path.exists(
-            logger.info(f"Download poster for new collection {collection.MovieTMDbCollectionName}...")
+                    Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}"):
-            os.mkdir(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}")
+                logger.info(f"Download poster for new collection {collection.MovieTMDbCollectionName}...")
-            urllib.request.urlretrieve(Config.tmdbImgUrl + f"{collection.MovieTMDbCollectionPosterPath}", Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}/poster.jpeg")
+                os.mkdir(Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}")
+                urllib.request.urlretrieve(Config.tmdbImgUrl + f"{collection.MovieTMDbCollectionPosterPath}",
+                                           Config.PlexAutoCollectionConfigFileImagesPath + f"/{collection.MovieTMDbCollectionName}/poster.jpeg")
+
+UpdateYAML(Config.PlexAutoCollectionConfigFilePath)
+UpdateYAML(Config.PlexAutoCollectionConfigFilePath4K)
 
 logger.info(f"All collections have been updated in plex...")

config/config.py

@@ -6,4 +6,5 @@ class Config(object):
     plexBaseUrl = 'https://plex.rouggy.ovh'
     plexToken = 'VeQQwtf-sNtWWLwzCEih'
     PlexAutoCollectionConfigFilePath = '/opt/Plex-Auto-Collections/config/config.yml'
+    PlexAutoCollectionConfigFilePath4K = '/opt/Plex-Auto-Collections/config/config-4k.yml'
     PlexAutoCollectionConfigFileImagesPath = '/opt/Plex-Auto-Collections/config/images'