Merge pull request kkroening#51 from kkroening/compile

kkroening · web-flow · commit 5b813cdecf73 · 2018-01-07T18:32:18.000-08:00
Add `compile` operator
diff --git a/ffmpeg/_run.py b/ffmpeg/_run.py
@@ -131,22 +131,28 @@ def get_args(stream_spec, overwrite_output=False):
     return args
 
 
+@output_operator()
+def compile(stream_spec, cmd='ffmpeg', **kwargs):
+    """Build command-line for ffmpeg."""
+    if isinstance(cmd, basestring):
+        cmd = [cmd]
+    elif type(cmd) != list:
+        cmd = list(cmd)
+    return cmd + get_args(stream_spec, **kwargs)
+
+
 @output_operator()
 def run(stream_spec, cmd='ffmpeg', **kwargs):
     """Run ffmpeg on node graph.
 
     Args:
         **kwargs: keyword-arguments passed to ``get_args()`` (e.g. ``overwrite_output=True``).
     """
-    if isinstance(cmd, basestring):
-        cmd = [cmd]
-    elif type(cmd) != list:
-        cmd = list(cmd)
-    args = cmd + get_args(stream_spec, **kwargs)
-    _subprocess.check_call(args)
+    _subprocess.check_call(compile(stream_spec, cmd, **kwargs))
 
 
 __all__ = [
+    'compile',
     'get_args',
     'run',
 ]
diff --git a/ffmpeg/tests/test_ffmpeg.py b/ffmpeg/tests/test_ffmpeg.py
@@ -219,6 +219,12 @@ def _get_drawtext_text_repr(text):
 #    subprocess.check_call(['ffmpeg', '-version'])
 
 
+def test_compile():
+    out_file = ffmpeg.input('dummy.mp4').output('dummy2.mp4')
+    assert out_file.compile() == ['ffmpeg', '-i', 'dummy.mp4', 'dummy2.mp4']
+    assert out_file.compile(cmd='ffmpeg.old') == ['ffmpeg.old', '-i', 'dummy.mp4', 'dummy2.mp4']
+
+
 def test_run():
     stream = _get_complex_filter_example()
     ffmpeg.run(stream)
