]>
Commit | Line | Data |
---|---|---|
dcdb292f | 1 | # coding: utf-8 |
ddbed364 | 2 | from __future__ import unicode_literals |
3 | ||
4 | ||
5 | import os | |
6 | import subprocess | |
7 | ||
31fd9c76 | 8 | from .ffmpeg import FFmpegPostProcessor |
9 | ||
ddbed364 | 10 | from ..utils import ( |
11 | check_executable, | |
2cc6d135 | 12 | encodeArgument, |
ddbed364 | 13 | encodeFilename, |
14 | PostProcessingError, | |
15 | prepend_extension, | |
bff857a8 | 16 | replace_extension, |
f5b1bca9 | 17 | shell_quote, |
18 | process_communicate_or_kill, | |
ddbed364 | 19 | ) |
20 | ||
21 | ||
22 | class EmbedThumbnailPPError(PostProcessingError): | |
23 | pass | |
24 | ||
25 | ||
31fd9c76 | 26 | class EmbedThumbnailPP(FFmpegPostProcessor): |
1b77b347 | 27 | PP_NAME = 'EmbedThumbnail' |
28 | ||
8e595397 YCH |
29 | def __init__(self, downloader=None, already_have_thumbnail=False): |
30 | super(EmbedThumbnailPP, self).__init__(downloader) | |
31 | self._already_have_thumbnail = already_have_thumbnail | |
32 | ||
ddbed364 | 33 | def run(self, info): |
34 | filename = info['filepath'] | |
35 | temp_filename = prepend_extension(filename, 'temp') | |
ddbed364 | 36 | |
8e595397 | 37 | if not info.get('thumbnails'): |
1b77b347 | 38 | self.to_screen('There aren\'t any thumbnails to embed') |
b5cbe3d6 | 39 | return [], info |
ddbed364 | 40 | |
8e595397 | 41 | thumbnail_filename = info['thumbnails'][-1]['filename'] |
ddbed364 | 42 | |
c33a8639 | 43 | if not os.path.exists(encodeFilename(thumbnail_filename)): |
f446cc66 | 44 | self.report_warning('Skipping embedding the thumbnail because the file is missing.') |
c33a8639 YCH |
45 | return [], info |
46 | ||
bff857a8 S |
47 | def is_webp(path): |
48 | with open(encodeFilename(path), 'rb') as f: | |
49 | b = f.read(12) | |
50 | return b[0:4] == b'RIFF' and b[8:] == b'WEBP' | |
51 | ||
52 | # Correct extension for WebP file with wrong extension (see #25687, #25717) | |
53 | _, thumbnail_ext = os.path.splitext(thumbnail_filename) | |
54 | if thumbnail_ext: | |
55 | thumbnail_ext = thumbnail_ext[1:].lower() | |
56 | if thumbnail_ext != 'webp' and is_webp(thumbnail_filename): | |
1b77b347 | 57 | self.to_screen('Correcting extension to webp and escaping path for thumbnail "%s"' % thumbnail_filename) |
bff857a8 S |
58 | thumbnail_webp_filename = replace_extension(thumbnail_filename, 'webp') |
59 | os.rename(encodeFilename(thumbnail_filename), encodeFilename(thumbnail_webp_filename)) | |
60 | thumbnail_filename = thumbnail_webp_filename | |
61 | thumbnail_ext = 'webp' | |
62 | ||
63 | # Convert unsupported thumbnail formats to JPEG (see #25687, #25717) | |
64 | if thumbnail_ext not in ['jpg', 'png']: | |
65 | # NB: % is supposed to be escaped with %% but this does not work | |
66 | # for input files so working around with standard substitution | |
67 | escaped_thumbnail_filename = thumbnail_filename.replace('%', '#') | |
68 | os.rename(encodeFilename(thumbnail_filename), encodeFilename(escaped_thumbnail_filename)) | |
69 | escaped_thumbnail_jpg_filename = replace_extension(escaped_thumbnail_filename, 'jpg') | |
1b77b347 | 70 | self.to_screen('Converting thumbnail "%s" to JPEG' % escaped_thumbnail_filename) |
bff857a8 S |
71 | self.run_ffmpeg(escaped_thumbnail_filename, escaped_thumbnail_jpg_filename, ['-bsf:v', 'mjpeg2jpeg']) |
72 | os.remove(encodeFilename(escaped_thumbnail_filename)) | |
73 | thumbnail_jpg_filename = replace_extension(thumbnail_filename, 'jpg') | |
74 | # Rename back to unescaped for further processing | |
75 | os.rename(encodeFilename(escaped_thumbnail_jpg_filename), encodeFilename(thumbnail_jpg_filename)) | |
76 | thumbnail_filename = thumbnail_jpg_filename | |
777d5a45 | 77 | |
67002a5a | 78 | success = True |
8e2915d7 | 79 | if info['ext'] == 'mp3': |
92995e62 | 80 | options = [ |
c76eb41b | 81 | '-c', 'copy', '-map', '0:0', '-map', '1:0', '-id3v2_version', '3', |
e51f368c | 82 | '-metadata:s:v', 'title="Album cover"', '-metadata:s:v', 'comment="Cover (front)"'] |
ddbed364 | 83 | |
1b77b347 | 84 | self.to_screen('Adding thumbnail to "%s"' % filename) |
bb8ca1d1 | 85 | self.run_ffmpeg_multiple_files([filename, thumbnail_filename], temp_filename, options) |
ddbed364 | 86 | |
62566a41 | 87 | elif info['ext'] == 'mkv': |
62566a41 | 88 | options = [ |
67002a5a | 89 | '-c', 'copy', '-map', '0', '-dn', '-attach', thumbnail_filename, |
90 | '-metadata:s:t', 'mimetype=image/jpeg', '-metadata:s:t', 'filename=cover.jpg'] | |
62566a41 | 91 | |
1b77b347 | 92 | self.to_screen('Adding thumbnail to "%s"' % filename) |
62566a41 | 93 | self.run_ffmpeg_multiple_files([filename], temp_filename, options) |
94 | ||
d6aa68ce | 95 | elif info['ext'] in ['m4a', 'mp4']: |
ddbed364 | 96 | if not check_executable('AtomicParsley', ['-v']): |
97 | raise EmbedThumbnailPPError('AtomicParsley was not found. Please install.') | |
98 | ||
2cc6d135 YCH |
99 | cmd = [encodeFilename('AtomicParsley', True), |
100 | encodeFilename(filename, True), | |
101 | encodeArgument('--artwork'), | |
102 | encodeFilename(thumbnail_filename, True), | |
103 | encodeArgument('-o'), | |
104 | encodeFilename(temp_filename, True)] | |
ddbed364 | 105 | |
1b77b347 | 106 | self.to_screen('Adding thumbnail to "%s"' % filename) |
806b05cf | 107 | self.write_debug('AtomicParsley command line: %s' % shell_quote(cmd)) |
ddbed364 | 108 | |
109 | p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE) | |
f5b1bca9 | 110 | stdout, stderr = process_communicate_or_kill(p) |
ddbed364 | 111 | |
112 | if p.returncode != 0: | |
113 | msg = stderr.decode('utf-8', 'replace').strip() | |
114 | raise EmbedThumbnailPPError(msg) | |
ddbed364 | 115 | # for formats that don't support thumbnails (like 3gp) AtomicParsley |
116 | # won't create to the temporary file | |
117 | if b'No changes' in stdout: | |
f446cc66 | 118 | self.report_warning('The file format doesn\'t support embedding a thumbnail') |
67002a5a | 119 | success = False |
120 | ||
ddbed364 | 121 | else: |
958804ad | 122 | raise EmbedThumbnailPPError('Only mp3, mkv, m4a and mp4 are supported for thumbnail embedding for now.') |
ddbed364 | 123 | |
67002a5a | 124 | if success: |
125 | os.remove(encodeFilename(filename)) | |
126 | os.rename(encodeFilename(temp_filename), encodeFilename(filename)) | |
127 | ||
128 | files_to_delete = [] if self._already_have_thumbnail else [thumbnail_filename] | |
129 | return files_to_delete, info |