xZero707/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2024-11-27 06:10:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

Transform in place. Tag in mimetype.

Browse Source
This commit is contained in:
tripulse 2021-04-27 18:46:34 +05:30
parent 9ca71e76ba
commit 80a88a15ab
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
youtube_dl/postprocessor/embedthumbnail.py
View File

@ -4,6 +4,7 @@ from __future__ import unicode_literals
import os
import subprocess
import imghdr
from .ffmpeg import FFmpegPostProcessor
@ -18,7 +19,6 @@ from ..utils import (
)
from base64 import b64encode
from shutil import copyfile
class EmbedThumbnailPPError(PostProcessingError):
@ -136,13 +136,12 @@ class EmbedThumbnailPP(FFmpegPostProcessor):
except ImportError:
raise EmbedThumbnailPPError('mutagen was not found. Please install.')
# to prevent the behaviour of in-place modification of Mutagen
copyfile(filename, temp_filename)
aufile = {'opus': OggOpus, 'flac': FLAC, 'ogg': OggVorbis}[info['ext']](temp_filename)
aufile = {'opus': OggOpus, 'flac': FLAC, 'ogg': OggVorbis}[info['ext']](filename)
thumbfile = open(thumbnail_filename, 'rb')
covart = Picture()
covart.data = open(thumbnail_filename, 'rb').read()
covart.mime = 'image/' + imghdr.what(thumbfile)
covart.data = thumbfile.read()
covart.type = 3 # as the front cover
if info['ext'] == 'flac':
@ -156,8 +155,6 @@ class EmbedThumbnailPP(FFmpegPostProcessor):
if not self._already_have_thumbnail:
os.remove(encodeFilename(thumbnail_filename))
os.remove(encodeFilename(filename))
os.rename(encodeFilename(temp_filename), encodeFilename(filename))
else:
raise EmbedThumbnailPPError('Only mp3, m4a/mp4, ogg, opus and flac are supported for thumbnail embedding for now.')