Fixed missing imports, ran Flake8 over the fille.

This commit is contained in:
tripulse 2021-04-27 12:00:50 +05:30
parent de3865fec0
commit 9ca71e76ba

View File

@ -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()