def has_invalid_cce(yaml_file, product_yaml=None): rule = yaml.open_and_macro_expand(yaml_file, product_yaml) if 'identifiers' in rule and rule['identifiers'] is not None: for i_type, i_value in rule['identifiers'].items(): if i_type[0:3] == 'cce': if not cce.is_cce_value_valid("CCE-" + str(i_value)): return True return False
def fix_invalid_cce(file_contents, yaml_contents): section = 'identifiers' invalid_identifiers = [] if yaml_contents[section] is not None: for i_type, i_value in yaml_contents[section].items(): if i_type[0:3] == 'cce': if not cce.is_cce_value_valid("CCE-" + str(i_value)): invalid_identifiers.append(i_type) return remove_section_keys(file_contents, yaml_contents, section, invalid_identifiers)