pepepper/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl synced 2025-01-07 12:00:09 +09:00
Code Issues Projects Releases Wiki Activity

Fixed missing imports, ran Flake8 over the fille.

Browse Source
This commit is contained in:
tripulse 2021-04-27 12:00:50 +05:30
parent de3865fec0
commit 9ca71e76ba
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
youtube_dl/postprocessor/embedthumbnail.py
View File

@ -18,6 +18,8 @@ from ..utils import (
) )
from base64 import b64encode from base64 import b64encode
from shutil import copyfile
class EmbedThumbnailPPError(PostProcessingError): class EmbedThumbnailPPError(PostProcessingError):
pass pass
@ -129,15 +131,15 @@ class EmbedThumbnailPP(FFmpegPostProcessor):
elif info['ext'] in ['opus', 'ogg', 'flac']: elif info['ext'] in ['opus', 'ogg', 'flac']:
try: try:
from mutagen.oggvorbis import OggVorbis from mutagen.oggvorbis import OggVorbis
from mutagen.oggopus import OggOpus from mutagen.oggopus import OggOpus
from mutagen.flac import Picture, FLAC from mutagen.flac import Picture, FLAC
except ImportError: except ImportError:
raise EmbedThumbnailPPError('mutagen was not found. Please install.') raise EmbedThumbnailPPError('mutagen was not found. Please install.')
# to prevent the behaviour of in-place modification of Mutagen # to prevent the behaviour of in-place modification of Mutagen
shutil.copyfile(filename, temp_filename) copyfile(filename, temp_filename)
aufile = {'opus': OggOpus, 'flac': FLAC, 'ogg', OggVorbis}[info['ext']](temp_filename) aufile = {'opus': OggOpus, 'flac': FLAC, 'ogg': OggVorbis}[info['ext']](temp_filename)
covart = Picture() covart = Picture()
covart.data = open(thumbnail_filename, 'rb').read() covart.data = open(thumbnail_filename, 'rb').read()
@ -151,7 +153,7 @@ class EmbedThumbnailPP(FFmpegPostProcessor):
self._downloader.to_screen('[mutagen] Adding thumbnail to "%s"' % temp_filename) self._downloader.to_screen('[mutagen] Adding thumbnail to "%s"' % temp_filename)
aufile.save() aufile.save()
if not self._already_have_thumbnail: if not self._already_have_thumbnail:
os.remove(encodeFilename(thumbnail_filename)) os.remove(encodeFilename(thumbnail_filename))
os.remove(encodeFilename(filename)) os.remove(encodeFilename(filename))