fix for #37

modules/config.py

@@ -894,7 +894,6 @@ class Config:
                         show_tag = details["arr_tag"] if "arr_tag" in details else None
 
                         items_found = 0
-                        library.clear_collection_missing(collection_name)
 
                         for method, values in methods:
                             logger.debug("")
@@ -1009,7 +1008,7 @@ class Config:
                                             except Failed as e:
                                                 logger.error(e)
                                         logger.info("{} Movie{} Missing".format(len(missing_movies_with_names), "s" if len(missing_movies_with_names) > 1 else ""))
-                                        library.save_missing(collection_name, missing_movies_with_names, True)
+                                        library.add_missing(collection_name, missing_movies_with_names, True)
                                         if do_arr and library.Radarr:
                                             library.Radarr.add_tmdb([missing_id for title, missing_id in missing_movies_with_names], tag=movie_tag)
                                     if len(missing_shows) > 0 and library.is_show:
@@ -1022,7 +1021,7 @@ class Config:
                                             except Failed as e:
                                                 logger.error(e)
                                         logger.info("{} Show{} Missing".format(len(missing_shows_with_names), "s" if len(missing_shows_with_names) > 1 else ""))
-                                        library.save_missing(c, missing_shows_with_names, False)
+                                        library.add_missing(c, missing_shows_with_names, False)
                                         if do_arr and library.Sonarr:
                                             library.Sonarr.add_tvdb([missing_id for title, missing_id in missing_shows_with_names], tag=show_tag)
 

modules/plex.py

@@ -82,6 +82,7 @@ class PlexAPI:
         self.radarr = params["radarr"]
         self.sonarr = params["sonarr"]
         self.tautulli = params["tautulli"]
+        self.missing = {}
 
     @retry(stop_max_attempt_number=6, wait_fixed=10000)
     def search(self, title, libtype=None, year=None):
@@ -129,37 +130,18 @@ class PlexAPI:
         except yaml.scanner.ScannerError as e:
             logger.error("YAML Error: {}".format(str(e).replace("\n", "\n|\t      ")))
 
-    def clear_collection_missing(self, collection):
-        missing_data = {}
-        if not os.path.exists(self.missing_path):
-            with open(self.missing_path, "w"): pass
-        try:
-            missing_data, ind, bsi = yaml.util.load_yaml_guess_indent(open(self.missing_path))
-            if not missing_data:
-                missing_data = {}
-            if collection in missing_data:
-                missing_data[collection.encode("ascii", "replace").decode()] = {}
-            yaml.round_trip_dump(missing_data, open(self.missing_path, "w"), indent=ind, block_seq_indent=bsi)
-        except yaml.scanner.ScannerError as e:
-            logger.error("YAML Error: {}".format(str(e).replace("\n", "\n|\t      ")))
-
-    def save_missing(self, collection, items, is_movie):
-        missing_data = {}
-        if not os.path.exists(self.missing_path):
-            with open(self.missing_path, "w"): pass
+    def add_missing(self, collection, items, is_movie):
+        col_name = collection.encode("ascii", "replace").decode()
+        if col_name not in self.missing:
+            self.missing[col_name] = {}
+        section = "Movies Missing (TMDb IDs)" if is_movie else "Shows Missing (TVDb IDs)"
+        if section not in self.missing[col_name]:
+            self.missing[col_name][section] = {}
+        for title, item_id in items:
+            self.missing[col_name][section][int(item_id)] = str(title).encode("ascii", "replace").decode()
+        with open(self.missing_path, "w"): pass
         try:
-            missing_data, ind, bsi = yaml.util.load_yaml_guess_indent(open(self.missing_path))
-            if not missing_data:
-                missing_data = {}
-            col_name = collection.encode("ascii", "replace").decode()
-            if col_name not in missing_data:
-                missing_data[col_name] = {}
-            section = "Movies Missing (TMDb IDs)" if is_movie else "Shows Missing (TVDb IDs)"
-            if section not in missing_data[col_name]:
-                missing_data[col_name][section] = {}
-            for title, item_id in items:
-                missing_data[col_name][section][int(item_id)] = str(title).encode("ascii", "replace").decode()
-            yaml.round_trip_dump(missing_data, open(self.missing_path, "w"), indent=ind, block_seq_indent=bsi)
+            yaml.round_trip_dump(self.missing, open(self.missing_path, "w"))
         except yaml.scanner.ScannerError as e:
             logger.error("YAML Error: {}".format(str(e).replace("\n", "\n|\t      ")))