1
0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-12-11 20:34:18 +01:00
svtplay-dl/lib/svtplay_dl/__init__.py

77 lines
2.2 KiB
Python
Raw Normal View History

2013-03-23 15:02:41 +01:00
# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
import logging
2019-08-25 00:40:39 +02:00
import sys
2013-03-23 15:02:41 +01:00
2019-08-25 00:40:39 +02:00
import yaml
2017-09-16 23:45:29 +02:00
from svtplay_dl.service.cmore import Cmore
2019-08-25 00:40:39 +02:00
from svtplay_dl.utils.getmedia import get_media
from svtplay_dl.utils.getmedia import get_multiple_media
from svtplay_dl.utils.parser import parser
from svtplay_dl.utils.parser import parsertoconfig
from svtplay_dl.utils.parser import setup_defaults
2018-07-10 21:32:54 +02:00
from .__version__ import get_versions
2019-08-25 00:27:31 +02:00
__version__ = get_versions()["version"]
2018-07-10 21:32:54 +02:00
del get_versions
2018-05-13 13:06:45 +02:00
2019-08-25 00:27:31 +02:00
log = logging.getLogger("svtplay_dl")
2013-03-23 15:02:41 +01:00
2018-01-30 20:11:37 +01:00
2014-01-05 16:30:45 +01:00
def setup_log(silent, verbose=False):
logging.addLevelName(25, "INFO")
2019-08-25 00:27:31 +02:00
fmt = "%(levelname)s: %(message)s"
2013-03-23 15:02:41 +01:00
if silent:
stream = sys.stderr
level = 25
2014-01-05 16:30:45 +01:00
elif verbose:
stream = sys.stderr
level = logging.DEBUG
2019-08-25 00:27:31 +02:00
fmt = "%(levelname)s [%(created)s] %(pathname)s/%(funcName)s: %(message)s"
2013-03-23 15:02:41 +01:00
else:
stream = sys.stdout
level = logging.INFO
2018-03-13 00:34:39 +01:00
logging.basicConfig(level=level, format=fmt)
2013-03-23 15:02:41 +01:00
hdlr = logging.StreamHandler(stream)
2018-03-13 00:34:39 +01:00
log.addHandler(hdlr)
2013-03-23 15:02:41 +01:00
2015-09-15 20:10:32 +02:00
2013-03-23 15:02:41 +01:00
def main():
2021-12-18 21:37:09 +01:00
"""Main program"""
2018-01-14 00:39:16 +01:00
parse, options = parser(__version__)
2013-03-23 15:02:41 +01:00
2013-03-23 18:26:48 +01:00
if options.flexibleq and not options.quality:
logging.error("flexible-quality requires a quality")
if options.only_audio and options.only_video:
logging.error("Only use one of them, not both at the same time")
sys.exit(2)
if len(options.urls) == 0:
parse.print_help()
sys.exit(0)
urls = options.urls
config = parsertoconfig(setup_defaults(), options)
if len(urls) < 1:
parse.error("Incorrect number of arguments")
setup_log(config.get("silent"), config.get("verbose"))
if options.cmoreoperatorlist:
config = parsertoconfig(setup_defaults(), options)
c = Cmore(config, urls)
c.operatorlist()
sys.exit(0)
try:
if len(urls) == 1:
get_media(urls[0], config, __version__)
else:
get_multiple_media(urls, config)
except KeyboardInterrupt:
2014-05-02 12:06:53 +02:00
print("")
except (yaml.YAMLError, yaml.MarkedYAMLError) as e:
2021-12-18 21:36:16 +01:00
logging.error("Your settings file(s) contain invalid YAML syntax! Please fix and restart!, %s", str(e))
sys.exit(2)