railiance-infra/ansible/inventory_from_yaml.py

40 lines
1.2 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
import json, yaml, subprocess, os, sys, pathlib
def load_servers():
with open(os.path.join(os.path.dirname(__file__), '..', 'inventory', 'servers.yaml')) as f:
data = yaml.safe_load(f)
servers = data.get('servers', [])
return servers
def load_tf_outputs():
# Try to read terraform outputs to attach IPs, if available.
try:
out = subprocess.check_output(['terraform', '-chdir=../terraform/hetzner', 'output', '-json'], stderr=subprocess.DEVNULL, text=True)
j = json.loads(out)
servers = j.get('servers', {}).get('value', {})
return servers # {name: ip}
except Exception:
return {}
def main():
server_list = load_servers()
tf = load_tf_outputs()
host_names = []
hostvars = {}
for s in server_list:
name = s['name']
host_names.append(name)
hostvars[name] = {
"ansible_host": tf.get(name) or s.get('ip'),
"ansible_user": s.get('ssh_user', 'admin')
}
inv = {
"all": {"hosts": host_names},
"_meta": {"hostvars": hostvars}
}
print(json.dumps(inv))
if __name__ == "__main__":
main()