diff options
author | Paul Durivage <paul.durivage@rackspace.com> | 2013-10-01 13:23:39 -0500 |
---|---|---|
committer | Paul Durivage <pauldurivage@gmail.com> | 2013-10-29 13:47:25 -0500 |
commit | 25c3bec6ef70627eba2ca6c0a397bbce2a075fc6 (patch) | |
tree | ec574db567d2070d6668ef42d8466ea96b52a423 /plugins/inventory/rax.py | |
parent | ccc607eaa5265ee6295ab74e427e07a765d47d93 (diff) | |
download | ansible-25c3bec6ef70627eba2ca6c0a397bbce2a075fc6.tar.gz |
Eliminate the RAX_REGION environment variable; iterate through all regions available to an account and present servers in region groups; default to using server name to identify servers in groups; set ansible_ssh_host key for each server; utilize the _meta key to provide all the hostvars for each server; all grouping by web, db, sql, lb, app based on server name; pretty print JSON output for some human-readable inventory action
Add set_setting to alleviate requirement that ~/.pyrax.cfg exists
Add regions as a optional env var to narrow our inventory scope
Clean up
Diffstat (limited to 'plugins/inventory/rax.py')
-rwxr-xr-x | plugins/inventory/rax.py | 181 |
1 files changed, 118 insertions, 63 deletions
diff --git a/plugins/inventory/rax.py b/plugins/inventory/rax.py index cbf781cfea..de5e5c1cd9 100755 --- a/plugins/inventory/rax.py +++ b/plugins/inventory/rax.py @@ -74,6 +74,7 @@ import sys import re import os import argparse +import collections try: import json @@ -86,69 +87,123 @@ except ImportError: print('pyrax required for this module') sys.exit(1) -# Setup the parser -parser = argparse.ArgumentParser(description='List active instances', - epilog='List by itself will list all the active \ - instances. Listing a specific instance will show \ - all the details about the instance.') -parser.add_argument('--list', action='store_true', default=True, - help='List active servers') -parser.add_argument('--host', - help='List details about the specific host (IP address)') - -args = parser.parse_args() - -# setup the auth -try: - creds_file = os.environ['RAX_CREDS_FILE'] - region = os.environ['RAX_REGION'] -except KeyError, e: - sys.stderr.write('Unable to load %s\n' % e.message) - sys.exit(1) - -pyrax.set_setting('identity_type', 'rackspace') - -try: - pyrax.set_setting("identity_type", "rackspace") - pyrax.set_credential_file(os.path.expanduser(creds_file), - region=region) -except Exception, e: - sys.stderr.write("%s: %s\n" % (e, e.message)) - sys.exit(1) - -# Execute the right stuff -if not args.host: - groups = {} - - # Cycle on servers - for server in pyrax.cloudservers.list(): - # Define group (or set to empty string) - try: - group = server.metadata['group'] - except KeyError: - group = 'undefined' - - # Create group if not exist and add the server - groups.setdefault(group, []).append(server.accessIPv4) - - # Return server list - print(json.dumps(groups)) +def host(hostname): + hostvars = {} + + for region in pyrax.regions: + # Connect to the region + cs = pyrax.connect_to_cloudservers(region=region) + for server in cs.servers.list(): + if server.name == hostname: + keys = [key for key in vars(server) if key not in ('manager', '_info')] + for key in keys: + # Extract value + value = getattr(server, key) + + # Generate sanitized key + key = 'rax_' + (re.sub("[^A-Za-z0-9\-]", "_", key) + .lower() + .lstrip("_")) + hostvars[key] = value + + # And finally, add an IP address + hostvars['ansible_ssh_host'] = server.accessIPv4 + print(json.dumps(hostvars, sort_keys=True, indent=4)) + + +def _list(region): + groups = collections.defaultdict(list) + hostvars = collections.defaultdict(dict) + + if region and region.upper() in pyrax.regions: + pyrax.regions = (region,) + + # Go through all the regions looking for servers + for region in pyrax.regions: + # Connect to the region + cs = pyrax.connect_to_cloudservers(region=region) + for server in cs.servers.list(): + # Create a group on region + groups[region].append(server.name) + + # Anything we can discern from the hostname? + try: + subdom = server.name.split('.')[0] + except IndexError: + pass + else: + for name in ('web', 'db', 'sql', 'lb', 'app'): + if name in subdom: + groups[name].append(server.name) + + # Check if group metadata key in servers' metadata + try: + group = server.metadata['group'] + except KeyError: + pass + else: + # Create group if not exist and add the server + groups[group].append(server.name) + + # Add host metadata + keys = [key for key in vars(server) if key not in ('manager', '_info')] + for key in keys: + # Extract value + value = getattr(server, key) + + # Generate sanitized key + key = 'rax_' + (re.sub("[^A-Za-z0-9\-]", "_", key) + .lower() + .lstrip('_')) + hostvars[server.name][key] = value + + # And finally, add an IP address + hostvars[server.name]['ansible_ssh_host'] = server.accessIPv4 + + if hostvars: + groups['_meta'] = {'hostvars': hostvars} + print(json.dumps(groups, sort_keys=True, indent=4)) + + +def parse_args(): + parser = argparse.ArgumentParser(description='Ansible Rackspace Cloud ' + 'inventory module') + group = parser.add_mutually_exclusive_group(required=True) + group.add_argument('--list', action='store_true', + help='List active servers') + group.add_argument('--host', help='List details about the specific host') + return parser.parse_args() + + +def setup(): + try: + creds_file = os.environ['RAX_CREDS_FILE'] + region = os.getenv('RAX_REGION') + except KeyError, e: + sys.stderr.write('Unable to load environment ' + 'variable %s\n' % e.message) + sys.exit(1) + + pyrax.set_setting('identity_type', 'rackspace') + + try: + pyrax.set_credential_file(os.path.expanduser(creds_file)) + except Exception, e: + sys.stderr.write("%s: %s\n" % (e, e.message)) + sys.exit(1) + + return region + + +def main(): + args = parse_args() + region = setup() + if args.list: + _list(region) + elif args.host: + host(args.host) sys.exit(0) -# Get the deets for the instance asked for -results = {} -# This should be only one, but loop anyway -for server in pyrax.cloudservers.list(): - if server.accessIPv4 == args.host: - for key in [key for key in vars(server) if - key not in ('manager', '_info')]: - # Extract value - value = getattr(server, key) - - # Generate sanitized key - key = 'rax_' + re.sub("[^A-Za-z0-9\-]", "_", key).lower() - results[key] = value - -print(json.dumps(results)) -sys.exit(0) +if __name__ == '__main__': + main() |