ansible/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2024-09-14 20:13:21 +02:00
Code Releases Activity

Address comments

Browse source
This commit is contained in:
Andre keedy 2016-02-04 16:58:10 -05:00
parent 083530d8fa
commit e46074c791
1 changed files with 31 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
contrib/inventory/rackhd.py
View file

@ -1,4 +1,5 @@
#!/usr/bin/python
import json
import requests
import os
@ -12,13 +13,8 @@ class RackhdInventory(object):
self._inventory = {}
for nodeid in nodeids:
self._load_inventory_data(nodeid)
output = '{\n'
for nodeid,info in self._inventory.iteritems():
output += self._format_output(nodeid, info)
output += ',\n'
output = output[:-2]
output += '}\n'
print (output)
print(json.dumps(self._format_output(nodeid, info)))
def _load_inventory_data(self, nodeid):
info = {}
@ -29,28 +25,29 @@ class RackhdInventory(object):
for key,url in info.iteritems():
r = requests.get( url, verify=False)
results[key] = r.text
self._inventory[nodeid] = results
def _format_output(self, nodeid, info):
output = ''
try:
node_info = json.loads(info['lookup'])
ipaddress = ''
if len(node_info) > 0:
ipaddress = node_info[0]["ipAddress"]
output += ' "' + nodeid + '" : {\n'
output += ' "hosts": [ "' + ipaddress + '" ],\n'
output += ' "vars" : {\n'
ipaddress = node_info[0]['ipAddress']
output = {nodeid:{ 'hosts':[ipaddress],'vars':{}}}
for key,result in info.iteritems():
output += ' "' + key + '": ' + json.dumps(json.loads(result), sort_keys=True, indent=2) + ',\n'
output += ' "ansible_ssh_user": "renasar"\n'
output += ' }\n'
output += ' }\n'
output[nodeid]['vars'][key] = json.loads(result)
output[nodeid]['vars']['ansible_ssh_user'] = 'monorail'
except KeyError:
pass
return output
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--host')
parser.add_argument('--list', action='store_true')
return parser.parse_args()
try:
#check if rackhd url(ie:10.1.1.45:8080) is specified in the environment
RACKHD_URL = 'http://' + str(os.environ['RACKHD_URL'])
@ -61,13 +58,21 @@ except:
# Use the nodeid specified in the environment to limit the data returned
# or return data for all available nodes
nodeids = []
try:
nodeids += os.environ['nodeid'].split(',')
except KeyError:
url = RACKHD_URL + '/api/common/nodes'
r = requests.get( url, verify=False)
data = json.loads(r.text)
for entry in data:
if entry['type'] == 'compute':
nodeids.append(entry['id'])
RackhdInventory(nodeids)
if (parse_args().host):
try:
nodeids += parse_args().host.split(',')
RackhdInventory(nodeids)
except:
pass
if (parse_args().list):
try:
url = RACKHD_URL + '/api/common/nodes'
r = requests.get( url, verify=False)
data = json.loads(r.text)
for entry in data:
if entry['type'] == 'compute':
nodeids.append(entry['id'])
RackhdInventory(nodeids)
except:
pass