Improve visualize.py to load data directly

visualize.py

@@ -4,6 +4,7 @@ import json
 import sys
 
 import jinja2
+import hcl2
 
 icon_overrides = {
     "acme_dns": "lets-encrypt",
@@ -38,9 +39,8 @@ def service_key(svc, data, hosts):
 
 
 def service_key_find(svc_name, services, hosts):
-    for svc, data in services.items():
-        if svc == svc_name:
-            return service_key(svc, data, hosts)
+    if svc := services.get(svc_name):
+        return service_key(svc_name, svc, hosts)
     return None
 
 def parse_hosts(hosts):
@@ -74,13 +74,13 @@ def parse_services(services, hosts):
     return result
 
 if __name__ == '__main__':
-    hosts = []
-    services = []
+    hosts = {}
+    services = {}
 
-    data = json.loads(sys.stdin.read())
-
-    services = data["services"]
-    hosts = data["hosts"]
+    with open('./hosts.auto.tfvars', 'r') as file:
+        hosts = hcl2.load(file)["hosts"][0]
+    with open('./services.auto.tfvars', 'r') as file:
+        services = hcl2.load(file)["services"][0]
 
     db_key = service_key_find("postgresql", services, hosts)
     auth_key = service_key_find("authentik", services, hosts)