mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-24 12:15:40 +01:00
243 lines
8.7 KiB
Python
243 lines
8.7 KiB
Python
import copy
|
|
import logging
|
|
import os
|
|
import sys
|
|
from datetime import datetime
|
|
from shutil import which
|
|
|
|
from svtplay_dl.error import UIException
|
|
from svtplay_dl.fetcher import VideoRetriever
|
|
from svtplay_dl.postprocess import postprocess
|
|
from svtplay_dl.service import Generic
|
|
from svtplay_dl.service import service_handler
|
|
from svtplay_dl.service.services import Raw
|
|
from svtplay_dl.service.services import sites
|
|
from svtplay_dl.subtitle import subtitle
|
|
from svtplay_dl.utils.nfo import write_nfo_episode
|
|
from svtplay_dl.utils.nfo import write_nfo_tvshow
|
|
from svtplay_dl.utils.output import filename
|
|
from svtplay_dl.utils.output import find_dupes
|
|
from svtplay_dl.utils.output import formatname
|
|
from svtplay_dl.utils.stream import list_quality
|
|
from svtplay_dl.utils.stream import select_quality
|
|
from svtplay_dl.utils.text import exclude
|
|
|
|
|
|
def get_multiple_media(urls, config):
|
|
if config.get("output") and os.path.isfile(config.get("output")):
|
|
logging.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(f"version: {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 or url.find(".mpd") > 1:
|
|
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") or stream.config.get("all_episodes"):
|
|
get_all_episodes(stream, url, options)
|
|
else:
|
|
get_one_media(stream)
|
|
|
|
|
|
def get_all_episodes(stream, url, options):
|
|
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)
|
|
|
|
if not (options.get("get_url") and options.get("get_only_episode_url")):
|
|
# 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:
|
|
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"):
|
|
if not subs:
|
|
logging.info("No subtitles available")
|
|
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 = f"{errormsg}. {str(exc)}"
|
|
else:
|
|
errormsg = str(exc)
|
|
if errormsg:
|
|
logging.error(f"No videos found. {errormsg}")
|
|
else:
|
|
logging.error("No videos found.")
|
|
else:
|
|
if stream.config.get("list_quality"):
|
|
list_quality(videos)
|
|
return
|
|
if stream.config.get("nfo"):
|
|
# Create NFO files
|
|
write_nfo_episode(stream.output, stream.config)
|
|
write_nfo_tvshow(stream.output, stream.config)
|
|
if stream.config.get("force_nfo"):
|
|
return
|
|
fstream = select_quality(stream.config, videos)
|
|
if fstream.config.get("get_url"):
|
|
print(fstream.url)
|
|
return
|
|
|
|
dupe, fileame = find_dupes(fstream.output, stream.config)
|
|
if dupe and not stream.config.get("force"):
|
|
logging.warning(f"File ({fileame.name}) already exists. Use --force to overwrite")
|
|
return
|
|
try:
|
|
logging.info("Selected to download %s, bitrate: %s format: %s", fstream.name, fstream.bitrate, fstream.format)
|
|
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)
|
|
|
|
post = postprocess(fstream, fstream.config, subfixes)
|
|
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")
|
|
elif fstream.audio and post.detect:
|
|
post.merge()
|
|
elif 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)
|