cmueller/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

digital_ocean: whitespace changes

Browse source
This commit is contained in:
Evan Wies 2013-07-05 10:38:04 -04:00
parent 9b1a57f1f3
commit 98bbf9708a
1 changed files with 33 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

66
plugins/inventory/digital_ocean.py
View file

@ -201,7 +201,7 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
self.load_droplets_from_digital_ocean() self.load_droplets_from_digital_ocean()
# Pick the json_data to print based on the CLI command # Pick the json_data to print based on the CLI command
if self.args.droplets: json_data = { 'droplets': self.data['droplets'] } if self.args.droplets: json_data = { 'droplets': self.data['droplets'] }
elif self.args.regions: json_data = { 'regions': self.data['regions'] } elif self.args.regions: json_data = { 'regions': self.data['regions'] }
elif self.args.images: json_data = { 'images': self.data['images'] } elif self.args.images: json_data = { 'images': self.data['images'] }
elif self.args.sizes: json_data = { 'sizes': self.data['sizes'] } elif self.args.sizes: json_data = { 'sizes': self.data['sizes'] }
@ -214,9 +214,9 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
json_data = self.inventory json_data = self.inventory
if self.args.pretty: if self.args.pretty:
print json.dumps( json_data, sort_keys=True, indent=2 ) print json.dumps(json_data, sort_keys=True, indent=2)
else: else:
print json.dumps( json_data ) print json.dumps(json_data)
# That's all she wrote... # That's all she wrote...
@ -295,22 +295,22 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
def load_all_data_from_digital_ocean(self): def load_all_data_from_digital_ocean(self):
''' Use dopy to get all the information from DigitalOcean and save data in cache files ''' ''' Use dopy to get all the information from DigitalOcean and save data in cache files '''
manager = DoManager( self.client_id, self.api_key ) manager = DoManager(self.client_id, self.api_key)
self.data = {} self.data = {}
self.data['droplets'] = self.sanitize_list( manager.all_active_droplets() ) self.data['droplets'] = self.sanitize_list(manager.all_active_droplets())
self.data['regions'] = self.sanitize_list( manager.all_regions() ) self.data['regions'] = self.sanitize_list(manager.all_regions())
self.data['images'] = self.sanitize_list( manager.all_images(filter=None) ) self.data['images'] = self.sanitize_list(manager.all_images(filter=None))
self.data['sizes'] = self.sanitize_list( manager.sizes() ) self.data['sizes'] = self.sanitize_list(manager.sizes())
self.data['ssh_keys'] = self.sanitize_list( manager.all_ssh_keys() ) self.data['ssh_keys'] = self.sanitize_list(manager.all_ssh_keys())
self.data['domains'] = self.sanitize_list( manager.all_domains() ) self.data['domains'] = self.sanitize_list(manager.all_domains())
self.index = {} self.index = {}
self.index['region_to_name'] = self.build_index( self.data['regions'], 'id', 'name' ) self.index['region_to_name'] = self.build_index(self.data['regions'], 'id', 'name')
self.index['size_to_name'] = self.build_index( self.data['sizes'], 'id', 'name' ) self.index['size_to_name'] = self.build_index(self.data['sizes'], 'id', 'name')
self.index['image_to_name'] = self.build_index( self.data['images'], 'id', 'name' ) self.index['image_to_name'] = self.build_index(self.data['images'], 'id', 'name')
self.index['image_to_distro'] = self.build_index( self.data['images'], 'id', 'distribution' ) self.index['image_to_distro'] = self.build_index(self.data['images'], 'id', 'distribution')
self.index['host_to_droplet'] = self.build_index( self.data['droplets'], 'ip_address', 'id', False ) self.index['host_to_droplet'] = self.build_index(self.data['droplets'], 'ip_address', 'id', False)
self.build_inventory() self.build_inventory()
@ -319,9 +319,9 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
def load_droplets_from_digital_ocean(self): def load_droplets_from_digital_ocean(self):
''' Use dopy to get droplet information from DigitalOcean and save data in cache files ''' ''' Use dopy to get droplet information from DigitalOcean and save data in cache files '''
manager = DoManager( self.client_id, self.api_key ) manager = DoManager(self.client_id, self.api_key)
self.data['droplets'] = self.sanitize_list( manager.all_active_droplets() ) self.data['droplets'] = self.sanitize_list(manager.all_active_droplets())
self.index['host_to_droplet'] = self.build_index( self.data['droplets'], 'ip_address', 'id', False ) self.index['host_to_droplet'] = self.build_index(self.data['droplets'], 'ip_address', 'id', False)
self.build_inventory() self.build_inventory()
self.write_to_cache() self.write_to_cache()
@ -344,23 +344,23 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
dest = droplet['ip_address'] dest = droplet['ip_address']
self.inventory[droplet['id']] = [dest] self.inventory[droplet['id']] = [dest]
self.push( self.inventory, droplet['name'], dest ) self.push(self.inventory, droplet['name'], dest)
self.push( self.inventory, 'region_'+droplet['region_id'], dest ) self.push(self.inventory, 'region_'+droplet['region_id'], dest)
self.push( self.inventory, 'image_' +droplet['image_id'], dest ) self.push(self.inventory, 'image_' +droplet['image_id'], dest)
self.push( self.inventory, 'size_' +droplet['size_id'], dest ) self.push(self.inventory, 'size_' +droplet['size_id'], dest)
self.push( self.inventory, 'status_'+droplet['status'], dest ) self.push(self.inventory, 'status_'+droplet['status'], dest)
region_name = self.index['region_to_name'][droplet['region_id']] region_name = self.index['region_to_name'][droplet['region_id']]
self.push( self.inventory, 'region_'+region_name, dest ) self.push(self.inventory, 'region_'+region_name, dest)
size_name = self.index['size_to_name'][droplet['size_id']] size_name = self.index['size_to_name'][droplet['size_id']]
self.push( self.inventory, 'size_'+size_name, dest ) self.push(self.inventory, 'size_'+size_name, dest)
image_name = self.index['image_to_name'][droplet['image_id']] image_name = self.index['image_to_name'][droplet['image_id']]
self.push( self.inventory, 'image_'+image_name, dest ) self.push(self.inventory, 'image_'+image_name, dest)
distro_name = self.index['image_to_distro'][droplet['image_id']] distro_name = self.index['image_to_distro'][droplet['image_id']]
self.push( self.inventory, 'distro_'+distro_name, dest ) self.push(self.inventory, 'distro_'+distro_name, dest)
def load_droplet_variables_for_host(self): def load_droplet_variables_for_host(self):
@ -379,13 +379,13 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
if self.cache_refreshed: if self.cache_refreshed:
for drop in self.data['droplets']: for drop in self.data['droplets']:
if drop['ip_address'] == host: if drop['ip_address'] == host:
droplet = self.sanitize_dict( drop ) droplet = self.sanitize_dict(drop)
break break
else: else:
# Cache wasn't refreshed this run, so hit DigitalOcean API # Cache wasn't refreshed this run, so hit DigitalOcean API
manager = DoManager( self.client_id, self.api_key ) manager = DoManager(self.client_id, self.api_key)
droplet_id = self.index['host_to_droplet'][host] droplet_id = self.index['host_to_droplet'][host]
droplet = self.sanitize_dict( manager.show_droplet( droplet_id ) ) droplet = self.sanitize_dict(manager.show_droplet(droplet_id))
if not droplet: if not droplet:
return {} return {}
@ -427,7 +427,7 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
cache = open(self.cache_filename, 'r') cache = open(self.cache_filename, 'r')
json_data = cache.read() json_data = cache.read()
cache.close() cache.close()
data = json.loads( json_data ) data = json.loads(json_data)
self.data = data['data'] self.data = data['data']
self.inventory = data['inventory'] self.inventory = data['inventory']
@ -437,7 +437,7 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
def write_to_cache(self): def write_to_cache(self):
''' Writes data in JSON format to a file ''' ''' Writes data in JSON format to a file '''
data = { 'data': self.data, 'index': self.index, 'inventory': self.inventory } data = { 'data': self.data, 'index': self.index, 'inventory': self.inventory }
json_data = json.dumps( data, sort_keys=True, indent=2 ) json_data = json.dumps(data, sort_keys=True, indent=2)
cache = open(self.cache_filename, 'w') cache = open(self.cache_filename, 'w')
cache.write(json_data) cache.write(json_data)
@ -473,7 +473,7 @@ or environment variables (DO_CLIENT_ID and DO_API_KEY)'''
def sanitize_list(self, seq): def sanitize_list(self, seq):
new_seq = [] new_seq = []
for d in seq: for d in seq:
new_seq.append( self.sanitize_dict(d) ) new_seq.append(self.sanitize_dict(d))
return new_seq return new_seq