Created
March 24, 2017 12:08
-
-
Save coolflame87/a731754f99cb31f9e6f36cf9b13f6824 to your computer and use it in GitHub Desktop.
ec2metadata
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #! /usr/bin/python | |
| instdata_host = "169.254.169.254" | |
| instdata_ver = "2009-04-04" | |
| instdata_url = "http://%s/%s" % (instdata_host, instdata_ver) | |
| __doc__ = """ | |
| Query and display EC2 metadata. | |
| If no options are provided, all options will be displayed | |
| Options: | |
| -h --help show this help | |
| --kernel-id display the kernel id | |
| --ramdisk-id display the ramdisk id | |
| --reservation-id display the reservation id | |
| --ami-id display the ami id | |
| --ami-launch-index display the ami launch index | |
| --ami-manifest-path display the ami manifest path | |
| --ancestor-ami-ids display the ami ancestor id | |
| --product-codes display the ami associated product codes | |
| --availability-zone display the ami placement zone | |
| --instance-id display the instance id | |
| --instance-type display the instance type | |
| --local-hostname display the local hostname | |
| --public-hostname display the public hostname | |
| --local-ipv4 display the local ipv4 ip address | |
| --public-ipv4 display the public ipv4 ip address | |
| --block-device-mapping display the block device id | |
| --security-groups display the security groups | |
| --mac display the instance mac address | |
| --profile display the instance profile | |
| --instance-action display the instance-action | |
| --public-keys display the openssh public keys | |
| --user-data display the user data (not actually metadata) | |
| -u | --url URL use URL (default: %s) | |
| """ % instdata_url | |
| import sys | |
| import time | |
| import getopt | |
| import urllib2 | |
| import socket | |
| import urlparse | |
| METAOPTS = ['ami-id', 'ami-launch-index', 'ami-manifest-path', | |
| 'ancestor-ami-ids', 'availability-zone', 'block-device-mapping', | |
| 'instance-action', 'instance-id', 'instance-type', | |
| 'local-hostname', 'local-ipv4', 'kernel-id', 'mac', | |
| 'profile', 'product-codes', 'public-hostname', 'public-ipv4', | |
| 'public-keys', 'ramdisk-id', 'reserveration-id', 'security-groups', | |
| 'user-data'] | |
| class Error(Exception): | |
| pass | |
| class EC2Metadata: | |
| """Class for querying metadata from EC2""" | |
| def __init__(self, burl=instdata_url): | |
| self.burl = burl | |
| s = urlparse.urlsplit(burl) | |
| addr = s.netloc.split(":")[0] | |
| port = s.port | |
| if s.port is None: | |
| port = 80 | |
| if not self._test_connectivity(addr, port): | |
| raise Error("could not establish connection to: %s:%s" % | |
| (addr, port)) | |
| @staticmethod | |
| def _test_connectivity(addr, port): | |
| for i in range(6): | |
| s = socket.socket() | |
| try: | |
| s.connect((addr, port)) | |
| s.close() | |
| return True | |
| except socket.error, e: | |
| time.sleep(1) | |
| return False | |
| def _get(self, uri): | |
| url = "%s/%s" % (self.burl, uri) | |
| try: | |
| resp = urllib2.urlopen(urllib2.Request(url)) | |
| value = resp.read() | |
| except urllib2.HTTPError as e: | |
| if e.code == 404: | |
| return None | |
| # Eucalyptus may raise a 500 (Internal Server Error) | |
| if e.code == 500: | |
| return None | |
| raise | |
| return value | |
| def get(self, metaopt): | |
| """return value of metaopt""" | |
| if metaopt not in METAOPTS: | |
| raise Error('unknown metaopt', metaopt, METAOPTS) | |
| if metaopt == 'availability-zone': | |
| return self._get('meta-data/placement/availability-zone') | |
| if metaopt == 'public-keys': | |
| data = self._get('meta-data/public-keys') | |
| if data is None: | |
| return None | |
| keyids = [line.split('=')[0] for line in data.splitlines()] | |
| public_keys = [] | |
| for keyid in keyids: | |
| uri = 'meta-data/public-keys/%d/openssh-key' % int(keyid) | |
| public_keys.append(self._get(uri).rstrip()) | |
| return public_keys | |
| if metaopt == 'user-data': | |
| return self._get('user-data') | |
| return self._get('meta-data/' + metaopt) | |
| def get(metaopt): | |
| """primitive: return value of metaopt""" | |
| m = EC2Metadata() | |
| return m.get(metaopt) | |
| def display(metaopts, burl, prefix=False): | |
| """primitive: display metaopts (list) values with optional prefix""" | |
| m = EC2Metadata(burl) | |
| for metaopt in metaopts: | |
| value = m.get(metaopt) | |
| if not value: | |
| value = "unavailable" | |
| if prefix: | |
| print "%s: %s" % (metaopt, value) | |
| else: | |
| print value | |
| def usage(s=None): | |
| """display usage and exit""" | |
| if s: | |
| print >> sys.stderr, "Error:", s | |
| print >> sys.stderr, "Syntax: %s [options]" % sys.argv[0] | |
| print >> sys.stderr, __doc__ | |
| sys.exit(1) | |
| def main(): | |
| """handle cli options""" | |
| try: | |
| getopt_metaopts = METAOPTS[:] | |
| getopt_metaopts.append('help') | |
| getopt_metaopts.append('url=') | |
| opts, args = getopt.gnu_getopt(sys.argv[1:], "hu:", getopt_metaopts) | |
| except getopt.GetoptError, e: | |
| usage(e) | |
| burl = instdata_url | |
| metaopts = [] | |
| prefix = False | |
| for opt, val in opts: | |
| if opt in ('-h', '--help'): | |
| usage() | |
| if opt in ('-u', '--url'): | |
| burl = val | |
| continue | |
| metaopts.append(opt.replace('--', '')) | |
| if len(metaopts) == 0: | |
| prefix = True | |
| metaopts = METAOPTS | |
| display(metaopts, burl, prefix) | |
| if __name__ == "__main__": | |
| main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment