diff options
Diffstat (limited to 'app/dl_formats.py')
| -rw-r--r-- | app/dl_formats.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/app/dl_formats.py b/app/dl_formats.py index e2f6591..8de79e9 100644 --- a/app/dl_formats.py +++ b/app/dl_formats.py @@ -1,5 +1,7 @@ import copy +AUDIO_FORMATS = ("m4a", "mp3", "opus", "wav") + def get_format(format: str, quality: str) -> str: """ Returns format for download @@ -23,7 +25,7 @@ def get_format(format: str, quality: str) -> str: # Quality is irrelevant in this case since we skip the download return "bestaudio/best" - if format in ("m4a", "mp3"): + if format in AUDIO_FORMATS: # Audio quality needs to be set post-download, set in opts return "bestaudio/best" @@ -59,17 +61,19 @@ def get_opts(format: str, quality: str, ytdl_opts: dict) -> dict: if "postprocessors" not in opts: opts["postprocessors"] = [] - if format in ("m4a", "mp3"): + if format in AUDIO_FORMATS: opts["postprocessors"].append({ "key": "FFmpegExtractAudio", "preferredcodec": format, "preferredquality": 0 if quality == "best" else quality, }) - opts["writethumbnail"] = True - opts["postprocessors"].append({"key": "FFmpegThumbnailsConvertor", "format": "jpg", "when": "before_dl"}) - opts["postprocessors"].append({"key": "FFmpegMetadata"}) - opts["postprocessors"].append({"key": "EmbedThumbnail"}) + #Audio formats without thumbnail + if format not in ("wav"): + opts["writethumbnail"] = True + opts["postprocessors"].append({"key": "FFmpegThumbnailsConvertor", "format": "jpg", "when": "before_dl"}) + opts["postprocessors"].append({"key": "FFmpegMetadata"}) + opts["postprocessors"].append({"key": "EmbedThumbnail"}) if format == "thumbnail": opts["skip_download"] = True |