This commit is contained in:
Your Name 2021-12-01 21:48:01 +01:00
parent 0aa5571730
commit e635a4a7f5
1 changed files with 14 additions and 5 deletions

View File

@ -1,3 +1,5 @@
import copy
def get_format(format: str, quality: str) -> str: def get_format(format: str, quality: str) -> str:
""" """
Returns format for download Returns format for download
@ -48,15 +50,22 @@ def get_opts(format: str, quality: str, ytdl_opts: dict) -> dict:
Returns: Returns:
ytdl_opts: Extra options ytdl_opts: Extra options
""" """
if "postprocessors" not in ytdl_opts:
ytdl_opts["postprocessors"] = [] opts = copy.deepcopy(ytdl_opts)
if "postprocessors" not in opts:
opts["postprocessors"] = []
if format == "mp3": if format == "mp3":
extra_args = {} extra_args = {}
if quality != "best": if quality != "best":
extra_args = {"preferredquality": quality} extra_args = {"preferredquality": quality}
ytdl_opts["postprocessors"].append(
opts["postprocessors"].append(
{"key": "FFmpegExtractAudio", "preferredcodec": "mp3", **extra_args}, {"key": "FFmpegExtractAudio", "preferredcodec": "mp3", **extra_args},
) )
opts["writethumbnail"] = True
return ytdl_opts opts["postprocessors"].append({"key": "FFmpegMetadata"})
opts["postprocessors"].append({"key": "EmbedThumbnail"})
return opts