mirror of
https://github.com/alexta69/metube.git
synced 2025-04-04 12:47:38 +03:00
ebbe0ac7aa
iOS has strict requirements for video files, requiring h264 or h265 video codec and aac audio codec in MP4 container. This update to the MP4 format string tries to get a fully compatible file first, followed by the right video codec and any M4A audio (audio is much faster to convert if needed), and then falls back to the original behaviour of getting the best available MP4 video and M4A audio.
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
import copy
|
|
|
|
AUDIO_FORMATS = ("m4a", "mp3", "opus", "wav")
|
|
|
|
|
|
def get_format(format: str, quality: str) -> str:
|
|
"""
|
|
Returns format for download
|
|
|
|
Args:
|
|
format (str): format selected
|
|
quality (str): quality selected
|
|
|
|
Raises:
|
|
Exception: unknown quality, unknown format
|
|
|
|
Returns:
|
|
dl_format: Formatted download string
|
|
"""
|
|
format = format or "any"
|
|
|
|
if format.startswith("custom:"):
|
|
return format[7:]
|
|
|
|
if format == "thumbnail":
|
|
# Quality is irrelevant in this case since we skip the download
|
|
return "bestaudio/best"
|
|
|
|
if format in AUDIO_FORMATS:
|
|
# Audio quality needs to be set post-download, set in opts
|
|
return f"bestaudio[ext={format}]/bestaudio/best"
|
|
|
|
if format in ("mp4", "any"):
|
|
if quality == "audio":
|
|
return "bestaudio/best"
|
|
# video {res} {vfmt} + audio {afmt} {res} {vfmt}
|
|
vfmt, afmt = ("[ext=mp4]", "[ext=m4a]") if format == "mp4" else ("", "")
|
|
vres = f"[height<={quality}]" if quality != "best" else ""
|
|
vcombo = vres + vfmt
|
|
|
|
# iOS has strict requirements for video files, requiring h264 or h265
|
|
# video codec and aac audio codec in MP4 container. This format string
|
|
# attempts to get the fully compatible formats first, then the h264/h265
|
|
# video codec with any M4A audio codec (because audio is faster to
|
|
# convert if needed), and falls back to getting the best available MP4
|
|
# file.
|
|
return f"bestvideo[vcodec~='^((he|a)vc|h26[45])']{vres}+bestaudio[acodec=aac]/bestvideo[vcodec~='^((he|a)vc|h26[45])']{vres}+bestaudio{afmt}/bestvideo{vcombo}+bestaudio{afmt}/best{vcombo}"
|
|
|
|
raise Exception(f"Unkown format {format}")
|
|
|
|
|
|
def get_opts(format: str, quality: str, ytdl_opts: dict) -> dict:
|
|
"""
|
|
Returns extra download options
|
|
Mostly postprocessing options
|
|
|
|
Args:
|
|
format (str): format selected
|
|
quality (str): quality of format selected (needed for some formats)
|
|
ytdl_opts (dict): current options selected
|
|
|
|
Returns:
|
|
ytdl_opts: Extra options
|
|
"""
|
|
|
|
opts = copy.deepcopy(ytdl_opts)
|
|
|
|
postprocessors = []
|
|
|
|
if format in AUDIO_FORMATS:
|
|
postprocessors.append(
|
|
{
|
|
"key": "FFmpegExtractAudio",
|
|
"preferredcodec": format,
|
|
"preferredquality": 0 if quality == "best" else quality,
|
|
}
|
|
)
|
|
|
|
# Audio formats without thumbnail
|
|
if format not in ("wav") and "writethumbnail" not in opts:
|
|
opts["writethumbnail"] = True
|
|
postprocessors.append(
|
|
{
|
|
"key": "FFmpegThumbnailsConvertor",
|
|
"format": "jpg",
|
|
"when": "before_dl",
|
|
}
|
|
)
|
|
postprocessors.append({"key": "FFmpegMetadata"})
|
|
postprocessors.append({"key": "EmbedThumbnail"})
|
|
|
|
if format == "thumbnail":
|
|
opts["skip_download"] = True
|
|
opts["writethumbnail"] = True
|
|
postprocessors.append(
|
|
{"key": "FFmpegThumbnailsConvertor", "format": "jpg", "when": "before_dl"}
|
|
)
|
|
|
|
opts["postprocessors"] = postprocessors + (
|
|
opts["postprocessors"] if "postprocessors" in opts else []
|
|
)
|
|
return opts
|