From 1dbfe7c248674c791264a697d1e04196a8f7ff87 Mon Sep 17 00:00:00 2001 From: Marcus Bakker Date: Thu, 30 Jan 2020 16:28:46 +0100 Subject: [PATCH] Another fix for when 'software_id' is not present in a group YAML file --- group_mapping.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/group_mapping.py b/group_mapping.py index 7344883..c3cad6c 100644 --- a/group_mapping.py +++ b/group_mapping.py @@ -92,9 +92,9 @@ def _get_software_techniques(groups, stage, platform): groups_dict[group_id]['techniques'] = set() if campaign != '': groups_dict[group_id]['campaign'] = str(campaign) - groups_dict[group_id]['software'] = group['software_id'] + groups_dict[group_id]['software'] = group.get('software_id', None) - if group['software_id']: + if 'software_id' in group and group['software_id']: for soft_id in group['software_id']: try: groups_dict[group_id]['techniques'].update(software_dict[soft_id])