Inventory for servers in Google Cloud

google_inventory

#!/usr/bin/env python

import requests
import subprocess
import json

STAGE = 'production'

def use_internal(instance):
    if in_gce:
        return True
    return False

in_gce = True

try:
    requests.get('http://169.254.169.254/computeMetadata/v1/',timeout=2)
except:
    in_gce = False

groups = {}
instances = json.loads(subprocess.check_output(["gcloud","compute","instances","list","--format","json"]))

for instance in instances:
    if STAGE not in instance['tags']['items']:
        continue
    for item in instance['tags']['items']:
        if not groups.get(item):
            groups[item] = []
        if use_internal(instance):
            groups[item].append(instance['networkInterfaces'][0]['networkIP'])
        else:
            groups[item].append(instance['networkInterfaces'][0]['accessConfigs'][0]['natIP'])         

print json.dumps(groups, indent=4)