view python/example/m3u.py @ 881:4ebb6aff9d67

no more has_key
author Jeff Hammel <k0scist@gmail.com>
date Tue, 06 Aug 2019 14:22:00 -0700 (2019-08-06)
parents e6fb1a8fe66b
children
line wrap: on
line source
#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
m3u playlist manipulator

See:
- https://pypi.python.org/pypi/m3u8
- http://gonze.com/playlists/playlist-format-survey.html#M3U
- http://www.assistanttools.com/articles/m3u_playlist_format.shtml
- http://en.wikipedia.org/wiki/M3U
- http://tools.ietf.org/html/draft-pantos-http-live-streaming-08#section-3.3.1
"""

# imports
import argparse
import m3u8
import os
import subprocess
import sys

# module globals
__all__ = ['main', 'M3UParser']
here = os.path.dirname(os.path.realpath(__file__))
string = (str, unicode)

class Playlist(object):
    def __init__(self, playlists=(), base_path=None):
        self.clips = []
        for playlist in playlists:
            assert os.path.isfile(playlist)
            _base_path = os.path.abspath(base_path or os.path.dirname(os.path.abspath(playlist))) # TODO: this doesn't fix URLs
            content = open(playlist).read()
            m3u = m3u8.M3U8(content, base_path=_base_path)
            for segment in m3u.segments:
                path = segment.uri
                title = segment.title
                self.clips.append((title, path))

    def __str__(self):
        retval = []
        for title, path in self.clips:
            if title:
                line = '{} # {}'.format(os.path.abspath(path), title)
            else:
                line = os.path.abspath(path)
            retval.append(line)
        return '\n'.join(retval)

class M3UParser(argparse.ArgumentParser):
    """CLI option parser"""
    def __init__(self, **kwargs):
        kwargs.setdefault('description', __doc__)
        argparse.ArgumentParser.__init__(self, **kwargs)
        self.add_argument('playlists', metavar='playlist', nargs='+',
                          help="playlist to parse")
        self.add_argument('-b', '--base', dest='base_path', default=os.getcwd(),
                          help="base path [DEFAULT: %(default)s]")
        self.options = None

    def parse_args(self, *args, **kw):
        options = argparse.ArgumentParser.parse_args(self, *args, **kw)
        self.validate(options)
        self.options = options
        return options

    def validate(self, options):
        """validate options"""
        if not os.path.exists(options.base_path):
            self.error("Base path '{}' does not exist".format(options.base_path))
        missing = [i for i in options.playlists
                   if not os.path.exists(i)]
        if missing:
            self.error("Playlist not found: {}".format(', '.join(missing)))

def main(args=sys.argv[1:]):
    """CLI"""

    # parse command line options
    parser = M3UParser()
    options = parser.parse_args(args)

    # create playlist
    playlist = Playlist(options.playlists, options.base_path)

    # print out some things
    print ('{}'.format(playlist))

if __name__ == '__main__':
    main()