From 3501200c8c2fc425ba83bbebe7b69fd25cc97163 Mon Sep 17 00:00:00 2001
From: alex <alex@pdp7.net>
Date: Sun, 18 Feb 2024 12:00:53 +0100
Subject: [PATCH] Add ansible_facts Python lib

---
 personal_infra/ansible_facts.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 personal_infra/ansible_facts.py

diff --git a/personal_infra/ansible_facts.py b/personal_infra/ansible_facts.py
new file mode 100644
index 0000000..02a744a
--- /dev/null
+++ b/personal_infra/ansible_facts.py
@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+import collections
+import json
+import subprocess
+
+
+def setup(pattern, setup_options):
+    res = dict()
+    out = subprocess.run(["ansible", "-m", "setup", pattern, "-a", setup_options], env={"ANSIBLE_LOAD_CALLBACK_PLUGINS": "true", "ANSIBLE_STDOUT_CALLBACK": "ansible.posix.jsonl"}, check=True, stdout=subprocess.PIPE).stdout
+    out = out.decode("utf8").split("\n")[0:-4]
+    out = [json.loads(l) for l in out if l.startswith("{")]
+    out = [j for j in out if "hosts" in j and j["hosts"]]
+    for j in out:
+        assert len(j["hosts"]) == 1
+        k, v = list(j["hosts"].items())[0]
+        res[k] = v["ansible_facts"]
+    return res
+
+
+def distros():
+    res = collections.defaultdict(list)
+    for k, v in setup("!k8s", "gather_subset=all").items():
+        res[f'{v["ansible_distribution_file_variety"]} {v["ansible_distribution_major_version"]}'].append(k)
+    return res
-- 
2.47.3