Parse ffprobe output and print results in predictable format

ffparse.py

#!/usr/bin/env python3
#
# Parse ffprobe output and print results in a predictable format
# for other tools to use.
#
# Copyright (c) 2017, Alexandre Hamelin <alexandre.hamelin gmail.com>


import sys
import re
import subprocess


class MediaFile:
    def __init__(self):
        self.chapters = []

class Chapter:
    def __init__(self):
        self.id = None
        self.start = 0
        self.end = 0
        self.title = None


if __name__ == '__main__':
    mediafile = sys.argv[1]
    output = subprocess.getoutput('ffmpeg -i {}'.format(mediafile))
    for line in output.splitlines():
        line = line.strip()
        if line.startswith('Input #'):
            pass
        elif line.startswith('Duration:'):
            pass
        elif line.startswith('Chapter #'):
            try:
                id, start, end = re.search(
                    r'#(\d+:\d+): start (\S+), end (\S+)$', line).groups()
            except AttributeError:
                print('Invalid Chapter line: {}'.format(line)
                raise SystemExit(1)
            ch = Chapter()
            ch.id = id
            ch.start = float(start)
            ch.end = float(end)
            media.chapters.append(ch)
        elif line.startswith('title '):
            title = re.search(r'title\s+: (.*)', line).group(1)
            media.chapters[-1].title = title
        elif line.startswith('Stream #'):
            stream_info = re.search(
                r'#(\d+:\d+)(?:\((...)\))?: (\S+): (\S+) (.*)', line).groups()

SAMPLE.csv