]>
jfr.im git - yt-dlp.git/blob - youtube_dl/postprocessor/embedthumbnail.py
1 # -*- coding: utf-8 -*-
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 raise EmbedThumbnailPPError('Thumbnail was not found. Nothing to do.')
36 thumbnail_filename
= info
['thumbnails'][-1]['filename']
38 if not os
.path
.exists(encodeFilename(thumbnail_filename
)):
39 self
._downloader
.report_warning(
40 'Skipping embedding the thumbnail because the file is missing.')
43 if info
['ext'] in ('mp3', 'mkv'):
45 '-c', 'copy', '-map', '0', '-map', '1',
46 '-metadata:s:v', 'title="Album cover"', '-metadata:s:v', 'comment="Cover (Front)"']
48 self
._downloader
.to_screen('[ffmpeg] Adding thumbnail to "%s"' % filename
)
50 self
.run_ffmpeg_multiple_files([filename
, thumbnail_filename
], temp_filename
, options
)
52 if not self
._already
_have
_thumbnail
:
53 os
.remove(encodeFilename(thumbnail_filename
))
54 os
.remove(encodeFilename(filename
))
55 os
.rename(encodeFilename(temp_filename
), encodeFilename(filename
))
57 elif info
['ext'] in ['m4a', 'mp4']:
58 if not check_executable('AtomicParsley', ['-v']):
59 raise EmbedThumbnailPPError('AtomicParsley was not found. Please install.')
61 cmd
= [encodeFilename('AtomicParsley', True),
62 encodeFilename(filename
, True),
63 encodeArgument('--artwork'),
64 encodeFilename(thumbnail_filename
, True),
66 encodeFilename(temp_filename
, True)]
68 self
._downloader
.to_screen('[atomicparsley] Adding thumbnail to "%s"' % filename
)
70 if self
._downloader
.params
.get('verbose', False):
71 self
._downloader
.to_screen('[debug] AtomicParsley command line: %s' % shell_quote(cmd
))
73 p
= subprocess
.Popen(cmd
, stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
)
74 stdout
, stderr
= p
.communicate()
77 msg
= stderr
.decode('utf-8', 'replace').strip()
78 raise EmbedThumbnailPPError(msg
)
80 if not self
._already
_have
_thumbnail
:
81 os
.remove(encodeFilename(thumbnail_filename
))
82 # for formats that don't support thumbnails (like 3gp) AtomicParsley
83 # won't create to the temporary file
84 if b
'No changes' in stdout
:
85 self
._downloader
.report_warning('The file format doesn\'t support embedding a thumbnail')
87 os
.remove(encodeFilename(filename
))
88 os
.rename(encodeFilename(temp_filename
), encodeFilename(filename
))
90 raise EmbedThumbnailPPError('Only mp3 and m4a/mp4 are supported for thumbnail embedding for now.')