Add an additional way to dereference a variable in a playbook, $foo

(Using varReplace function originally from yum, thanks Seth)
pull/70/head
Michael DeHaan 12 years ago
parent af9596307d
commit 5ed2b894d9

@ -20,6 +20,7 @@
import sys
import os
import shlex
import re
import jinja2
try:
@ -224,8 +225,41 @@ def parse_json(data):
return { "failed" : True, "parsed" : False, "msg" : data }
return results
_KEYCRE = re.compile(r"\$(\w+)")
def varReplace(raw, vars):
'''Perform variable replacement of $vars
@param raw: String to perform substitution on.
@param vars: Dictionary of variables to replace. Key is variable name
(without $ prefix). Value is replacement string.
@return: Input raw string with substituted values.
'''
# this code originally from yum
done = [] # Completed chunks to return
while raw:
m = _KEYCRE.search(raw)
if not m:
done.append(raw)
break
# Determine replacement value (if unknown variable then preserve
# original)
varname = m.group(1).lower()
replacement = vars.get(varname, m.group())
start, end = m.span()
done.append(raw[:start]) # Keep stuff leading up to token
done.append(replacement) # Append replacement value
raw = raw[end:] # Continue with remainder of string
return ''.join(done)
def template(text, vars):
''' run a text buffer through the templating engine '''
text = varReplace(text, vars)
template = jinja2.Template(text)
return template.render(vars)

@ -52,7 +52,7 @@
[
"task start",
[
"test basic shell",
"test basic shell, plus two ways to dereference a variable",
false
]
],
@ -61,10 +61,10 @@
[
"127.0.0.1",
{
"cmd": "echo $HOME ",
"cmd": "echo $HOME 5150 5150 ",
"rc": 0,
"stderr": "",
"stdout": "/root"
"stdout": "/root 5150 5150"
}
]
],
@ -200,4 +200,3 @@
}
}
}

@ -13,8 +13,8 @@
- name: test basic success command 2
action: command /bin/true
- name: test basic shell
action: shell echo $HOME
- name: test basic shell, plus two ways to dereference a variable
action: shell echo $HOME $port {{ port }}
# in the command below, the test file should contain a valid template
# and trigger the change handler

Loading…
Cancel
Save