svtplay-dl/lib/svtplay_dl/__init__.py

# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
from __future__ import absolute_import, unicode_literals

import sys
import logging

from svtplay_dl.utils.parser import setup_defaults, parser, mergeparseroption
from svtplay_dl.utils.getmedia import get_media, get_multiple_media
from svtplay_dl.service.cmore import Cmore


__version__ = "1.9.11"

log = logging.getLogger('svtplay_dl')


def setup_log(silent, verbose=False):
    logging.addLevelName(25, "INFO")
    fmt = '%(levelname)s: %(message)s'
    if silent:
        stream = sys.stderr
        level = 25
    elif verbose:
        stream = sys.stderr
        level = logging.DEBUG
        fmt = '%(levelname)s [%(created)s] %(filename)s/%(funcName)s: %(message)s'
    else:
        stream = sys.stdout
        level = logging.INFO

    logging.basicConfig(level=level, format=fmt)
    hdlr = logging.StreamHandler(stream)
    log.addHandler(hdlr)


def main():
    """ Main program """
    parse, options = parser(__version__)
    if options.require_subtitle:
        if options.merge_subtitle:
            options.merge_subtitle = True
        else:
            options.subtitle = True
    if options.merge_subtitle:
        options.remux = True

    if options.silent_semi:
        options.silent = True

    if options.cmoreoperatorlist:
        c = Cmore(options, None)
        c.operatorlist()
        sys.exit(0)

    if options.proxy:
        options.proxy = options.proxy.replace("socks5", "socks5h", 1)
        options.proxy = dict(http=options.proxy,
                             https=options.proxy)

    if options.flexibleq and not options.quality:
        logging.error("flexible-quality requires a quality")

    if len(options.urls) == 0:
        parse.print_help()
        sys.exit(0)
    urls = options.urls
    options = mergeparseroption(setup_defaults(), options)
    if len(urls) < 1:
        parse.error("Incorrect number of arguments")
    setup_log(options.get("silent"), options.get("verbose"))
    try:
        if len(urls) == 1:
            get_media(urls[0], options, __version__)
        else:
            get_multiple_media(urls, options)
    except KeyboardInterrupt:
        print("")