import os import sys import copy import logging from shutil import which from datetime import datetime from svtplay_dl.log import log from svtplay_dl.service import service_handler, Generic from svtplay_dl.service.services import sites, Raw from svtplay_dl.fetcher import VideoRetriever from svtplay_dl.subtitle import subtitle from svtplay_dl.utils.output import filename, formatname from svtplay_dl.postprocess import postprocess from svtplay_dl.utils.stream import select_quality, list_quality from svtplay_dl.utils.text import exclude from svtplay_dl.error import UIException from svtplay_dl.utils.nfo import write_nfo_episode, write_nfo_tvshow def get_multiple_media(urls, config): if config.get("output") and os.path.isfile(config.output): log.error("Output must be a directory if used with multiple URLs") sys.exit(2) elif config.get("output") and not os.path.exists(config.get("output")): try: os.makedirs(config.get("output")) except OSError as e: logging.error("%s: %s", e.strerror, e.filename) return for url in urls: get_media(url, copy.copy(config)) def get_media(url, options, version="Unknown"): if "http" not in url[:4]: url = "http://%s" % url if options.get("verbose"): logging.debug("version: {0}".format(version)) stream = service_handler(sites, options, url) if not stream: generic = Generic(options, url) url, stream = generic.get(sites) if not stream: if url.find(".f4m") > 0 or url.find(".m3u8") > 0: stream = Raw(options, url) if not stream: logging.error("That site is not supported. Make a ticket or send a message") sys.exit(2) if options.get("all_episodes"): get_all_episodes(stream, url) else: get_one_media(stream) def get_all_episodes(stream, url): name = os.path.dirname(formatname({"basedir": True}, stream.config)) if name and os.path.isfile(name): logging.error("Output must be a directory if used with --all-episodes") sys.exit(2) elif name and not os.path.exists(name): try: os.makedirs(name) except OSError as e: logging.error("%s: %s", e.strerror, e.filename) return episodes = stream.find_all_episodes(stream.config) if episodes is None: return for idx, o in enumerate(episodes): if o == url: substream = stream else: substream = service_handler(sites, copy.copy(stream.config), o) logging.info("Episode %d of %d", idx + 1, len(episodes)) logging.info("Url: %s", o) # get_one_media overwrites options.output... get_one_media(substream) def get_one_media(stream): # Make an automagic filename if not filename(stream): return if stream.config.get("merge_subtitle"): if not which('ffmpeg'): logging.error("--merge-subtitle needs ffmpeg. Please install ffmpeg.") logging.info("https://ffmpeg.org/download.html") sys.exit(2) videos = [] subs = [] subfixes = [] error = [] streams = stream.get() try: for i in streams: if isinstance(i, Exception): error.append(i) elif not exclude(stream.config, formatname(i.output, stream.config)): if isinstance(i, VideoRetriever): if stream.config.get("preferred"): if stream.config.get("preferred").lower() == i.name: videos.append(i) else: videos.append(i) if isinstance(i, subtitle): subs.append(i) except Exception as e: if stream.config.get("verbose"): raise else: logging.error("svtplay-dl crashed") logging.error("Run again and add --verbose as an argument, to get more information") logging.error("If the error persists, you can report it at https://github.com/spaam/svtplay-dl/issues") logging.error("Include the URL used, the stack trace and the output of svtplay-dl --version in the issue") return try: after_date = datetime.strptime(stream.config.get("after_date"), "%Y-%m-%d") except (ValueError, TypeError, KeyError, AttributeError): # gotta catch em all.. after_date = None try: pub_date = datetime.fromtimestamp(stream.output["publishing_datetime"]) except (ValueError, TypeError, KeyError): pub_date = None if after_date is not None and pub_date is not None and pub_date.date() < after_date.date(): logging.info("Video {}S{}E{} skipped since published {} before {}. ".format( stream.output["title"], stream.output["season"], stream.output["episode"], pub_date.date(), after_date.date())) return if stream.config.get("require_subtitle") and not subs: logging.info("No subtitles available") return if stream.config.get("subtitle") and stream.config.get("get_url"): if subs: if stream.config.get("get_all_subtitles"): for sub in subs: print(sub.url) else: print(subs[0].url) if stream.config.get("force_subtitle"): return def options_subs_dl(subfixes): if subs: if stream.config.get("get_all_subtitles"): for sub in subs: sub.download() if stream.config.get("merge_subtitle"): if sub.subfix: subfixes += [sub.subfix] else: stream.config.set("get_all_subtitles", False) else: subs[0].download() elif stream.config.get("merge_subtitle"): stream.config.set("merge_subtitle", False) if stream.config.get("subtitle") and not stream.config.get("get_url"): options_subs_dl(subfixes) if stream.config.get("force_subtitle"): return if stream.config.get("merge_subtitle") and not stream.config.get("subtitle"): options_subs_dl(subfixes) if not videos: errormsg = None for exc in error: if errormsg: errormsg = "{}. {}".format(errormsg, str(exc)) else: errormsg = str(exc) logging.error("No videos found. {}".format(errormsg)) else: if stream.config.get("list_quality"): list_quality(videos) return try: fstream = select_quality(stream.config, videos) if fstream.config.get("get_url"): print(fstream.url) return logging.info("Selected to download %s, bitrate: %s", fstream.name, fstream.bitrate) fstream.download() except UIException as e: if fstream.config.get("verbose"): raise e logging.error(e) sys.exit(2) if fstream.config.get("thumbnail") and hasattr(stream, "get_thumbnail"): stream.get_thumbnail(stream.config) if stream.config.get("nfo"): # Create NFO files write_nfo_episode(stream.output, stream.config) write_nfo_tvshow(stream.output, stream.config) post = postprocess(fstream, fstream.config, subfixes) if fstream.audio and post.detect: post.merge() if fstream.audio and not post.detect and fstream.finished: logging.warning("Cant find ffmpeg/avconv. audio and video is in seperate files. if you dont want this use -P hls or hds") if fstream.name == "hls" or fstream.config.get("remux"): post.remux() if fstream.config.get("silent_semi") and fstream.finished: logging.log(25, "Download of %s was completed" % fstream.options.output)