]>
Commit | Line | Data |
---|---|---|
dcdb292f | 1 | # coding: utf-8 |
ddbed364 | 2 | from __future__ import unicode_literals |
3 | ||
4 | ||
5 | import os | |
6 | import subprocess | |
06167fbb | 7 | import struct |
8 | import re | |
9 | import base64 | |
10 | ||
11 | try: | |
12 | import mutagen | |
13 | _has_mutagen = True | |
14 | except ImportError: | |
15 | _has_mutagen = False | |
ddbed364 | 16 | |
31fd9c76 | 17 | from .ffmpeg import FFmpegPostProcessor |
18 | ||
ddbed364 | 19 | from ..utils import ( |
20 | check_executable, | |
2cc6d135 | 21 | encodeArgument, |
ddbed364 | 22 | encodeFilename, |
06167fbb | 23 | error_to_compat_str, |
ddbed364 | 24 | PostProcessingError, |
25 | prepend_extension, | |
06167fbb | 26 | process_communicate_or_kill, |
bff857a8 | 27 | replace_extension, |
f5b1bca9 | 28 | shell_quote, |
ddbed364 | 29 | ) |
30 | ||
31 | ||
32 | class EmbedThumbnailPPError(PostProcessingError): | |
33 | pass | |
34 | ||
35 | ||
31fd9c76 | 36 | class EmbedThumbnailPP(FFmpegPostProcessor): |
1b77b347 | 37 | |
8e595397 YCH |
38 | def __init__(self, downloader=None, already_have_thumbnail=False): |
39 | super(EmbedThumbnailPP, self).__init__(downloader) | |
40 | self._already_have_thumbnail = already_have_thumbnail | |
41 | ||
ddbed364 | 42 | def run(self, info): |
43 | filename = info['filepath'] | |
44 | temp_filename = prepend_extension(filename, 'temp') | |
de6000d9 | 45 | files_to_delete = [] |
ddbed364 | 46 | |
8e595397 | 47 | if not info.get('thumbnails'): |
1b77b347 | 48 | self.to_screen('There aren\'t any thumbnails to embed') |
b5cbe3d6 | 49 | return [], info |
ddbed364 | 50 | |
8e595397 | 51 | thumbnail_filename = info['thumbnails'][-1]['filename'] |
ddbed364 | 52 | |
c33a8639 | 53 | if not os.path.exists(encodeFilename(thumbnail_filename)): |
f446cc66 | 54 | self.report_warning('Skipping embedding the thumbnail because the file is missing.') |
c33a8639 YCH |
55 | return [], info |
56 | ||
bff857a8 S |
57 | def is_webp(path): |
58 | with open(encodeFilename(path), 'rb') as f: | |
59 | b = f.read(12) | |
60 | return b[0:4] == b'RIFF' and b[8:] == b'WEBP' | |
61 | ||
62 | # Correct extension for WebP file with wrong extension (see #25687, #25717) | |
63 | _, thumbnail_ext = os.path.splitext(thumbnail_filename) | |
64 | if thumbnail_ext: | |
65 | thumbnail_ext = thumbnail_ext[1:].lower() | |
66 | if thumbnail_ext != 'webp' and is_webp(thumbnail_filename): | |
1b77b347 | 67 | self.to_screen('Correcting extension to webp and escaping path for thumbnail "%s"' % thumbnail_filename) |
bff857a8 S |
68 | thumbnail_webp_filename = replace_extension(thumbnail_filename, 'webp') |
69 | os.rename(encodeFilename(thumbnail_filename), encodeFilename(thumbnail_webp_filename)) | |
70 | thumbnail_filename = thumbnail_webp_filename | |
71 | thumbnail_ext = 'webp' | |
72 | ||
73 | # Convert unsupported thumbnail formats to JPEG (see #25687, #25717) | |
74 | if thumbnail_ext not in ['jpg', 'png']: | |
75 | # NB: % is supposed to be escaped with %% but this does not work | |
76 | # for input files so working around with standard substitution | |
77 | escaped_thumbnail_filename = thumbnail_filename.replace('%', '#') | |
78 | os.rename(encodeFilename(thumbnail_filename), encodeFilename(escaped_thumbnail_filename)) | |
79 | escaped_thumbnail_jpg_filename = replace_extension(escaped_thumbnail_filename, 'jpg') | |
1b77b347 | 80 | self.to_screen('Converting thumbnail "%s" to JPEG' % escaped_thumbnail_filename) |
bff857a8 | 81 | self.run_ffmpeg(escaped_thumbnail_filename, escaped_thumbnail_jpg_filename, ['-bsf:v', 'mjpeg2jpeg']) |
de6000d9 | 82 | files_to_delete.append(escaped_thumbnail_filename) |
bff857a8 S |
83 | thumbnail_jpg_filename = replace_extension(thumbnail_filename, 'jpg') |
84 | # Rename back to unescaped for further processing | |
85 | os.rename(encodeFilename(escaped_thumbnail_jpg_filename), encodeFilename(thumbnail_jpg_filename)) | |
86 | thumbnail_filename = thumbnail_jpg_filename | |
06167fbb | 87 | thumbnail_ext = 'jpg' |
777d5a45 | 88 | |
67002a5a | 89 | success = True |
8e2915d7 | 90 | if info['ext'] == 'mp3': |
92995e62 | 91 | options = [ |
c76eb41b | 92 | '-c', 'copy', '-map', '0:0', '-map', '1:0', '-id3v2_version', '3', |
e51f368c | 93 | '-metadata:s:v', 'title="Album cover"', '-metadata:s:v', 'comment="Cover (front)"'] |
ddbed364 | 94 | |
1b77b347 | 95 | self.to_screen('Adding thumbnail to "%s"' % filename) |
bb8ca1d1 | 96 | self.run_ffmpeg_multiple_files([filename, thumbnail_filename], temp_filename, options) |
ddbed364 | 97 | |
06167fbb | 98 | elif info['ext'] in ['mkv', 'mka']: |
99 | options = ['-c', 'copy', '-map', '0', '-dn'] | |
ddbed364 | 100 | |
06167fbb | 101 | mimetype = 'image/%s' % ('png' if thumbnail_ext == 'png' else 'jpeg') |
102 | old_stream, new_stream = self.get_stream_number( | |
103 | filename, ('tags', 'mimetype'), mimetype) | |
104 | if old_stream is not None: | |
105 | options.extend(['-map', '-0:%d' % old_stream]) | |
106 | new_stream -= 1 | |
107 | options.extend([ | |
108 | '-attach', thumbnail_filename, | |
109 | '-metadata:s:%d' % new_stream, 'mimetype=%s' % mimetype, | |
110 | '-metadata:s:%d' % new_stream, 'filename=cover.%s' % thumbnail_ext]) | |
ddbed364 | 111 | |
1b77b347 | 112 | self.to_screen('Adding thumbnail to "%s"' % filename) |
06167fbb | 113 | self.run_ffmpeg(filename, temp_filename, options) |
114 | ||
115 | elif info['ext'] in ['m4a', 'mp4', 'mov']: | |
116 | try: | |
117 | options = ['-c', 'copy', '-map', '0', '-dn', '-map', '1'] | |
118 | ||
119 | old_stream, new_stream = self.get_stream_number( | |
120 | filename, ('disposition', 'attached_pic'), 1) | |
121 | if old_stream is not None: | |
122 | options.extend(['-map', '-0:%d' % old_stream]) | |
123 | new_stream -= 1 | |
124 | options.extend(['-disposition:%s' % new_stream, 'attached_pic']) | |
125 | ||
126 | self.to_screen('Adding thumbnail to "%s"' % filename) | |
127 | self.run_ffmpeg_multiple_files([filename, thumbnail_filename], temp_filename, options) | |
128 | ||
129 | except PostProcessingError as err: | |
130 | self.report_warning('unable to embed using ffprobe & ffmpeg; %s' % error_to_compat_str(err)) | |
131 | if not check_executable('AtomicParsley', ['-v']): | |
132 | raise EmbedThumbnailPPError('AtomicParsley was not found. Please install.') | |
133 | ||
134 | cmd = [encodeFilename('AtomicParsley', True), | |
135 | encodeFilename(filename, True), | |
136 | encodeArgument('--artwork'), | |
137 | encodeFilename(thumbnail_filename, True), | |
138 | encodeArgument('-o'), | |
139 | encodeFilename(temp_filename, True)] | |
140 | cmd += [encodeArgument(o) for o in self._configuration_args(exe='AtomicParsley')] | |
141 | ||
142 | self.to_screen('Adding thumbnail to "%s"' % filename) | |
143 | self.write_debug('AtomicParsley command line: %s' % shell_quote(cmd)) | |
144 | p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) | |
145 | stdout, stderr = process_communicate_or_kill(p) | |
146 | if p.returncode != 0: | |
147 | msg = stderr.decode('utf-8', 'replace').strip() | |
148 | raise EmbedThumbnailPPError(msg) | |
149 | # for formats that don't support thumbnails (like 3gp) AtomicParsley | |
150 | # won't create to the temporary file | |
151 | if b'No changes' in stdout: | |
152 | self.report_warning('The file format doesn\'t support embedding a thumbnail') | |
153 | success = False | |
154 | ||
155 | elif info['ext'] in ['ogg', 'opus']: | |
156 | if not _has_mutagen: | |
e38df8f9 | 157 | raise EmbedThumbnailPPError('module mutagen was not found. Please install using `python -m pip install mutagen`') |
e3b771a8 | 158 | self.to_screen('Adding thumbnail to "%s"' % filename) |
159 | ||
06167fbb | 160 | size_regex = r',\s*(?P<w>\d+)x(?P<h>\d+)\s*[,\[]' |
ece8a2a1 | 161 | size_result = self.run_ffmpeg(thumbnail_filename, thumbnail_filename, ['-hide_banner']) |
06167fbb | 162 | mobj = re.search(size_regex, size_result) |
163 | width, height = int(mobj.group('w')), int(mobj.group('h')) | |
164 | mimetype = ('image/%s' % ('png' if thumbnail_ext == 'png' else 'jpeg')).encode('ascii') | |
165 | ||
166 | # https://xiph.org/flac/format.html#metadata_block_picture | |
167 | data = bytearray() | |
168 | data += struct.pack('>II', 3, len(mimetype)) | |
169 | data += mimetype | |
170 | data += struct.pack('>IIIIII', 0, width, height, 8, 0, os.stat(thumbnail_filename).st_size) # 32 if png else 24 | |
171 | ||
172 | fin = open(thumbnail_filename, "rb") | |
173 | data += fin.read() | |
174 | fin.close() | |
175 | ||
176 | temp_filename = filename | |
177 | f = mutagen.File(temp_filename) | |
178 | f.tags['METADATA_BLOCK_PICTURE'] = base64.b64encode(data).decode('ascii') | |
179 | f.save() | |
67002a5a | 180 | |
ddbed364 | 181 | else: |
06167fbb | 182 | raise EmbedThumbnailPPError('Supported filetypes for thumbnail embedding are: mp3, mkv/mka, ogg/opus, m4a/mp4/mov') |
ddbed364 | 183 | |
06167fbb | 184 | if success and temp_filename != filename: |
67002a5a | 185 | os.remove(encodeFilename(filename)) |
186 | os.rename(encodeFilename(temp_filename), encodeFilename(filename)) | |
de6000d9 | 187 | if self._already_have_thumbnail: |
188 | info['__files_to_move'][thumbnail_filename] = replace_extension( | |
189 | info['__thumbnail_filename'], os.path.splitext(thumbnail_filename)[1][1:]) | |
190 | else: | |
191 | files_to_delete.append(thumbnail_filename) | |
67002a5a | 192 | return files_to_delete, info |