Introduce per task variables and push them to templates.

This commit is contained in:
Jeroen Hoekx 2012-04-04 13:38:21 +02:00
parent edd5baad8b
commit ab86726a15
3 changed files with 38 additions and 8 deletions

View file

@ -113,14 +113,19 @@ class PlayBook(object):
# include: some.yml a=2 b=3 c=4 # include: some.yml a=2 b=3 c=4
include_tokens = task['include'].split() include_tokens = task['include'].split()
path = utils.path_dwim(dirname, include_tokens[0]) path = utils.path_dwim(dirname, include_tokens[0])
inject_vars = self._get_vars(play, dirname) play_vars = self._get_vars(play, dirname)
include_vars = {}
for i,x in enumerate(include_tokens): for i,x in enumerate(include_tokens):
if x.find("=") != -1: if x.find("=") != -1:
(k,v) = x.split("=") (k,v) = x.split("=")
inject_vars[k] = v include_vars[k] = v
inject_vars = play_vars.copy()
inject_vars.update(include_vars)
included = utils.template_from_file(path, inject_vars) included = utils.template_from_file(path, inject_vars)
included = utils.parse_yaml(included) included = utils.parse_yaml(included)
for x in included: for x in included:
if len(include_vars):
x["vars"] = include_vars
new_tasks.append(x) new_tasks.append(x)
# ***************************************************** # *****************************************************
@ -266,7 +271,7 @@ class PlayBook(object):
# ***************************************************** # *****************************************************
def _run_module(self, pattern, host_list, module, args, remote_user, def _run_module(self, pattern, host_list, module, args, vars, remote_user,
async_seconds, async_poll_interval, only_if, sudo): async_seconds, async_poll_interval, only_if, sudo):
''' run a particular module step in a playbook ''' ''' run a particular module step in a playbook '''
@ -277,7 +282,7 @@ class PlayBook(object):
module_args=args, host_list=hosts, forks=self.forks, module_args=args, host_list=hosts, forks=self.forks,
remote_pass=self.remote_pass, module_path=self.module_path, remote_pass=self.remote_pass, module_path=self.module_path,
timeout=self.timeout, remote_user=remote_user, timeout=self.timeout, remote_user=remote_user,
remote_port=self.remote_port, remote_port=self.remote_port, module_vars=vars,
setup_cache=SETUP_CACHE, basedir=self.basedir, setup_cache=SETUP_CACHE, basedir=self.basedir,
conditional=only_if, callbacks=self.runner_callbacks, conditional=only_if, callbacks=self.runner_callbacks,
extra_vars=self.extra_vars, debug=self.debug, sudo=sudo extra_vars=self.extra_vars, debug=self.debug, sudo=sudo
@ -309,6 +314,9 @@ class PlayBook(object):
module_name = tokens[0] module_name = tokens[0]
module_args = tokens[1] module_args = tokens[1]
# include task specific vars
module_vars = task.get('vars')
# tasks can be direct (run on all nodes matching # tasks can be direct (run on all nodes matching
# the pattern) or conditional, where they ran # the pattern) or conditional, where they ran
# as the result of a change handler on a subset # as the result of a change handler on a subset
@ -319,7 +327,7 @@ class PlayBook(object):
# load up an appropriate ansible runner to # load up an appropriate ansible runner to
# run the task in parallel # run the task in parallel
results = self._run_module(pattern, host_list, module_name, results = self._run_module(pattern, host_list, module_name,
module_args, remote_user, async_seconds, module_args, module_vars, remote_user, async_seconds,
async_poll_interval, only_if, sudo) async_poll_interval, only_if, sudo)
self.stats.compute(results) self.stats.compute(results)

View file

@ -74,7 +74,7 @@ class Runner(object):
remote_user=C.DEFAULT_REMOTE_USER, remote_pass=C.DEFAULT_REMOTE_PASS, remote_user=C.DEFAULT_REMOTE_USER, remote_pass=C.DEFAULT_REMOTE_PASS,
remote_port=C.DEFAULT_REMOTE_PORT, background=0, basedir=None, setup_cache=None, remote_port=C.DEFAULT_REMOTE_PORT, background=0, basedir=None, setup_cache=None,
transport='paramiko', conditional='True', groups={}, callbacks=None, verbose=False, transport='paramiko', conditional='True', groups={}, callbacks=None, verbose=False,
debug=False, sudo=False, extra_vars=None): debug=False, sudo=False, extra_vars=None, module_vars=None):
if setup_cache is None: if setup_cache is None:
setup_cache = {} setup_cache = {}
@ -101,6 +101,7 @@ class Runner(object):
self.forks = int(forks) self.forks = int(forks)
self.pattern = pattern self.pattern = pattern
self.module_args = module_args self.module_args = module_args
self.module_vars = module_vars
self.extra_vars = extra_vars self.extra_vars = extra_vars
self.timeout = timeout self.timeout = timeout
self.debug = debug self.debug = debug
@ -498,8 +499,16 @@ class Runner(object):
# install the template module # install the template module
template_module = self._transfer_module(conn, tmp, 'template') template_module = self._transfer_module(conn, tmp, 'template')
# transfer module vars
if self.module_vars:
vars = utils.bigjson(self.module_vars)
vars_path = self._transfer_str(conn, tmp, 'module_vars', vars)
vars_arg=" vars=%s"%(vars_path)
else:
vars_arg=""
# run the template module # run the template module
args = "src=%s dest=%s metadata=%s" % (temppath, dest, metadata) args = "src=%s dest=%s metadata=%s%s" % (temppath, dest, metadata, vars_arg)
(result1, err, executed) = self._execute_module(conn, tmp, template_module, args) (result1, err, executed) = self._execute_module(conn, tmp, template_module, args)
(host, ok, data, err) = self._return_from_module(conn, host, result1, err, executed) (host, ok, data, err) = self._return_from_module(conn, host, result1, err, executed)

View file

@ -48,7 +48,7 @@ for x in items:
source = params['src'] source = params['src']
dest = params['dest'] dest = params['dest']
metadata = params.get('metadata', '/etc/ansible/setup') metadata = params.get('metadata', '/etc/ansible/setup')
module_vars = params.get('vars')
# raise an error if there is no template metadata # raise an error if there is no template metadata
if not os.path.exists(metadata): if not os.path.exists(metadata):
@ -71,6 +71,19 @@ except:
}) })
sys.exit(1) sys.exit(1)
if module_vars:
try:
f = open(module_vars)
vars = json.loads(f.read())
data.update(vars)
f.close()
except:
print json.dumps({
"failed" : 1,
"msg" : "Failed to parse/load %s." % module_vars
})
sys.exit(1)
if not os.path.exists(source): if not os.path.exists(source):
print json.dumps({ print json.dumps({
"failed" : 1, "failed" : 1,