def collect_node_for_rule_with_group()

in visualize_rules_graph.py [0:0]

• 10 lines of code
• 4 McCabe index (conditional complexity)

def collect_node_for_rule_with_group(rule_name: str, rule_groups: 'list[str]'):
    for group_name in rule_groups:
        group_name = sanitize_name(group_name)
        if group_name != 'Cleanup_Rule':
            if group_name in rules_by_group_dict:
                rules_by_group_dict[group_name].append(rule_name)
            else:
                rules_by_group_dict[group_name] = [rule_name]
        else:  # we don't want to group `Cleanup Rule`s under the same graphviz shape
            cleanup_rules.add(rule_name)