get_host_names.py

#!/usr/bin/env python

import sys
import requests


def get_info(terrible_key, cluster, product=None):
    if product:
        url = 'https://terrible.sanomaservices.nl/env/list/{}/{}'.format(cluster, product)
    else:
        url = 'https://terrible.sanomaservices.nl/env/list/{}'.format(cluster)

    return requests.get(url, headers={'terrible-key': terrible_key})


def print_hostnames(environment_info):
    for env in environment_info:
        if 'Ec2' in env:
            for machine in env.get('Ec2', []):
                print('Host {}-{}-{}'.format(
                    env['InfraJSON'].get('Product'),
                    env['InfraJSON'].get('Environment'),
                    machine['Name'])
                )
                print('    Hostname {}'.format(machine['Ip']))


if __name__ == "__main__":
    if len(sys.argv) == 4:
        info = get_info(sys.argv[1], sys.argv[2], sys.argv[3])
    elif len(sys.argv) == 3:
        info = get_info(sys.argv[1], sys.argv[2])
    else:
        print('Generate ssh config hostnames for terrible')
        print('')
        print('Usage: ./get_host_names.py <terrible_key> <cluster> <product>')
        print('')
        print('example: ./get_host_name.py ABCD1234 cluster product >> ~/.ssh/config')
        sys.exit(2)

    if info.status_code == 200:
        print_hostnames(info.json())