diff options
| author | pukkandan <pukkandan.ytdlp@gmail.com> | 2022-04-12 05:27:17 +0530 |
|---|---|---|
| committer | pukkandan <pukkandan.ytdlp@gmail.com> | 2022-04-12 05:32:53 +0530 |
| commit | 66cf3e1001b6d9a2829fe834c3f9103b0890918e (patch) | |
| tree | 9356d80a87cefc631964cf81aee1cf6220f1fb49 /yt_dlp/postprocessor/ffmpeg.py | |
| parent | b07897ef5bcdb865991d8601faef64a451da39fc (diff) | |
| download | hypervideo-pre-66cf3e1001b6d9a2829fe834c3f9103b0890918e.tar.lz hypervideo-pre-66cf3e1001b6d9a2829fe834c3f9103b0890918e.tar.xz hypervideo-pre-66cf3e1001b6d9a2829fe834c3f9103b0890918e.zip | |
[EmbedSubtitle] Enable for more video extensions
Closes #3382
Diffstat (limited to 'yt_dlp/postprocessor/ffmpeg.py')
| -rw-r--r-- | yt_dlp/postprocessor/ffmpeg.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/yt_dlp/postprocessor/ffmpeg.py b/yt_dlp/postprocessor/ffmpeg.py index 69182618b..4b61693a2 100644 --- a/yt_dlp/postprocessor/ffmpeg.py +++ b/yt_dlp/postprocessor/ffmpeg.py @@ -577,14 +577,16 @@ class FFmpegVideoRemuxerPP(FFmpegVideoConvertorPP): class FFmpegEmbedSubtitlePP(FFmpegPostProcessor): + SUPPORTED_EXTS = ('mp4', 'mov', 'm4a', 'webm', 'mkv', 'mka') + def __init__(self, downloader=None, already_have_subtitle=False): super().__init__(downloader) self._already_have_subtitle = already_have_subtitle @PostProcessor._restrict_to(images=False) def run(self, info): - if info['ext'] not in ('mp4', 'webm', 'mkv'): - self.to_screen('Subtitles can only be embedded in mp4, webm or mkv files') + if info['ext'] not in self.SUPPORTED_EXTS: + self.to_screen(f'Subtitles can only be embedded in {", ".join(self.SUPPORTED_EXTS)} files') return [], info subtitles = info.get('requested_subtitles') if not subtitles: |