AWS custom grains

custom_aws_grains.py

#!/usr/bin/env python
# encoding: utf-8

# Standard Libs
import logging

# First Party Libs
import requests
from requests.adapters import HTTPAdapter


def ec2():
    """Returns custom grains from AWS EC2.
    """

    urls = {
        'id': 'http://169.254.169.254/latest/meta-data/instance-id',
        'ip': 'http://169.254.169.254/latest/meta-data/local-ipv4'
    }

    grains = {
        'ec2': {}
    }

    s = requests.Session()
    s.mount('http://169.254.169.254', HTTPAdapter(max_retries=1))

    for name, url in urls.iteritems():
        try:
            req = s.prepare_request(requests.Request('GET', url))
            rsp = s.send(req, timeout=0.5)
        except:
            logging.warn('Not able to get AWS Meta Data: {0}'.format(name))
        else:
            grains['ec2'][name] = rsp.text

    return grains