From b751cf4590b6cba8ec70ffbbe64e8e7f55e58013 Mon Sep 17 00:00:00 2001 From: Dan Brodjieski Date: Tue, 22 Sep 2020 16:01:11 -0400 Subject: [PATCH] create "all" keyword --- scripts/generate_baseline.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/scripts/generate_baseline.py b/scripts/generate_baseline.py index bf3b841a..3ef2e725 100755 --- a/scripts/generate_baseline.py +++ b/scripts/generate_baseline.py @@ -299,13 +299,14 @@ def main(): found_rules = [] for rule in all_rules: - if args.keyword in rule.rule_tags: + if args.keyword in rule.rule_tags or args.keyword == "all": found_rules.append(rule) # assume all baselines will contain the supplemental rules if "supplemental" in rule.rule_tags: - found_rules.append(rule) + if rule not in found_rules: + found_rules.append(rule) - if len(found_rules) == 0: + if args.keyword == None: print("No rules found for the keyword provided, please verify from the following list:") available_tags(all_rules) else: