diff options
Diffstat (limited to 'inventory')
| -rw-r--r-- | inventory/.gitignore | 1 | ||||
| -rwxr-xr-x | inventory/proxmox-ipam-inventory.py | 87 |
2 files changed, 88 insertions, 0 deletions
diff --git a/inventory/.gitignore b/inventory/.gitignore new file mode 100644 index 0000000..2eea525 --- /dev/null +++ b/inventory/.gitignore @@ -0,0 +1 @@ +.env
\ No newline at end of file diff --git a/inventory/proxmox-ipam-inventory.py b/inventory/proxmox-ipam-inventory.py new file mode 100755 index 0000000..cb0daf7 --- /dev/null +++ b/inventory/proxmox-ipam-inventory.py @@ -0,0 +1,87 @@ +#!/usr/bin/env python3 + +import requests +import ipaddress +import json +import dotenv + +def main(): + pass + +config = dotenv.dotenv_values() +proxmox_host = config["PROXMOX_URL"] +api_key = config["PROXMOX_API_KEY"] +reqs = requests.Session() +reqs.headers.update( + {"Authorization": api_key}, +) + + +def do_req(request): + request.raise_for_status() + return request + + +vms, ipam = [ + do_req(request).json() + for request in [ + reqs.get(f"{proxmox_host}/api2/json/cluster/resources"), + reqs.get(f"{proxmox_host}/api2/json/cluster/sdn/ipams/pve/status"), + ] +] + +inventory = { + "_meta": { + "hostvars": {}, + }, + "lxc": {"hosts": [], "vars": {"ansible_user": "root"}}, + "qemu": {"hosts": [], "vars": {"ansible_user": "matthieu"}}, + "vms": { + "children": ["lxc", "qemu"], + }, + "pve": { + "hosts": ["10.80.255.2", "10.80.255.200", "10.80.255.201", "10.80.255.202"], + "vars": {"ansible_connection": "ssh", "ansible_user": "root"}, + }, +} + +vmIpamDict = {} +for record in ipam["data"]: + valid = "vmid" in record and "ip" in record + if not valid: + continue + if record["vmid"] in vmIpamDict: + vmIpamDict[record["vmid"]].append(record["ip"]) + else: + vmIpamDict[record["vmid"]] = [record["ip"]] + +for vm in vms["data"]: + valid = ( + "type" in vm + and vm["type"] in ["lxc", "qemu"] + and "status" in vm + and vm["status"] == "running" + and "name" in vm + ) + if not valid: + continue + + # We only use the IPv4 IPs for Ansible since the IPv6 evpn fabric is still unstable + # because of the evpn redistribution issue. + + if f"{vm['vmid']}" in vmIpamDict: + ipamIPs = [ + ip + for ip in [ipaddress.ip_address(ip) for ip in vmIpamDict[f"{vm['vmid']}"]] + if ip.version == 4 + ] + fqdn = f"{vm['name']}.pantheon.lab.mpgn.dev" + type_ = vm["type"] + + for ip in ipamIPs: + inventory[type_]["hosts"].append(fqdn) + inventory["_meta"]["hostvars"][fqdn] = { + "ansible_host": str(ip), + } + +print(json.dumps(inventory)) |