From a8b1cb63f23ed4c3acfa73fe638ef6df486ecb0f Mon Sep 17 00:00:00 2001 From: Karl Kroening Date: Thu, 6 Jul 2017 03:41:42 -0600 Subject: [PATCH] #18: use `short_repr` in `view` --- ffmpeg/_view.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/ffmpeg/_view.py b/ffmpeg/_view.py index 607c45d..cd5e14e 100644 --- a/ffmpeg/_view.py +++ b/ffmpeg/_view.py @@ -58,12 +58,9 @@ def view(stream_spec, **kwargs): raise ValueError('Invalid kwargs key(s): {}'.format(', '.join(list(kwargs.keys())))) 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) - 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, {}) for edge in get_outgoing_edges(node, outgoing_edge_map):