Compare commits

..

No commits in common. "83d4013ccb37c36790e6ce4eeed8b5598e84f16e" and "e9a49de89cb68c1f674d53003a6b07643edb2ab0" have entirely different histories.

2 changed files with 19 additions and 22 deletions

View File

@ -107,10 +107,8 @@ class EmbedThumbnailPP(FFmpegPostProcessor):
except Exception as err: except Exception as err:
self.report_warning(f'unable to embed using mutagen; {err}') self.report_warning(f'unable to embed using mutagen; {err}')
success = False success = False
else:
success = False
# Method 2: Use ffmpeg # Method 2: Use ffmpeg
if not success: else:
options = [ options = [
'-c', 'copy', '-map', '0:0', '-map', '1:0', '-write_id3v1', '1', '-id3v2_version', '3', '-c', 'copy', '-map', '0:0', '-map', '1:0', '-write_id3v1', '1', '-id3v2_version', '3',
'-metadata:s:v', 'title="Album cover"', '-metadata:s:v', 'comment=Cover (front)'] '-metadata:s:v', 'title="Album cover"', '-metadata:s:v', 'comment=Cover (front)']

View File

@ -587,8 +587,8 @@ class FFmpegVideoRemuxerPP(FFmpegVideoConvertorPP):
class FFmpegEmbedSubtitlePP(FFmpegPostProcessor): class FFmpegEmbedSubtitlePP(FFmpegPostProcessor):
SUPPORTS_LYRICS = ('mp3', 'm4a', 'flac', 'opus') AUDIO_EXTS = ('mp3', 'm4a', 'flac', 'opus')
SUPPORTED_EXTS = ('mp4', 'mov', 'm4a', 'webm', 'mkv', 'mka', *SUPPORTS_LYRICS) SUPPORTED_EXTS = ('mp4', 'mov', 'm4a', 'webm', 'mkv', 'mka')
def __init__(self, downloader=None, already_have_subtitle=False): def __init__(self, downloader=None, already_have_subtitle=False):
super().__init__(downloader) super().__init__(downloader)
@ -596,8 +596,8 @@ class FFmpegEmbedSubtitlePP(FFmpegPostProcessor):
@PostProcessor._restrict_to(images=False) @PostProcessor._restrict_to(images=False)
def run(self, info): def run(self, info):
if info['ext'] not in self.SUPPORTED_EXTS: if info['ext'] not in self.SUPPORTED_EXTS + self.AUDIO_EXTS:
self.to_screen(f'Subtitles can only be embedded in {", ".join(self.SUPPORTED_EXTS)} files') self.to_screen(f'Subtitles can only be embedded in {", ".join(self.SUPPORTED_EXTS+self.AUDIO_EXTS)} files')
return [], info return [], info
subtitles = info.get('requested_subtitles') subtitles = info.get('requested_subtitles')
if not subtitles: if not subtitles:
@ -661,8 +661,8 @@ class FFmpegEmbedSubtitlePP(FFmpegPostProcessor):
temp_filename = prepend_extension(filename, 'temp') temp_filename = prepend_extension(filename, 'temp')
self.to_screen('Embedding subtitles in "%s"' % filename) self.to_screen('Embedding subtitles in "%s"' % filename)
if info['ext'] in self.SUPPORTS_LYRICS: if info['ext'] in self.AUDIO_EXTS:
self.embed_lyrics(info['filepath'], sub_dict=info['requested_subtitles']) self.embed_lyrics(input_files)
else: else:
self.run_ffmpeg_multiple_files(input_files, temp_filename, opts) self.run_ffmpeg_multiple_files(input_files, temp_filename, opts)
os.replace(temp_filename, filename) os.replace(temp_filename, filename)
@ -670,24 +670,23 @@ class FFmpegEmbedSubtitlePP(FFmpegPostProcessor):
files_to_delete = [] if self._already_have_subtitle else sub_filenames files_to_delete = [] if self._already_have_subtitle else sub_filenames
return files_to_delete, info return files_to_delete, info
def embed_lyrics(self, audio_file, sub_dict): def embed_lyrics(self, input_files):
if len(sub_dict) > 1: audio_file = input_files[0]
self.report_warning('More than one subtitle file found. Your media player will likely be unable to display all of them.') subs = input_files[1]
if len(input_files) > 2:
if sub_dict[list(sub_dict.keys())[0]]['ext'] != 'lrc': self.report_warning('More than one subtitle file found. Only one will be embedded')
if not subs.endswith('.lrc'):
raise PostProcessingError('LRC subtitles required. Use "--convert-subs lrc" to convert') raise PostProcessingError('LRC subtitles required. Use "--convert-subs lrc" to convert')
lyrics_list = [] with open(subs, 'r', encoding='utf-8') as f:
for lyrics in sub_dict.keys(): lyrics = f.read().strip()
lyrics_list.append(sub_dict[lyrics]['data'])
if audio_file.endswith('.mp3'): if audio_file.endswith('.mp3'):
for lyrics in lyrics_list: audio = mutagen.id3.ID3(audio_file)
audio = mutagen.id3.ID3(audio_file) audio.add(mutagen.id3.USLT(encoding=mutagen.id3.Encoding.UTF8, lang='und', desc='', text=lyrics))
audio.add(mutagen.id3.USLT(encoding=3, lang='und', desc='', text=lyrics)) audio.save()
audio.save()
else: else:
metadata = mutagen.File(audio_file) metadata = mutagen.File(audio_file)
metadata['©lyr' if audio_file.endswith('.m4a') else 'lyrics'] = lyrics_list metadata['©lyr' if audio_file.endswith('.m4a') else 'lyrics'] = [lyrics]
metadata.save() metadata.save()