From 929d4b6ab83363864cea826728c3db05d10fe872 Mon Sep 17 00:00:00 2001
From: Daniel Robertson <52652357+endail@users.noreply.github.com>
Date: Thu, 27 Jun 2024 00:07:17 +1000
Subject: [PATCH] Update generate.py

---
 scripts/generate.py | 32 +++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/scripts/generate.py b/scripts/generate.py
index 698ecb3..117d32e 100644
--- a/scripts/generate.py
+++ b/scripts/generate.py
@@ -30,12 +30,14 @@ def ntpdig(endpoint):
       eptresult = { "endpoint": endpoint, "dns": False, "ntp": None }
       
       if resolvedns(endpoint):
-        eptresult.dns = True
+        eptresult["dns"] = True
         
         try:
           dig = ntpdig(endpoint)
-          eptresult.ntp.stratum = dig.stratum
-          eptresult.ntp.ip = dig.ip
+          eptresult["ntp"] = {
+            "stratum": dig["stratum"],
+            "ip": dig["dig"]["ip"]
+          }
         except:
           print("ntpdig failed for " + endpoint)
       
@@ -46,19 +48,19 @@ def ntpdig(endpoint):
 
 # print results
 for list in results:
-  print("List name: " + list.name)
-  print("List description: " + list.description)
+  print("List name: " + list["name"])
+  print("List description: " + list["description"])
+  print("")
+  for srv in list["servers"]:
+    print("\tServer name: " + srv["name"])
+    for ept in srv["endpoints"]:
+      print("\t\t" + ept["endpoint"])
+      print("\t\t\t" + ept["dns"])
+      print("\t\t\t" + ept["ip"])
+      print("\t\t\t" + ept["stratum"])
+      print("")
+    print("")
   print("")
-  for srv in list.servers:
-    print("\tServer name: " + srv.name)
-    for ept in srv.endpoints:
-      print("\t\t" + ept.endpoint)
-      print("\t\t\t" + ept.dns)
-      print("\t\t\t" + ept.ip)
-      print("\t\t\t" + ept.stratum)
-      print()
-    print()
-  print()
 
 print("DONE!")