Add inventory for working with broker

Signed-off-by: Eric D. Helms <ericdhelms@gmail.com>

Author: ehelms

inventories/broker.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python3
+
+import argparse
+import json
+import yaml
+import os
+import subprocess
+import sys
+from io import StringIO
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(description="Broker inventory script")
+    group = parser.add_mutually_exclusive_group(required=True)
+    group.add_argument('--list', action='store_true')
+    group.add_argument('--host')
+    return parser.parse_args()
+
+
+def get_running_hosts():
+    cmd = ["broker", "inventory", "--details"]
+
+    try:
+        output = subprocess.check_output(cmd, universal_newlines=True).rstrip()
+    except FileNotFoundError:
+        return
+
+    hosts = yaml.safe_load(output)
+    return hosts.values()
+
+
+def list_running_hosts():
+    hosts = get_running_hosts()
+    variables = dict(get_configs(hosts))
+
+    return {
+        "_meta": {
+            "hostvars": variables
+        },
+        "all": {
+            "hosts": list(variables.keys())
+        },
+    }
+
+
+def get_configs(hosts):
+    if not hosts:
+        return
+
+    for host in hosts:
+        details = {
+            'ansible_host': host['ip'],
+            'ansible_port': '22',
+            'ansible_user': 'root'
+        }
+        yield host['hostname'], details
+
+
+def main():
+    args = parse_args()
+    hosts = list_running_hosts()
+
+    if args.list:
+        json.dump(hosts, sys.stdout)
+    elif args.host:
+        details = hosts['_meta']['hostvars']
+        json.dump(details[args.host], sys.stdout)
+
+
+if __name__ == '__main__':
+    main()