cmueller/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Get with_items to work with new play/task architecture.

Browse source
This commit is contained in:
Michael DeHaan 2012-05-26 01:49:23 -04:00
parent ecb944892d
commit 4664e354c3
2 changed files with 25 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
lib/ansible/playbook/play.py
View file

@ -80,13 +80,23 @@ class Play(object):
include_file = tokens[0]
data = utils.parse_yaml_from_file(utils.path_dwim(self.playbook.basedir, include_file))
for y in data:
items = y.get('with_items',None)
if items is None:
items = [ '' ]
for item in items:
t = Task(self,y)
# TODO: rename this to just 'vars'
t.module_vars = self.vars.copy()
t.module_vars.update(task_vars)
results.append(t)
mv = self.vars.copy()
mv.update(task_vars)
mv['item'] = item
results.append(Task(self,y,module_vars=mv))
elif type(x) == dict:
results.append(Task(self, x))
items = x.get('with_items', None)
if items is None:
items = [ '' ]
for item in items:
mv = self.vars.copy()
mv['item'] = item
results.append(Task(self,x,module_vars=mv))
else:
raise Exception("unexpected task type")
return results

12
lib/ansible/playbook/task.py
View file

@ -18,6 +18,7 @@
#############################################
from ansible import errors
from ansible import utils
class Task(object):
@ -26,10 +27,13 @@ class Task(object):
'notify', 'module_name', 'module_args', 'module_vars', 'play', 'notified_by',
]
def __init__(self, play, ds):
def __init__(self, play, ds, module_vars=None):
''' constructor loads from a task or handler datastructure '''
# TODO: more error handling
# include task specific vars
self.module_vars = module_vars
self.play = play
self.name = ds.get('name', None)
@ -55,8 +59,10 @@ class Task(object):
if len(tokens) > 1:
self.module_args = tokens[1]
# include task specific vars
self.module_vars = ds.get('vars', {})
self.name = utils.template(self.name, self.module_vars)
self.action = utils.template(self.name, self.module_vars)
if 'first_available_file' in ds:
self.module_vars['first_available_file'] = ds.get('first_available_file')