Allow with_items to specify a variable

This commit is contained in:
Daniel Hokka Zakrisson 2012-06-09 21:59:22 +02:00
parent 933b5e8631
commit a05da2d312
3 changed files with 11 additions and 3 deletions

View file

@ -89,6 +89,8 @@ class Play(object):
items = y.get('with_items',None) items = y.get('with_items',None)
if items is None: if items is None:
items = [ '' ] items = [ '' ]
elif isinstance(items, basestring):
items = utils.varLookup(items, task_vars)
for item in items: for item in items:
mv = task_vars.copy() mv = task_vars.copy()
mv['item'] = item mv['item'] = item

View file

@ -201,7 +201,7 @@ def parse_json(data):
_LISTRE = re.compile(r"(\w+)\[(\d+)\]") _LISTRE = re.compile(r"(\w+)\[(\d+)\]")
def varLookup(name, vars): def _varLookup(name, vars):
''' find the contents of a possibly complex variable in vars. ''' ''' find the contents of a possibly complex variable in vars. '''
path = name.split('.') path = name.split('.')
space = vars space = vars
@ -223,6 +223,12 @@ def varLookup(name, vars):
_KEYCRE = re.compile(r"\$(?P<complex>\{){0,1}((?(complex)[\w\.\[\]]+|\w+))(?(complex)\})") _KEYCRE = re.compile(r"\$(?P<complex>\{){0,1}((?(complex)[\w\.\[\]]+|\w+))(?(complex)\})")
# if { -> complex if complex, allow . and need trailing } # if { -> complex if complex, allow . and need trailing }
def varLookup(varname, vars):
m = _KEYCRE.search(varname)
if not m:
return None
return _varLookup(m.group(2), vars)
def varReplace(raw, vars): def varReplace(raw, vars):
'''Perform variable replacement of $vars '''Perform variable replacement of $vars
@ -245,7 +251,7 @@ def varReplace(raw, vars):
# original) # original)
varname = m.group(2) varname = m.group(2)
replacement = unicode(varLookup(varname, vars) or m.group()) replacement = unicode(_varLookup(varname, vars) or m.group())
start, end = m.span() start, end = m.span()
done.append(raw[:start]) # Keep stuff leading up to token done.append(raw[:start]) # Keep stuff leading up to token

View file

@ -17,7 +17,7 @@ class TestUtils(unittest.TestCase):
} }
} }
res = ansible.utils.varLookup('data.who', vars) res = ansible.utils._varLookup('data.who', vars)
assert sorted(res) == sorted(vars['data']['who']) assert sorted(res) == sorted(vars['data']['who'])