fix for #51

4 years ago · 9a782572cc
parent 7902cffec5
commit 9a782572cc
1 changed files with 9 additions and 9 deletions
--- a/modules/config.py
+++ b/modules/config.py
@ -62,15 +62,15 @@ class Config:
                        replace_attr(new_config["libraries"][library], "show_filtered", "plex")
                        replace_attr(new_config["libraries"][library], "show_missing", "plex")
                        replace_attr(new_config["libraries"][library], "save_missing", "plex")
-            new_config["libraries"] = new_config.pop("libraries")
-            new_config["settings"] = new_config.pop("settings")
-            new_config["plex"] = new_config.pop("plex")
-            new_config["tmdb"] = new_config.pop("tmdb")
-            new_config["tautulli"] = new_config.pop("tautulli")
-            new_config["radarr"] = new_config.pop("radarr")
-            new_config["sonarr"] = new_config.pop("sonarr")
-            new_config["trakt"] = new_config.pop("trakt")
-            new_config["mal"] = new_config.pop("mal")
+            if "libraries" in new_config:                   new_config["libraries"] = new_config.pop("libraries")
+            if "settings" in new_config:                    new_config["settings"] = new_config.pop("settings")
+            if "plex" in new_config:                        new_config["plex"] = new_config.pop("plex")
+            if "tmdb" in new_config:                        new_config["tmdb"] = new_config.pop("tmdb")
+            if "tautulli" in new_config:                    new_config["tautulli"] = new_config.pop("tautulli")
+            if "radarr" in new_config:                      new_config["radarr"] = new_config.pop("radarr")
+            if "sonarr" in new_config:                      new_config["sonarr"] = new_config.pop("sonarr")
+            if "trakt" in new_config:                       new_config["trakt"] = new_config.pop("trakt")
+            if "mal" in new_config:                         new_config["mal"] = new_config.pop("mal")
            yaml.round_trip_dump(new_config, open(self.config_path, "w"), indent=ind, block_seq_indent=bsi)
            self.data = new_config
        except yaml.scanner.ScannerError as e: