Add video_info example

kkroening · kkroening · commit 6274e7abf9bf · 2018-06-02T00:51:10.000-07:00
diff --git a/examples/video_info.py b/examples/video_info.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+from __future__ import unicode_literals, print_function
+import argparse
+import ffmpeg
+import sys
+
+
+parser = argparse.ArgumentParser(description='Get video information')
+parser.add_argument('in_filename', help='Input filename')
+
+
+if __name__ == '__main__':
+    args = parser.parse_args()
+    probe = ffmpeg.probe(args.in_filename)
+    video_info = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
+    if video_info is None:
+        print('No video stream found', file=sys.stderr)
+        sys.exit(1)
+
+    width = int(video_info['width'])
+    height = int(video_info['height'])
+    num_frames = int(video_info['nb_frames'])
+    print('width: {}'.format(width))
+    print('height: {}'.format(height))
+    print('num_frames: {}'.format(num_frames))
