]>
Raphaƫl G. Git Repositories - youtubedl/blob - youtube_dl/postprocessor/embedthumbnail.py
2 from __future__
import unicode_literals
8 from .ffmpeg
import FFmpegPostProcessor
20 class EmbedThumbnailPPError(PostProcessingError
):
24 class EmbedThumbnailPP(FFmpegPostProcessor
):
25 def __init__(self
, downloader
=None, already_have_thumbnail
=False):
26 super(EmbedThumbnailPP
, self
).__init
__(downloader
)
27 self
._already
_have
_thumbnail
= already_have_thumbnail
30 filename
= info
['filepath']
31 temp_filename
= prepend_extension(filename
, 'temp')
33 if not info
.get('thumbnails'):
34 self
._downloader
.to_screen('[embedthumbnail] There aren\'t any thumbnails to embed')
37 thumbnail_filename
= info
['thumbnails'][-1]['filename']
39 if not os
.path
.exists(encodeFilename(thumbnail_filename
)):
40 self
._downloader
.report_warning(
41 'Skipping embedding the thumbnail because the file is missing.')
44 if info
['ext'] == 'mp3':
46 '-c', 'copy', '-map', '0', '-map', '1',
47 '-metadata:s:v', 'title="Album cover"', '-metadata:s:v', 'comment="Cover (Front)"']
49 self
._downloader
.to_screen('[ffmpeg] Adding thumbnail to "%s"' % filename
)
51 self
.run_ffmpeg_multiple_files([filename
, thumbnail_filename
], temp_filename
, options
)
53 if not self
._already
_have
_thumbnail
:
54 os
.remove(encodeFilename(thumbnail_filename
))
55 os
.remove(encodeFilename(filename
))
56 os
.rename(encodeFilename(temp_filename
), encodeFilename(filename
))
58 elif info
['ext'] in ['m4a', 'mp4']:
59 if not check_executable('AtomicParsley', ['-v']):
60 raise EmbedThumbnailPPError('AtomicParsley was not found. Please install.')
62 cmd
= [encodeFilename('AtomicParsley', True),
63 encodeFilename(filename
, True),
64 encodeArgument('--artwork'),
65 encodeFilename(thumbnail_filename
, True),
67 encodeFilename(temp_filename
, True)]
69 self
._downloader
.to_screen('[atomicparsley] Adding thumbnail to "%s"' % filename
)
71 if self
._downloader
.params
.get('verbose', False):
72 self
._downloader
.to_screen('[debug] AtomicParsley command line: %s' % shell_quote(cmd
))
74 p
= subprocess
.Popen(cmd
, stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
75 stdout
, stderr
= p
.communicate()
78 msg
= stderr
.decode('utf-8', 'replace').strip()
79 raise EmbedThumbnailPPError(msg
)
81 if not self
._already
_have
_thumbnail
:
82 os
.remove(encodeFilename(thumbnail_filename
))
83 # for formats that don't support thumbnails (like 3gp) AtomicParsley
84 # won't create to the temporary file
85 if b
'No changes' in stdout
:
86 self
._downloader
.report_warning('The file format doesn\'t support embedding a thumbnail')
88 os
.remove(encodeFilename(filename
))
89 os
.rename(encodeFilename(temp_filename
), encodeFilename(filename
))
91 raise EmbedThumbnailPPError('Only mp3 and m4a/mp4 are supported for thumbnail embedding for now.')