From 78b3befcb37f62f1cb7d6f85c58a761a73c78263 Mon Sep 17 00:00:00 2001 From: fleaz Date: Fri, 24 Feb 2023 00:02:14 +0100 Subject: [PATCH] Catch invalid rules --- schmeckels/helper.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/schmeckels/helper.py b/schmeckels/helper.py index e4fffda..9f2cb9d 100644 --- a/schmeckels/helper.py +++ b/schmeckels/helper.py @@ -65,15 +65,23 @@ def get_rules(): data = yaml.load(fh, Loader=Loader) if data: for rule in data: - if rule.get("name"): + name = rule.get("name", None) + description = rule.get("description", None) + + if name == "" or description == "": + print("Found invalid rule with empty name or description string!") + print(rule) + sys.exit(1) + + if name: rule["name_regex"] = re.compile(rule["name"]) - if rule.get("description"): + if description: rule["desc_regex"] = re.compile(rule["description"]) else: data = [] return data else: - print(f"No rules.json found in {data_dir}! Did you run 'init'?") + print(f"No rules.yaml found in {data_dir}! Did you run 'init'?") sys.exit(1)