From 68af15f493aa125dfb4a840f750fa92ef426097a Mon Sep 17 00:00:00 2001 From: pyth0n1c <87383215+pyth0n1c@users.noreply.github.com> Date: Wed, 2 Mar 2022 16:56:46 -0800 Subject: [PATCH] Updates to field cleaner. Added Hunting as one of the types. Parse out false positive failures --- bin/required_fields_cleaner.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bin/required_fields_cleaner.py b/bin/required_fields_cleaner.py index 11d4761a61..5980cc782d 100644 --- a/bin/required_fields_cleaner.py +++ b/bin/required_fields_cleaner.py @@ -155,7 +155,7 @@ def update_required_fields_for_yaml(filename:str, search:str, required_fields:se # print(f"{filename} is {datamodels_from_datamodel_field}") for model in datamodels_from_datamodel_field: if model not in datamodels_with_submodel: - print (f"{filename} used datamodel {model} but not in search {datamodels_from_search}") + #print (f"{filename} used datamodel {model} but not in search {datamodels_from_search}") break @@ -280,8 +280,8 @@ def clean_folder(directory:str, defined_datamodels:dict): with open(filename,"r") as cfg: parsed = yaml.safe_load(cfg) - #if parsed["type"] not in ["TTP", "Anomaly" "TTP" ]: - # continue + if parsed["type"] not in ["Anomaly", "Hunting", "TTP" ]: + continue if "search" not in parsed: print(f"Failed to find ['search'] in {filename}")