mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
Support nested variables in varReplace.
This commit is contained in:
parent
036b779188
commit
82144fd543
2 changed files with 171 additions and 3 deletions
|
@ -198,7 +198,19 @@ def parse_json(data):
|
|||
return { "failed" : True, "parsed" : False, "msg" : data }
|
||||
return results
|
||||
|
||||
_KEYCRE = re.compile(r"\$(\w+)")
|
||||
def varLookup(name, vars):
|
||||
''' find the contents of a possibly complex variable in vars. '''
|
||||
path = name.split('.')
|
||||
space = vars
|
||||
for part in path:
|
||||
if part in space:
|
||||
space = space[part]
|
||||
else:
|
||||
return
|
||||
return space
|
||||
|
||||
_KEYCRE = re.compile(r"\$(?P<complex>\{){0,1}((?(complex)[\w\.]+|\w+))(?(complex)\})")
|
||||
# if { -> complex if complex, allow . and need trailing }
|
||||
|
||||
def varReplace(raw, vars):
|
||||
'''Perform variable replacement of $vars
|
||||
|
@ -220,8 +232,9 @@ def varReplace(raw, vars):
|
|||
|
||||
# Determine replacement value (if unknown variable then preserve
|
||||
# original)
|
||||
varname = m.group(1).lower()
|
||||
replacement = str(vars.get(varname, m.group()))
|
||||
varname = m.group(2).lower()
|
||||
|
||||
replacement = str(varLookup(varname, vars) or m.group())
|
||||
|
||||
start, end = m.span()
|
||||
done.append(raw[:start]) # Keep stuff leading up to token
|
||||
|
|
155
test/TestUtils.py
Normal file
155
test/TestUtils.py
Normal file
|
@ -0,0 +1,155 @@
|
|||
import os
|
||||
import unittest
|
||||
|
||||
import ansible.utils
|
||||
|
||||
class TestUtils(unittest.TestCase):
|
||||
|
||||
#####################################
|
||||
### varLookup function tests
|
||||
|
||||
def test_varLookup_list(self):
|
||||
vars = {
|
||||
'data': {
|
||||
'who': ['joe', 'jack', 'jeff']
|
||||
}
|
||||
}
|
||||
|
||||
res = ansible.utils.varLookup('data.who', vars)
|
||||
|
||||
assert sorted(res) == sorted(vars['data']['who'])
|
||||
|
||||
#####################################
|
||||
### varReplace function tests
|
||||
|
||||
def test_varReplace_simple(self):
|
||||
template = 'hello $who'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello world'
|
||||
|
||||
def test_varReplace_multiple(self):
|
||||
template = '$what $who'
|
||||
vars = {
|
||||
'what': 'hello',
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello world'
|
||||
|
||||
def test_varReplace_middle(self):
|
||||
template = 'hello $who!'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello world!'
|
||||
|
||||
def test_varReplace_alternative(self):
|
||||
template = 'hello ${who}'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello world'
|
||||
|
||||
def test_varReplace_almost_alternative(self):
|
||||
template = 'hello $who}'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello world}'
|
||||
|
||||
def test_varReplace_almost_alternative2(self):
|
||||
template = 'hello ${who'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == template
|
||||
|
||||
def test_varReplace_alternative_greed(self):
|
||||
template = 'hello ${who} }'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello world }'
|
||||
|
||||
def test_varReplace_notcomplex(self):
|
||||
template = 'hello $mydata.who'
|
||||
vars = {
|
||||
'data': {
|
||||
'who': 'world',
|
||||
},
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
print res
|
||||
assert res == template
|
||||
|
||||
def test_varReplace_nested(self):
|
||||
template = 'hello ${data.who}'
|
||||
vars = {
|
||||
'data': {
|
||||
'who': 'world'
|
||||
},
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello world'
|
||||
|
||||
def test_varReplace_nested_int(self):
|
||||
template = '$what ${data.who}'
|
||||
vars = {
|
||||
'data': {
|
||||
'who': 2
|
||||
},
|
||||
'what': 'hello',
|
||||
}
|
||||
|
||||
res = ansible.utils.varReplace(template, vars)
|
||||
|
||||
assert res == 'hello 2'
|
||||
|
||||
#####################################
|
||||
### Template function tests
|
||||
|
||||
def test_template_basic(self):
|
||||
template = 'hello {{ who }}'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.template(template, vars, {})
|
||||
|
||||
assert res == 'hello world'
|
||||
|
||||
def test_template_whitespace(self):
|
||||
template = 'hello {{ who }}\n'
|
||||
vars = {
|
||||
'who': 'world',
|
||||
}
|
||||
|
||||
res = ansible.utils.template(template, vars, {})
|
||||
|
||||
assert res == 'hello world\n'
|
Loading…
Reference in a new issue