From 8d68fd3cbb67aa30ad68c6b72ea0507503862b0f Mon Sep 17 00:00:00 2001 From: patrick Date: Sat, 28 Jun 2025 00:08:15 +0200 Subject: [PATCH] =?UTF-8?q?proxmox-inventory.py=20hinzugef=C3=BCgt?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- proxmox-inventory.py | 156 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 156 insertions(+) create mode 100644 proxmox-inventory.py diff --git a/proxmox-inventory.py b/proxmox-inventory.py new file mode 100644 index 0000000..c1be6c5 --- /dev/null +++ b/proxmox-inventory.py @@ -0,0 +1,156 @@ +#!/usr/bin/env python3 + +import sys +import re +import subprocess +import json +import argparse +import csv + +def run_local_cmd(cmd): + try: + result = subprocess.run(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, timeout=10) + return result.stdout.strip() if result.returncode == 0 else "" + except subprocess.TimeoutExpired: + return "" + +def run_ssh_cmd(host, user, cmd, key=None): + ssh_cmd = f"ssh -o BatchMode=yes -o ConnectTimeout=5 " + if key: + ssh_cmd += f"-i {key} " + ssh_cmd += f"{user}@{host} '{cmd}'" + return run_local_cmd(ssh_cmd) + +def parse_vm_list(output): + return [line.split()[0] for line in output.strip().splitlines()[1:] if line.strip()] + +def get_config_lines(host, user, use_ssh=False, key=None, vmid=None, ct=False): + cmd = f"{'pct' if ct else 'qm'} config {vmid}" + return (run_ssh_cmd(host, user, cmd, key) if use_ssh else run_local_cmd(cmd)).splitlines() + +def collect_node(host, user, use_ssh=False, key=None): + rows = [] + + hostname = run_ssh_cmd(host, user, "hostname -f", key) if use_ssh else run_local_cmd("hostname -f") + ip = run_ssh_cmd(host, user, "hostname -I | awk '{print $1}'", key) if use_ssh else run_local_cmd("hostname -I | awk '{print $1}'") + cpu = run_ssh_cmd(host, user, "nproc", key) if use_ssh else run_local_cmd("nproc") + ram = run_ssh_cmd(host, user, "free -m | awk '/Mem:/ {print $2\"M\"}'", key) if use_ssh else run_local_cmd("free -m | awk '/Mem:/ {print $2\"M\"}'") + + rows.append({ + 'ID': '0', + 'Hostname': hostname or '-', + 'IP': ip or '-', + 'Typ': 'Node', + 'CPU': cpu or '-', + 'RAM': ram or '-', + 'Storage': "-", + 'Tags': "-" + }) + + for vmid in parse_vm_list(run_ssh_cmd(host, user, "qm list", key) if use_ssh else run_local_cmd("qm list")): + cfg = get_config_lines(host, user, use_ssh, key, vmid) + rows.append(parse_config_line(cfg, vmid, "VM")) + + for ctid in parse_vm_list(run_ssh_cmd(host, user, "pct list", key) if use_ssh else run_local_cmd("pct list")): + cfg = get_config_lines(host, user, use_ssh, key, ctid, ct=True) + rows.append(parse_config_line(cfg, ctid, "CT")) + + return rows + +def parse_config_line(cfg_lines, vmid, vmtype): + def extract(field, default="-"): + return next((line.split(":", 1)[1].strip() for line in cfg_lines if line.startswith(f"{field}:")), default) + + name = extract("name", f"{vmtype}-{vmid}") + hostname = extract("hostname", name) + cores = extract("cores", "UNL") + memory = extract("memory", "0") + "M" + + ip = "-" + for line in cfg_lines: + if "ip=" in line: + ip = line.split("ip=")[1].split(',')[0] + break + + tags = "Yes" if any(line.startswith("tags:") for line in cfg_lines) else "No" + + storage_parts = [ + m.group(1) + for line in cfg_lines + if "none" not in line and any(d in line for d in ["scsi", "ide", "virtio", "sata", "mp", "rootfs"]) + if (m := re.search(r"size=([^, ]+)", line)) + ] + storage = "+".join(storage_parts) if storage_parts else "-" + + return { + 'ID': vmid, + 'Hostname': hostname, + 'IP': ip, + 'Typ': vmtype, + 'CPU': cores, + 'RAM': memory, + 'Storage': storage, + 'Tags': tags + } + +def output_table(rows, columns): + if not rows: + print("Keine Daten vorhanden.") + return + + # Dynamische Breite pro Spalte + col_widths = { + col: max(len(col), max(len(str(row.get(col, '-'))) for row in rows)) + for col in columns + } + + # Header + header = " | ".join(col.ljust(col_widths[col]) for col in columns) + separator = "-+-".join("-" * col_widths[col] for col in columns) + print(header) + print(separator) + + # Rows + for row in rows: + print(" | ".join(str(row.get(col, '-')).ljust(col_widths[col]) for col in columns)) + +def output_json(rows): + print(json.dumps(rows, indent=2)) + +def output_csv(rows, columns): + writer = csv.DictWriter(sys.stdout, fieldnames=columns) + writer.writeheader() + for row in rows: + writer.writerow({col: row.get(col, '') for col in columns}) + +def main(): + parser = argparse.ArgumentParser(description="Proxmox Inventory") + parser.add_argument("--local", action="store_true", help="Nur lokale Node abfragen") + parser.add_argument("--nodes", help="Komma-separierte Liste von Nodes") + parser.add_argument("--user", default="root", help="SSH Benutzer") + parser.add_argument("--key", help="Pfad zum SSH Key") + parser.add_argument("--columns", help="Welche Spalten anzeigen (Komma getrennt)") + parser.add_argument("--output", choices=["cli", "json", "csv"], default="cli", help="Ausgabeformat") + args = parser.parse_args() + + hosts = ["localhost"] if args.local else (args.nodes.split(",") if args.nodes else []) + if not hosts: + print("Fehler: --local oder --nodes muss angegeben werden") + sys.exit(1) + + use_ssh = not args.local + all_rows = [] + for host in hosts: + all_rows.extend(collect_node(host, args.user, use_ssh, args.key)) + + columns = args.columns.split(",") if args.columns else ['ID', 'Hostname', 'IP', 'Typ', 'CPU', 'RAM', 'Storage', 'Tags'] + + if args.output == "cli": + output_table(all_rows, columns) + elif args.output == "json": + output_json(all_rows) + elif args.output == "csv": + output_csv(all_rows, columns) + +if __name__ == "__main__": + main()