#328 added create_asset_folders to settings

Author: meisnate12

modules/builder.py

@@ -149,6 +149,7 @@ def __init__(self, config, library, metadata, name, data):
             "show_missing": self.library.show_missing,
             "save_missing": self.library.save_missing,
             "released_missing_only": self.library.released_missing_only,
+            "create_asset_folders": self.library.create_asset_folders,
             "item_assets": False
         }
         self.item_details = {}

modules/config.py

@@ -183,7 +183,8 @@ def check_for_attribute(data, attribute, parent=None, test_list=None, default=No
             "show_filtered": check_for_attribute(self.data, "show_filtered", parent="settings", var_type="bool", default=False),
             "show_missing": check_for_attribute(self.data, "show_missing", parent="settings", var_type="bool", default=True),
             "save_missing": check_for_attribute(self.data, "save_missing", parent="settings", var_type="bool", default=True),
-            "released_missing_only": check_for_attribute(self.data, "released_missing_only", parent="settings", var_type="bool", default=False)
+            "released_missing_only": check_for_attribute(self.data, "released_missing_only", parent="settings", var_type="bool", default=False),
+            "create_asset_folders": check_for_attribute(self.data, "create_asset_folders", parent="settings", var_type="bool", default=False)
         }
         if self.general["cache"]:
             util.separator()
@@ -349,6 +350,7 @@ def check_for_attribute(data, attribute, parent=None, test_list=None, default=No
             params["show_missing"] = check_for_attribute(lib, "show_missing", parent="settings", var_type="bool", default=self.general["show_missing"], do_print=False, save=False)
             params["save_missing"] = check_for_attribute(lib, "save_missing", parent="settings", var_type="bool", default=self.general["save_missing"], do_print=False, save=False)
             params["released_missing_only"] = check_for_attribute(lib, "released_missing_only", parent="settings", var_type="bool", default=self.general["released_missing_only"], do_print=False, save=False)
+            params["create_asset_folders"] = check_for_attribute(lib, "create_asset_folders", parent="settings", var_type="bool", default=self.general["create_asset_folders"], do_print=False, save=False)
 
             if lib and "mass_genre_update" in lib and lib["mass_genre_update"]:
                 params["mass_genre_update"] = check_for_attribute(lib, "mass_genre_update", test_list=mass_update_options, default_is_none=True, save=False)

modules/plex.py

@@ -280,6 +280,7 @@ def __init__(self, config, params):
         self.show_missing = params["show_missing"]
         self.save_missing = params["save_missing"]
         self.released_missing_only = params["released_missing_only"]
+        self.create_asset_folders = params["create_asset_folders"]
         self.mass_genre_update = params["mass_genre_update"]
         self.mass_audience_rating_update = params["mass_audience_rating_update"]
         self.mass_critic_rating_update = params["mass_critic_rating_update"]
@@ -746,7 +747,7 @@ def edit_tags(self, attr, obj, add_tags=None, remove_tags=None, sync_tags=None):
                 logger.info(f"Detail: {attr.capitalize()} {_remove} removed")
         return updated
 
-    def update_item_from_assets(self, item, overlay=None):
+    def update_item_from_assets(self, item, overlay=None, create=False):
         name = os.path.basename(os.path.dirname(str(item.locations[0])) if self.is_movie else str(item.locations[0]))
         logger.debug(name)
         found_folder = False
@@ -798,12 +799,15 @@ def update_item_from_assets(self, item, overlay=None):
                             self.upload_images(episode, poster=episode_poster)
         if not poster and overlay:
             self.upload_images(item, overlay=overlay)
-        if not overlay and self.asset_folders and not found_folder:
+        if create and self.asset_folders and not found_folder:
+            os.makedirs(os.path.join(self.asset_directory[0], name), exist_ok=True)
+            logger.info(f"Asset Directory Created: {os.path.join(self.asset_directory[0], name)}")
+        elif not overlay and self.asset_folders and not found_folder:
             logger.error(f"Asset Warning: No asset folder found called '{name}'")
         elif not poster and not background:
             logger.error(f"Asset Warning: No poster or background found in an assets folder for '{name}'")
 
-    def find_collection_assets(self, item, name=None):
+    def find_collection_assets(self, item, name=None, create=False):
         if name is None:
             name = item.title
         for ad in self.asset_directory:
@@ -825,4 +829,7 @@ def find_collection_assets(self, item, name=None):
                 background = ImageData("asset_directory", os.path.abspath(matches[0]), prefix=f"{item.title}'s ", is_poster=False, is_url=False)
             if poster or background:
                 return poster, background
+        if create and self.asset_folders and not os.path.isdir(os.path.join(self.asset_directory[0], name)):
+            os.makedirs(os.path.join(self.asset_directory[0], name), exist_ok=True)
+            logger.info(f"Asset Directory Created: {os.path.join(self.asset_directory[0], name)}")
         return None, None

plex_meta_manager.py

@@ -189,10 +189,10 @@ def update_libraries(config):
                 util.separator(f"All {'Movies' if library.is_movie else 'Shows'} Assets Check for {library.name} Library", space=False, border=False)
                 logger.info("")
                 for col in unmanaged_collections:
-                    poster, background = library.find_collection_assets(col)
+                    poster, background = library.find_collection_assets(col, create=library.create_asset_folders)
                     library.upload_images(col, poster=poster, background=background)
                 for item in library.get_all():
-                    library.update_item_from_assets(item)
+                    library.update_item_from_assets(item, create=library.create_asset_folders)
 
         logger.removeHandler(library_handler)