Add get_video_thumbnail example

kkroening · kkroening · commit 4a70f6a86884 · 2018-06-02T01:04:39.000-07:00
diff --git a/examples/get_video_thumbnail.py b/examples/get_video_thumbnail.py
@@ -0,0 +1,33 @@
+#!/usr/bin/env python
+from __future__ import unicode_literals, print_function
+import argparse
+import ffmpeg
+import sys
+
+
+parser = argparse.ArgumentParser(description='Extract video thumbnail')
+parser.add_argument('in_filename', help='Input filename')
+parser.add_argument('out_filename', help='Output filename')
+parser.add_argument(
+    '--time', type=int, default=0.1, help='Time offset')
+parser.add_argument(
+    '--width', type=int, default=120,
+    help='Width of output thumbnail (height automatically determined by aspect ratio)')
+
+
+def generate_thumbnail(in_filename, out_filename, time, width):
+    try:
+        (
+            ffmpeg
+            .input(in_filename, ss=time)
+            .filter_('scale', width, -1)
+            .output(out_filename, vframes=1, format='image2', vcodec='mjpeg')
+            .run(capture_stdout=True, capture_stderr=True, overwrite_output=True)
+        )
+    except ffmpeg.Error as e:
+        print(e.stderr.decode(), file=sys.stderr)
+
+
+if __name__ == '__main__':
+    args = parser.parse_args()
+    generate_thumbnail(args.in_filename, args.out_filename, args.time, args.width)
