From 9ca71e76baf958a22b7685cd8cf27aa1bbb2e9b4 Mon Sep 17 00:00:00 2001 From: tripulse Date: Tue, 27 Apr 2021 12:00:50 +0530 Subject: [PATCH] Fixed missing imports, ran Flake8 over the fille. --- youtube_dl/postprocessor/embedthumbnail.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/youtube_dl/postprocessor/embedthumbnail.py b/youtube_dl/postprocessor/embedthumbnail.py index c854cb541..69613ffaf 100644 --- a/youtube_dl/postprocessor/embedthumbnail.py +++ b/youtube_dl/postprocessor/embedthumbnail.py @@ -18,6 +18,8 @@ from ..utils import ( ) from base64 import b64encode +from shutil import copyfile + class EmbedThumbnailPPError(PostProcessingError): pass @@ -129,15 +131,15 @@ class EmbedThumbnailPP(FFmpegPostProcessor): elif info['ext'] in ['opus', 'ogg', 'flac']: try: from mutagen.oggvorbis import OggVorbis - from mutagen.oggopus import OggOpus - from mutagen.flac import Picture, FLAC + from mutagen.oggopus import OggOpus + from mutagen.flac import Picture, FLAC except ImportError: raise EmbedThumbnailPPError('mutagen was not found. Please install.') # 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.data = open(thumbnail_filename, 'rb').read() @@ -151,7 +153,7 @@ class EmbedThumbnailPP(FFmpegPostProcessor): self._downloader.to_screen('[mutagen] Adding thumbnail to "%s"' % temp_filename) aufile.save() - + if not self._already_have_thumbnail: os.remove(encodeFilename(thumbnail_filename)) os.remove(encodeFilename(filename))