From ecfc3a4df0f4f9f0cabcc3a37eeb6a203d13899f Mon Sep 17 00:00:00 2001
From: Chris Peterson <chris.peterson@canonical.com>
Date: Mon, 25 Mar 2024 21:30:54 -0700
Subject: [PATCH] scripts: support top-level autoinstall in validator

---
 scripts/validate-autoinstall-user-data.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/scripts/validate-autoinstall-user-data.py b/scripts/validate-autoinstall-user-data.py
index e59c3671..33bffbd2 100755
--- a/scripts/validate-autoinstall-user-data.py
+++ b/scripts/validate-autoinstall-user-data.py
@@ -60,7 +60,13 @@ def main() -> None:
         assert user_data.readline() == "#cloud-config\n"
         def get_autoinstall_data(data): return data["autoinstall"]
     else:
-        def get_autoinstall_data(data): return data
+        def get_autoinstall_data(data):
+            try:
+                cfg = data["autoinstall"]
+            except KeyError:
+                cfg = data
+            return cfg
+
 
     # Verify autoinstall doc link is in the file