diff --git a/common/common/segments.py b/common/common/segments.py index c5dc0a5..94f86ec 100644 --- a/common/common/segments.py +++ b/common/common/segments.py @@ -391,8 +391,13 @@ def ffmpeg_cut_segment(segment, cut_start=None, cut_end=None): return subprocess.Popen(args, stdout=subprocess.PIPE) +def ffmpeg_cut_one(segments, encode_args, output_file=subprocess.PIPE, input_args=[]): + """Wrapper for ffmpeg_cut_many() with a simpler API for the single-input case.""" + return ffmpeg_cut_many([(segments, input_args)], encode_args, output_file=output_file) + + @contextmanager -def ffmpeg_cut_many(inputs, output_file, encode_args): +def ffmpeg_cut_many(inputs, encode_args, output_file=subprocess.PIPE): """ Context manager that produces a Popen object which is ffmpeg cutting the given inputs. @@ -699,7 +704,7 @@ def full_cut_segments(segments, start, end, encode_args, stream=False): args += ['-t', duration] args += list(encode_args) - with ffmpeg_cut_many([segments, ()], tempfile, args) as ffmpeg: + with ffmpeg_cut_many([segments, ()], args, output_file=tempfile) as ffmpeg: # When streaming, we can return data as it is available # Otherwise, just exit the context manager so tempfile is fully written. if stream: @@ -737,8 +742,8 @@ def archive_cut_segments(segment_ranges, ranges, tempdir): tempfile_name = os.path.join(tempdir, "archive-temp-{}.mkv".format(uuid4())) try: with open(tempfile_name, "wb") as tempfile: - with ffmpeg_cut_many([(segments, [])], tempfile, encode_args): - # We just want ffmpeg to run to completion, which ffmpeg_cut_many() + with ffmpeg_cut_one(segments, encode_args, output_file=tempfile): + # We just want ffmpeg to run to completion, which ffmpeg_cut_one() # will do on exit for us. pass except: @@ -776,7 +781,7 @@ def render_segments_waveform(segments, size=(1024, 128), scale='sqrt', color='#0 # output as png '-f', 'image2', '-c', 'png', ] - with ffmpeg_cut_many([(segments, [])], subprocess.PIPE, args) as ffmpeg: + with ffmpeg_cut_one(segments, args) as ffmpeg: for chunk in read_chunks(ffmpeg.stdout): yield chunk @@ -803,7 +808,7 @@ def extract_frame(segments, timestamp): # output as png '-f', 'image2', '-c', 'png', ] - with ffmpeg_cut_many([(segments, input_args)], subprocess.PIPE, args) as ffmpeg: + with ffmpeg_cut_one(segments, args, input_args=input_args) as ffmpeg: for chunk in read_chunks(ffmpeg.stdout): yield chunk