From e02875274d2ef960d5ba237507ae2d2a5fda9871 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Soko=C5=82owski?= <jakub@status.im>
Date: Wed, 9 Feb 2022 19:57:35 +0100
Subject: [PATCH] ansible: add stats.py to compile counts of hosts
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Jakub Sokołowski <jakub@status.im>
---
 ansible/stats.py | 63 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100755 ansible/stats.py

diff --git a/ansible/stats.py b/ansible/stats.py
new file mode 100755
index 0000000..4779950
--- /dev/null
+++ b/ansible/stats.py
@@ -0,0 +1,63 @@
+#!/usr/bin/env python
+import sys
+import json
+from requests import get
+from optparse import OptionParser
+
+HELP_DESCRIPTION = '''
+Simple script to output counts of hosts in all fleets and all DCs.
+'''.strip()
+HELP_EXAMPLE = '''
+Example: ./stats.py -u http://consul.example.org:8400
+'''
+
+
+def parse_opts():
+    parser = OptionParser(description=HELP_DESCRIPTION, epilog=HELP_EXAMPLE)
+    parser.add_option('-u', '--consul-url', default='http://localhost:8500',
+                      help='Name of virtual network interface.')
+
+    return parser.parse_args()
+
+def main():
+    (opts, args) = parse_opts()
+
+    nodes = []
+    envs = {}
+    fleets = {}
+    dcs = {}
+
+    data_centers = get('%s/v1/catalog/datacenters' % opts.consul_url)
+    for dc in data_centers.json():
+        node = get('%s/v1/catalog/nodes?dc=%s' % (opts.consul_url,dc))
+        nodes.extend(node.json())
+
+    for node in nodes:
+        meta = node['Meta']
+        fleet_name = '%s.%s' % (meta['env'], meta['stage'])
+
+
+        # Fix for inconsistency in fleet naming
+        if meta['stage'] in ['hq', 'misc', 'office', 'bi', 'ci']:
+            env = envs.setdefault(meta['stage'], [])
+        else:
+            env = envs.setdefault(meta['env'], [])
+        dc = dcs.setdefault(node['Datacenter'], [])
+        fleet = fleets.setdefault(fleet_name, [])
+
+        dc.append(node['Node'])
+        fleet.append(node['Node'])
+        env.append(node['Node'])
+
+    out = {
+        'total': len(nodes),
+        'dcs': {dc: len(hosts) for dc, hosts in dcs.items()},
+        'envs': {env: len(hosts) for env, hosts in envs.items()},
+        #'fleets': {fleet: len(hosts) for fleet, hosts in fleets.items()},
+    }
+
+    print(json.dumps(out, indent=2))
+
+
+if __name__ == '__main__':
+    main()