From 34360ca49cc5b646ad2b128d6cd424ec3ba82a35 Mon Sep 17 00:00:00 2001
From: Lex <86126040+alexiacrumpton@users.noreply.github.com>
Date: Thu, 29 Feb 2024 16:34:28 -0500
Subject: [PATCH] Fix YAML load errors in Nav Script

---
 scripts/generate_attack_nav_layer.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/scripts/generate_attack_nav_layer.py b/scripts/generate_attack_nav_layer.py
index 3c57204b..4a4c8788 100644
--- 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: