Fix YAML load errors in Nav Script

mitre-attack · Feb 29, 2024 · 34360ca · 34360ca
1 parent 55b59f7
commit 34360ca
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/scripts/generate_attack_nav_layer.py b/scripts/generate_attack_nav_layer.py
@@ -29,7 +29,12 @@
 
 # Get all analytics and load as list of dicts
 analytics_files = glob.glob(path.join(path.dirname(__file__), "..", "analytics", "*.yaml"))
-analytics = [yaml.load(open(analytic_file,encoding='utf-8').read()) for analytic_file in analytics_files]
+analytics = []
+for analytic_file in analytics_files:
+ with open(analytic_file) as af:
+ files = yaml.safe_load(af)
+ analytics.append(files)
+#analytics = [yaml.load(open(analytic_file,encoding='utf-8').read()) for analytic_file in analytics_files]
 
 def addMapping(technique, name, attack_mappings):
  if technique not in attack_mappings: