mirror of
https://github.com/xcat2/confluent.git
synced 2025-09-26 12:08:13 +00:00
89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
#!/usr/bin/python3
|
|
import optparse
|
|
import signal
|
|
import sys
|
|
import os
|
|
|
|
try:
|
|
signal.signal(signal.SIGPIPE, signal.SIG_DFL)
|
|
except AttributeError:
|
|
pass
|
|
|
|
path = os.path.dirname(os.path.realpath(__file__))
|
|
path = os.path.realpath(os.path.join(path, '..', 'lib', 'python'))
|
|
if path.startswith('/opt'):
|
|
sys.path.append(path)
|
|
|
|
import confluent.client as client
|
|
import confluent.sortutil as sortutil
|
|
|
|
|
|
|
|
|
|
def main():
|
|
argparser = optparse.OptionParser(
|
|
usage='''\n %prog noderange -o ansible.hosts -a
|
|
''')
|
|
argparser.add_option('-o', '--output',
|
|
help='Writes an Ansible hosts file')
|
|
argparser.add_option('-a', '--append', action='store_true',
|
|
help='Appends to existing hosts file')
|
|
(options, args) = argparser.parse_args()
|
|
try:
|
|
noderange = args[0]
|
|
except IndexError:
|
|
argparser.print_help()
|
|
sys.exit(1)
|
|
if not options.output:
|
|
sys.stderr.write('Output file must be specified by -o\n')
|
|
sys.exit(1)
|
|
sess = client.Command()
|
|
databynode = {}
|
|
for res in sess.read('/noderange/{0}/attributes/all'.format(noderange)):
|
|
for node in res.get('databynode', {}):
|
|
if node not in databynode:
|
|
databynode[node] = {}
|
|
databynode[node].update(res['databynode'][node])
|
|
|
|
nodesbygroup = {}
|
|
for node in sortutil.natural_sort(databynode):
|
|
data = databynode[node]
|
|
groups = data.get('groups', [])
|
|
if not groups:
|
|
nodesbygroup.setdefault('', set()).add(node.strip().lower())
|
|
else:
|
|
for g in groups:
|
|
nodesbygroup.setdefault(g, set()).add(node.strip().lower())
|
|
existing_data = {}
|
|
if options.append and os.path.exists(options.output):
|
|
current_group = ''
|
|
with open(options.output, 'r') as f:
|
|
for line in f:
|
|
line = line.strip().lower()
|
|
if not line:
|
|
continue
|
|
if line.startswith('[') and line.endswith(']'):
|
|
current_group = line[1:-1]
|
|
existing_data.setdefault(current_group, set())
|
|
else:
|
|
existing_data.setdefault(current_group, set()).add(line)
|
|
|
|
for group, nodes in nodesbygroup.items():
|
|
nodes = {n.strip().lower() for n in nodes}
|
|
current_nodes = existing_data.get(group, set())
|
|
new_nodes = nodes - current_nodes
|
|
if new_nodes:
|
|
existing_data.setdefault(group, set()).update(nodes)
|
|
|
|
with open(options.output, 'w') as importfile:
|
|
for group in sortutil.natural_sort(existing_data.keys()):
|
|
if group:
|
|
importfile.write('[{0}]\n'.format(group))
|
|
for node in sortutil.natural_sort(existing_data[group]):
|
|
importfile.write('{0}\n'.format(node))
|
|
importfile.write('\n')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|