openstack/taskflow
Code Issues Proposed changes

Merge "Have reset state handlers go through a shared list"

Browse Source
This commit is contained in:
Jenkins 2015-04-17 06:33:31 +00:00 committed by Gerrit Code Review
commit 4ab27f54b4
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
taskflow/engines/action_engine/runtime.py
View File

@ -98,18 +98,23 @@ class Runtime(object):
def reset_nodes(self, nodes, state=st.PENDING, intention=st.EXECUTE):
tweaked = []
node_state_handlers = [
(self.task_action, {'progress': 0.0}),
(self.retry_action, {}),
]
for node in nodes:
if state or intention:
tweaked.append((node, state, intention))
if state:
if self.task_action.handles(node):
self.task_action.change_state(node, state,
progress=0.0)
elif self.retry_action.handles(node):
self.retry_action.change_state(node, state)
else:
raise TypeError("Unknown how to reset atom '%s' (%s)"
% (node, type(node)))
handled = False
for h, kwargs in node_state_handlers:
if h.handles(node):
h.change_state(node, state, **kwargs)
handled = True
break
if not handled:
raise TypeError("Unknown how to reset state of"
" node '%s' (%s)" % (node, type(node)))
if intention:
self.storage.set_atom_intention(node.name, intention)
return tweaked