From b2914546f03dcd2122951ef0da4d723abd32b9e4 Mon Sep 17 00:00:00 2001
From: meisnate12 <meisnate12@gmail.com>
Date: Sat, 20 Feb 2021 12:05:11 -0500
Subject: [PATCH] Added Collection Label Support #49

---
 modules/builder.py | 18 ++++++++++++++++--
 modules/util.py    |  6 +++---
 2 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/modules/builder.py b/modules/builder.py
index 1ef551de..9bbc2b66 100644
--- a/modules/builder.py
+++ b/modules/builder.py
@@ -247,11 +247,14 @@ class CollectionBuilder:
                 elif method_name == "file_background":
                     if os.path.exists(data[m]):                                 self.backgrounds.append(("file", os.path.abspath(data[m]), method_name))
                     else:                                                       raise Failed("Collection Error: Background Path Does Not Exist: {}".format(os.path.abspath(data[m])))
-                elif method_name == "arr_tag":
+                elif method_name == "label_sync_mode":
+                    if data[m] in ["append", "sync"]:                           self.details[method_name] = data[m]
+                    else:                                                       raise Failed("Collection Error: label_sync_mode attribute must be either 'append' or 'sync'")
+                elif method_name in ["arr_tag", "label"]:
                     self.details[method_name] = util.get_list(data[m])
                 elif method_name in util.boolean_details:
                     if isinstance(data[m], bool):                               self.details[method_name] = data[m]
-                    else:                                                       raise Failed("Collection Error: {} must be either true or false".format(method_name))
+                    else:                                                       raise Failed("Collection Error: {} attribute must be either true or false".format(method_name))
                 elif method_name in util.all_details:
                     self.details[method_name] = data[m]
                 elif method_name in ["year", "year.not"]:
@@ -689,6 +692,17 @@ class CollectionBuilder:
         if "collection_order" in self.details:
             collection.sortUpdate(sort=self.details["collection_order"])
 
+        if "label" in self.details:
+            item_labels = [label.tag for label in collection.labels]
+            labels = util.get_list(self.details["label"])
+            if "label_sync_mode" in self.details and self.details["label_sync_mode"] == "sync":
+                for label in (l for l in item_labels if l not in labels):
+                    collection.removeLabel(label)
+                    logger.info("Detail: Label {} removed".format(label))
+            for label in (l for l in labels if l not in item_labels):
+                collection.addLabel(label)
+                logger.info("Detail: Label {} added".format(label))
+
         if self.library.asset_directory:
             name_mapping = self.name
             if "name_mapping" in self.details:
diff --git a/modules/util.py b/modules/util.py
index ae5022f9..13a4e133 100644
--- a/modules/util.py
+++ b/modules/util.py
@@ -249,7 +249,7 @@ collectionless_lists = [
     "collection_order", "plex_collectionless",
     "url_poster", "tmdb_poster", "tmdb_profile", "file_poster",
     "url_background", "file_background",
-    "name_mapping"
+    "name_mapping", "label", "label_sync_mode"
 ]
 other_attributes = [
     "schedule",
@@ -389,7 +389,7 @@ movie_only_filters = [
 boolean_details = [
     "add_to_arr",
     "show_filtered",
-    "show_missing", 
+    "show_missing",
     "save_missing"
 ]
 all_details = [
@@ -398,7 +398,7 @@ all_details = [
     "collection_mode", "collection_order",
     "url_poster", "tmdb_poster", "tmdb_profile", "file_poster",
     "url_background", "file_background",
-    "name_mapping", "add_to_arr",
+    "name_mapping", "add_to_arr", "arr_tag", "label",
     "show_filtered", "show_missing", "save_missing"
 ]
 discover_movie = [