def runner(self):
return ru.Runner(self, self._task_executor)
def main():
parser = optparse.OptionParser()
parser.add_option("-f",
"--file",
dest="filename",
help="write svg to FILE",
metavar="FILE")
parser.add_option("-t",
"--tasks",
dest="tasks",
action='store_true',
help="use task state transitions",
default=False)
parser.add_option("-r",
"--retries",
dest="retries",
action='store_true',
help="use retry state transitions",
default=False)
parser.add_option("-e",
"--engines",
dest="engines",
action='store_true',
help="use engine state transitions",
default=False)
parser.add_option("-T",
"--format",
dest="format",
help="output in given format",
default='svg')
(options, args) = parser.parse_args()
if options.filename is None:
options.filename = 'states.%s' % options.format
types = [options.engines, options.retries, options.tasks]
if sum([int(i) for i in types]) > 1:
parser.error("Only one of task/retry/engines may be specified.")
internal_states = list()
ordering = 'in'
if options.tasks:
source_type = "Tasks"
source = make_machine(states.PENDING,
list(states._ALLOWED_TASK_TRANSITIONS),
[states.RETRYING])
elif options.retries:
source_type = "Retries"
source = make_machine(states.PENDING,
list(states._ALLOWED_TASK_TRANSITIONS), [])
elif options.engines:
source_type = "Engines"
r = runner.Runner(DummyRuntime(), None)
source, memory = r.builder.build()
internal_states.extend(runner._META_STATES)
ordering = 'out'
else:
source_type = "Flow"
source = make_machine(states.PENDING,
list(states._ALLOWED_FLOW_TRANSITIONS), [])
graph_name = "%s states" % source_type
g = pydot.Dot(graph_name=graph_name,
rankdir='LR',
nodesep='0.25',
overlap='false',
ranksep="0.5",
size="11x8.5",
splines='true',
ordering=ordering)
node_attrs = {
'fontsize': '11',
}
nodes = {}
for (start_state, _on_event, end_state) in source:
if start_state not in nodes:
start_node_attrs = node_attrs.copy()
text_color = map_color(internal_states, start_state)
if text_color:
start_node_attrs['fontcolor'] = text_color
nodes[start_state] = pydot.Node(start_state, **start_node_attrs)
g.add_node(nodes[start_state])
if end_state not in nodes:
end_node_attrs = node_attrs.copy()
text_color = map_color(internal_states, end_state)
if text_color:
end_node_attrs['fontcolor'] = text_color
nodes[end_state] = pydot.Node(end_state, **end_node_attrs)
g.add_node(nodes[end_state])
g.add_edge(pydot.Edge(nodes[start_state], nodes[end_state]))
start = pydot.Node("__start__",
shape="point",
width="0.1",
xlabel='start',
fontcolor='green',
**node_attrs)
g.add_node(start)
g.add_edge(pydot.Edge(start, nodes[source.start_state], style='dotted'))
print("*" * len(graph_name))
print(graph_name)
print("*" * len(graph_name))
print(source.pformat())
print(g.to_string().strip())
g.write(options.filename, format=options.format)
print("Created %s at '%s'" % (options.format, options.filename))