diff options
Diffstat (limited to 'bin/ansible-inventory')
| -rwxr-xr-x | bin/ansible-inventory | 100 |
1 files changed, 39 insertions, 61 deletions
diff --git a/bin/ansible-inventory b/bin/ansible-inventory index 3550079b..56c370cc 100755 --- a/bin/ansible-inventory +++ b/bin/ansible-inventory @@ -18,7 +18,7 @@ from ansible import constants as C from ansible import context from ansible.cli.arguments import option_helpers as opt_help from ansible.errors import AnsibleError, AnsibleOptionsError -from ansible.module_utils.common.text.converters import to_bytes, to_native, to_text +from ansible.module_utils._text import to_bytes, to_native, to_text from ansible.utils.vars import combine_vars from ansible.utils.display import Display from ansible.vars.plugins import get_vars_from_inventory_sources, get_vars_from_path @@ -72,6 +72,7 @@ class InventoryCLI(CLI): opt_help.add_runtask_options(self.parser) # remove unused default options + self.parser.add_argument('-l', '--limit', help=argparse.SUPPRESS, action=opt_help.UnrecognizedArgument, nargs='?') self.parser.add_argument('--list-hosts', help=argparse.SUPPRESS, action=opt_help.UnrecognizedArgument) self.parser.add_argument('args', metavar='host|group', nargs='?') @@ -79,10 +80,9 @@ class InventoryCLI(CLI): # Actions action_group = self.parser.add_argument_group("Actions", "One of following must be used on invocation, ONLY ONE!") action_group.add_argument("--list", action="store_true", default=False, dest='list', help='Output all hosts info, works as inventory script') - action_group.add_argument("--host", action="store", default=None, dest='host', - help='Output specific host info, works as inventory script. It will ignore limit') + action_group.add_argument("--host", action="store", default=None, dest='host', help='Output specific host info, works as inventory script') action_group.add_argument("--graph", action="store_true", default=False, dest='graph', - help='create inventory graph, if supplying pattern it must be a valid group name. It will ignore limit') + help='create inventory graph, if supplying pattern it must be a valid group name') self.parser.add_argument_group(action_group) # graph @@ -144,22 +144,17 @@ class InventoryCLI(CLI): # FIXME: should we template first? results = self.dump(myvars) - else: - if context.CLIARGS['subset']: - # not doing single host, set limit in general if given - self.inventory.subset(context.CLIARGS['subset']) - - if context.CLIARGS['graph']: - results = self.inventory_graph() - elif context.CLIARGS['list']: - top = self._get_group('all') - if context.CLIARGS['yaml']: - results = self.yaml_inventory(top) - elif context.CLIARGS['toml']: - results = self.toml_inventory(top) - else: - results = self.json_inventory(top) - results = self.dump(results) + elif context.CLIARGS['graph']: + results = self.inventory_graph() + elif context.CLIARGS['list']: + top = self._get_group('all') + if context.CLIARGS['yaml']: + results = self.yaml_inventory(top) + elif context.CLIARGS['toml']: + results = self.toml_inventory(top) + else: + results = self.json_inventory(top) + results = self.dump(results) if results: outfile = context.CLIARGS['output_file'] @@ -254,7 +249,7 @@ class InventoryCLI(CLI): return dump @staticmethod - def _remove_empty_keys(dump): + def _remove_empty(dump): # remove empty keys for x in ('hosts', 'vars', 'children'): if x in dump and not dump[x]: @@ -301,34 +296,33 @@ class InventoryCLI(CLI): def json_inventory(self, top): - seen_groups = set() + seen = set() - def format_group(group, available_hosts): + def format_group(group): results = {} results[group.name] = {} if group.name != 'all': - results[group.name]['hosts'] = [h.name for h in group.hosts if h.name in available_hosts] + results[group.name]['hosts'] = [h.name for h in group.hosts] results[group.name]['children'] = [] for subgroup in group.child_groups: results[group.name]['children'].append(subgroup.name) - if subgroup.name not in seen_groups: - results.update(format_group(subgroup, available_hosts)) - seen_groups.add(subgroup.name) + if subgroup.name not in seen: + results.update(format_group(subgroup)) + seen.add(subgroup.name) if context.CLIARGS['export']: results[group.name]['vars'] = self._get_group_variables(group) - self._remove_empty_keys(results[group.name]) - # remove empty groups + self._remove_empty(results[group.name]) if not results[group.name]: del results[group.name] return results - hosts = self.inventory.get_hosts(top.name) - results = format_group(top, frozenset(h.name for h in hosts)) + results = format_group(top) # populate meta results['_meta'] = {'hostvars': {}} + hosts = self.inventory.get_hosts() for host in hosts: hvars = self._get_host_variables(host) if hvars: @@ -338,10 +332,9 @@ class InventoryCLI(CLI): def yaml_inventory(self, top): - seen_hosts = set() - seen_groups = set() + seen = [] - def format_group(group, available_hosts): + def format_group(group): results = {} # initialize group + vars @@ -351,21 +344,15 @@ class InventoryCLI(CLI): results[group.name]['children'] = {} for subgroup in group.child_groups: if subgroup.name != 'all': - if subgroup.name in seen_groups: - results[group.name]['children'].update({subgroup.name: {}}) - else: - results[group.name]['children'].update(format_group(subgroup, available_hosts)) - seen_groups.add(subgroup.name) + results[group.name]['children'].update(format_group(subgroup)) # hosts for group results[group.name]['hosts'] = {} if group.name != 'all': for h in group.hosts: - if h.name not in available_hosts: - continue # observe limit myvars = {} - if h.name not in seen_hosts: # avoid defining host vars more than once - seen_hosts.add(h.name) + if h.name not in seen: # avoid defining host vars more than once + seen.append(h.name) myvars = self._get_host_variables(host=h) results[group.name]['hosts'][h.name] = myvars @@ -374,22 +361,17 @@ class InventoryCLI(CLI): if gvars: results[group.name]['vars'] = gvars - self._remove_empty_keys(results[group.name]) - # remove empty groups - if not results[group.name]: - del results[group.name] + self._remove_empty(results[group.name]) return results - available_hosts = frozenset(h.name for h in self.inventory.get_hosts(top.name)) - return format_group(top, available_hosts) + return format_group(top) def toml_inventory(self, top): - seen_hosts = set() - seen_hosts = set() + seen = set() has_ungrouped = bool(next(g.hosts for g in top.child_groups if g.name == 'ungrouped')) - def format_group(group, available_hosts): + def format_group(group): results = {} results[group.name] = {} @@ -399,14 +381,12 @@ class InventoryCLI(CLI): continue if group.name != 'all': results[group.name]['children'].append(subgroup.name) - results.update(format_group(subgroup, available_hosts)) + results.update(format_group(subgroup)) if group.name != 'all': for host in group.hosts: - if host.name not in available_hosts: - continue - if host.name not in seen_hosts: - seen_hosts.add(host.name) + if host.name not in seen: + seen.add(host.name) host_vars = self._get_host_variables(host=host) else: host_vars = {} @@ -418,15 +398,13 @@ class InventoryCLI(CLI): if context.CLIARGS['export']: results[group.name]['vars'] = self._get_group_variables(group) - self._remove_empty_keys(results[group.name]) - # remove empty groups + self._remove_empty(results[group.name]) if not results[group.name]: del results[group.name] return results - available_hosts = frozenset(h.name for h in self.inventory.get_hosts(top.name)) - results = format_group(top, available_hosts) + results = format_group(top) return results |