This repository has been archived on 2020-05-12. You can view files and clone it, but cannot push or open issues or pull requests.
HPCplaybooks/hosts.py

60 lines
1.5 KiB
Python
Raw Normal View History

2017-08-31 10:32:49 +02:00
#!/usr/bin/env python
import argparse
import json
import sys
def get_hosts(hosts_file='/etc/hosts'):
'''
Get the hostsnames from /etc/hosts.
Returns: A set of hostnames.
'''
rv = []
with open(hosts_file, 'r') as f:
for line in f:
if line == '\n':
continue
if line[0] == '#':
continue
rv.append(line.split()[1])
rv = set(rv)
ignore = {'localhost', 'ip6-allnodes', 'ip6-allrouters'}
return rv.difference(ignore)
def get_args(args_list):
"""
Parse the arguments and make sure only
that --list or --host is given, not both.
"""
parser = argparse.ArgumentParser(
description='ansible inventory script parsing /etc/hosts')
mutex_group = parser.add_mutually_exclusive_group(required=True)
help_list = 'list all hosts from /etc/hosts'
mutex_group.add_argument('--list', action='store_true', help=help_list)
help_host = 'display variables for a host'
mutex_group.add_argument('--host', help=help_host)
return parser.parse_args(args_list)
def main(args_list):
"""
Print a json list of the hosts if --list is given.
Does not support host vars.
Print an empty dictionary if --host is passed to remain valid.
"""
args = get_args(args_list)
if args.list:
print(json.dumps({
'all': {
'hosts': list(get_hosts()),
}
}))
if args.host:
print(json.dumps({}))
if __name__ == '__main__':
main(sys.argv[1:])