mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-05 07:31:49 -05:00
73 lines
2.9 KiB
Python
73 lines
2.9 KiB
Python
|
|
# AWX
|
|
from awx.main.scheduler.dag_simple import SimpleDAG
|
|
|
|
class WorkflowDAG(SimpleDAG):
|
|
def __init__(self, workflow_job=None):
|
|
super(WorkflowDAG, self).__init__()
|
|
if workflow_job:
|
|
self._init_graph(workflow_job)
|
|
|
|
def _init_graph(self, workflow_job):
|
|
workflow_nodes = workflow_job.workflow_job_nodes.all()
|
|
for workflow_node in workflow_nodes:
|
|
self.add_node(workflow_node)
|
|
|
|
for node_type in ['success_nodes', 'failure_nodes', 'always_nodes']:
|
|
for workflow_node in workflow_nodes:
|
|
related_nodes = getattr(workflow_node, node_type).all()
|
|
for related_node in related_nodes:
|
|
self.add_edge(workflow_node, related_node, node_type)
|
|
|
|
def bfs_nodes_to_run(self):
|
|
root_nodes = self.get_root_nodes()
|
|
nodes = root_nodes
|
|
nodes_found = []
|
|
|
|
for index, n in enumerate(nodes):
|
|
obj = n['node_object']
|
|
job = obj.job
|
|
|
|
if not job:
|
|
nodes_found.append(n)
|
|
# Job is about to run or is running. Hold our horses and wait for
|
|
# the job to finish. We can't proceed down the graph path until we
|
|
# have the job result.
|
|
elif job.status not in ['failed', 'error', 'successful']:
|
|
continue
|
|
elif job.status in ['failed', 'error']:
|
|
children_failed = self.get_dependencies(obj, 'failure_nodes')
|
|
children_always = self.get_dependencies(obj, 'always_nodes')
|
|
children_all = children_failed + children_always
|
|
nodes.extend(children_all)
|
|
elif job.status in ['successful']:
|
|
children_success = self.get_dependencies(obj, 'success_nodes')
|
|
nodes.extend(children_success)
|
|
return [n['node_object'] for n in nodes_found]
|
|
|
|
def is_workflow_done(self):
|
|
root_nodes = self.get_root_nodes()
|
|
nodes = root_nodes
|
|
|
|
for index, n in enumerate(nodes):
|
|
obj = n['node_object']
|
|
job = obj.job
|
|
|
|
if not job:
|
|
return False
|
|
# Job is about to run or is running. Hold our horses and wait for
|
|
# the job to finish. We can't proceed down the graph path until we
|
|
# have the job result.
|
|
elif job.status not in ['failed', 'error', 'successful']:
|
|
return False
|
|
elif job.status in ['failed', 'error']:
|
|
children_failed = self.get_dependencies(obj, 'failure_nodes')
|
|
children_always = self.get_dependencies(obj, 'always_nodes')
|
|
children_all = children_failed + children_always
|
|
nodes.extend(children_all)
|
|
elif job.status in ['successful']:
|
|
children_success = self.get_dependencies(obj, 'success_nodes')
|
|
nodes.extend(children_success)
|
|
return True
|
|
|