Skip to content

Commit 46aebcd

Browse files
committed
fix: handle missing x_mitre_data_source_ref in data components
1 parent 6b4af25 commit 46aebcd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mitreattack/attackToExcel/stixToDf.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -290,7 +290,8 @@ def datasourcesToDf(src):
290290
if "x_mitre_aliases" in data_object:
291291
row["aliases"] = ", ".join(sorted(data_object["x_mitre_aliases"][1:]))
292292
if data_object["type"] == "x-mitre-data-component":
293-
row["name"] = f"{source_lookup[data_object['x_mitre_data_source_ref']]}: {data_object['name']}"
293+
if "x_mitre_data_source_ref" in data_object:
294+
row["name"] = f"{source_lookup[data_object['x_mitre_data_source_ref']]}: {data_object['name']}"
294295
row["type"] = "datacomponent"
295296
else:
296297
row["type"] = "datasource"

0 commit comments

Comments
 (0)