kkroening#17: add short_repr for input and output nodes

kkroening · kkroening · commit c6e2f05e5b15 · 2017-07-06T03:42:03.000-06:00
diff --git a/ffmpeg/nodes.py b/ffmpeg/nodes.py
@@ -1,8 +1,9 @@
 from __future__ import unicode_literals
 
-from builtins import object
 from .dag import KwargReprNode
 from ._utils import get_hash_int
+from builtins import object
+import os
 
 
 def _is_of_types(obj, types):
@@ -115,6 +116,10 @@ def __init__(self, name, args=[], kwargs={}):
             kwargs=kwargs
         )
 
+    @property
+    def short_repr(self):
+        return os.path.basename(self.kwargs['filename'])
+
 
 class FilterNode(Node):
     def __init__(self, stream_spec, name, max_inputs=1, args=[], kwargs={}):
@@ -152,6 +157,10 @@ def __init__(self, stream, name, args=[], kwargs={}):
             kwargs=kwargs
         )
 
+    @property
+    def short_repr(self):
+        return os.path.basename(self.kwargs['filename'])
+
 
 class OutputStream(Stream):
     def __init__(self, upstream_node, upstream_label):
