digital_ocean: whitespace changes
This commit is contained in:
parent
9b1a57f1f3
commit
98bbf9708a
1 changed files with 33 additions and 33 deletions
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue