Mercurial > hg > hls
view hls/main.py @ 4:f2331455c15f default tip
stub
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Mon, 20 Jul 2015 16:47:30 -0700 |
parents | ed7e9734a088 |
children |
line wrap: on
line source
#!/usr/bin/env python # -*- coding: utf-8 -*- """ HLS client testing library for python """ # imports import argparse import m3u8 import os import subprocess import sys import time # python requirements # (key, value) = (module, PyPI name) requirements = () for module, package in requirements: try: globals()[module] = __import__(module) except ImportError: # install requirement and try again subprocess.check_call(['pip', 'install', package]) args = [sys.executable] + sys.argv os.execl(sys.executable, *args) # module globals __all__ = ['main', 'Parser'] here = os.path.dirname(os.path.realpath(__file__)) string = (str, unicode) def ensure_dir(directory): """ensure a directory exists""" if os.path.exists(directory): if not os.path.isdir(directory): raise OSError("Not a directory: '{}'".format(directory)) return directory os.makedirs(directory) return directory class Parser(argparse.ArgumentParser): """CLI option parser""" def __init__(self, **kwargs): kwargs.setdefault('formatter_class', argparse.RawTextHelpFormatter) kwargs.setdefault('description', __doc__) argparse.ArgumentParser.__init__(self, **kwargs) self.add_argument('url', help='URL of manifest') self.add_argument('--monitor', dest='monitor', type=float, metavar='SLEEP', help="run in monitor mode") 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""" def main(args=sys.argv[1:]): """CLI""" # parse command line options parser = Parser() options = parser.parse_args(args) try: while True: # load the manifest manifest = m3u8.load(options.url) # TODO: make sure each piece of content is loadable # loop control if options.monitor: time.sleep(options.monitor) else: break except KeyboardInterrupt: pass if __name__ == '__main__': main()