#!/usr/bin/env python

import argparse

import json

import logging

import subprocess

import sys

parser = argparse.ArgumentParser(description='Run ffprobe on one or more files with JSON output')

parser.add_argument('filenames', nargs='+', help='Filename(s)')

class ExecException(Exception):

def __init__(self, stderr_output):

super(ExecException, self).__init__('Execution error')

self.stderr_output = stderr_output

def run_ffprobe(filename):

args = ['ffprobe', '-show_format', '-show_streams', '-of', 'json', filename]

p = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

out, err = p.communicate()

if p.returncode != 0:

raise ExecException(err)

return json.loads(out)

def main(filenames):

failed = False

results = {}

for filename in sys.argv[1:]:

try:

results[filename] = run_ffprobe(filename)

except ExecException as e:

logging.error('Error processing {}:'.format(filename))

logging.error(e.stderr_output)

failed = True

except Exception as e:

logging.exception('Error processing {}:'.format(filename))

failed = True

if results != {}:

print(json.dumps(results, indent=4, sort_keys=True))

if failed:

logging.warning('One or more ffprobe calls failed. See above for detail.')

sys.exit(int(failed))

if __name__ == '__main__':

args = parser.parse_args()

main(args.filenames)