fixed enrich_techniques_data_sources function to not remote data sour…

…ces metadata.Fixes #45
OTRF · Jan 20, 2022 · 638e354 · 638e354
1 parent 5ec1292
commit 638e354
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/attackcti/attack_api.py b/attackcti/attack_api.py
@@ -1796,6 +1796,7 @@ def enrich_techniques_data_sources(self, stix_object):
                             technique_ds[dc_ds_ref]['data_components'] = list()
                         if dc not in technique_ds[dc_ds_ref]['data_components']:
                             technique_ds[dc_ds_ref]['data_components'].append(dc)
-                new_data_sources = [ v for v in technique_ds.values()]
-                stix_object[i] = stix_object[i].new_version(x_mitre_data_sources = new_data_sources)
-        return stix_object
+                if technique_ds:
+                    new_data_sources = [ v for v in technique_ds.values()]
+                    stix_object[i] = stix_object[i].new_version(x_mitre_data_sources = new_data_sources)
+        return stix_object