From 7effe85ae2b13cea5f2fee196ecbb845f9664df7 Mon Sep 17 00:00:00 2001 From: pyth0n1c <87383215+pyth0n1c@users.noreply.github.com> Date: Mon, 10 Jan 2022 15:50:17 -0800 Subject: [PATCH] Previous commit overwrote things we did not want to, like fields besides required_fields in the tags section. Now we are more careful. --- bin/required_fields_cleaner.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/bin/required_fields_cleaner.py b/bin/required_fields_cleaner.py index d531dc927c..bed04e2092 100644 --- a/bin/required_fields_cleaner.py +++ b/bin/required_fields_cleaner.py @@ -297,7 +297,11 @@ def clean_folder(directory:str, defined_datamodels:dict): required_fields_from_yaml = set(parsed['tags']['required_fields']) fields_to_update = update_required_fields_for_yaml(filename, parsed["search"], set(parsed["tags"]["required_fields"]), set(parsed["datamodel"]), defined_datamodels, required_fields_from_yaml) if fields_to_update != {}: - parsed.update(fields_to_update) + if 'datamodel' in fields_to_update: + parsed['datamodel'] = fields_to_update['datamodel'] + if 'tags' in fields_to_update and 'required_fields' in fields_to_update['tags']: + parsed['tags']['required_fields'] = fields_to_update['tags']['required_fields'] + required_updates += 1 with open(filename, 'w') as updated_cfg: yaml.safe_dump(parsed, updated_cfg)