#18: use short_repr in view

This commit is contained in:
Karl Kroening 2017-07-06 03:41:42 -06:00
parent 0b1259238f
commit a8b1cb63f2

View File

@ -58,12 +58,9 @@ def view(stream_spec, **kwargs):
raise ValueError('Invalid kwargs key(s): {}'.format(', '.join(list(kwargs.keys())))) raise ValueError('Invalid kwargs key(s): {}'.format(', '.join(list(kwargs.keys()))))
for node in sorted_nodes: for node in sorted_nodes:
name = node.name
if '_kwargs' in dir(node) and 'filename' in node._kwargs:
name = os.path.basename(node._kwargs['filename'])
color = _get_node_color(node) color = _get_node_color(node)
graph.node(str(hash(node)), name, shape='box', style='filled', fillcolor=color) graph.node(str(hash(node)), node.short_repr, shape='box', style='filled', fillcolor=color)
outgoing_edge_map = outgoing_edge_maps.get(node, {}) outgoing_edge_map = outgoing_edge_maps.get(node, {})
for edge in get_outgoing_edges(node, outgoing_edge_map): for edge in get_outgoing_edges(node, outgoing_edge_map):