From 39393482865201532da778f103aef35d35c38026 Mon Sep 17 00:00:00 2001 From: Toshio Kuratomi Date: Fri, 31 Jul 2015 12:47:54 -0700 Subject: [PATCH] Fix HostVars to support containment tests --- lib/ansible/vars/hostvars.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/lib/ansible/vars/hostvars.py b/lib/ansible/vars/hostvars.py index 29d1e1aa806..766efb5ed3b 100644 --- a/lib/ansible/vars/hostvars.py +++ b/lib/ansible/vars/hostvars.py @@ -19,13 +19,16 @@ from __future__ import (absolute_import, division, print_function) __metaclass__ = type +import collections + from jinja2 import Undefined as j2undefined from ansible.template import Templar __all__ = ['HostVars'] -class HostVars(dict): +# Note -- this is a Mapping, not a MutableMapping +class HostVars(collections.Mapping): ''' A special view of vars_cache that adds values from the inventory when needed. ''' def __init__(self, vars_manager, play, inventory, loader): @@ -36,7 +39,7 @@ class HostVars(dict): self._lookup = {} def __getitem__(self, host_name): - + if host_name not in self._lookup: host = self._inventory.get_host(host_name) if not host: @@ -46,3 +49,13 @@ class HostVars(dict): self._lookup[host_name] = templar.template(result, fail_on_undefined=False) return self._lookup[host_name] + def __contains__(self, host_name): + item = self.get(host_name) + if item and item is not j2undefined: + return True + return False + def __iter__(self): + raise NotImplementedError('HostVars does not support iteration as hosts are discovered on an as needed basis.') + + def __len__(self): + raise NotImplementedError('HostVars does not support len. hosts entries are discovered dynamically as needed')