kkroening#18: use get_stream_spec_nodes in view

kkroening · kkroening · commit 2d6b0d473013 · 2017-07-09T16:00:41.000-06:00
diff --git a/ffmpeg/_view.py b/ffmpeg/_view.py
@@ -8,7 +8,7 @@
 
 from ffmpeg.nodes import (
     FilterNode,
-    get_stream_map,
+    get_stream_spec_nodes,
     InputNode,
     OutputNode,
     Stream,
@@ -44,12 +44,7 @@ def view(stream_spec, **kwargs):
     if filename is None:
         filename = tempfile.mktemp()
 
-    nodes = []
-    stream_map = get_stream_map(stream_spec)
-    for stream in stream_map.values():
-        if not isinstance(stream, Stream):
-            raise TypeError('Expected Stream; got {}'.format(type(stream)))
-        nodes.append(stream.node)
+    nodes = get_stream_spec_nodes(stream_spec)
 
     sorted_nodes, outgoing_edge_maps = topo_sort(nodes)
     graph = graphviz.Digraph()
