115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
import mimetypes
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
from typing import Union, Dict
|
|
|
|
import mutagen
|
|
|
|
from . import abstract, parser_factory, video
|
|
|
|
|
|
class MutagenParser(abstract.AbstractParser):
|
|
def __init__(self, filename):
|
|
super().__init__(filename)
|
|
try:
|
|
if mutagen.File(self.filename) is None:
|
|
raise ValueError
|
|
except mutagen.MutagenError:
|
|
raise ValueError
|
|
|
|
def get_meta(self) -> Dict[str, Union[str, Dict]]:
|
|
f = mutagen.File(self.filename)
|
|
if f.tags:
|
|
return {k: ', '.join(map(str, v)) for k, v in f.tags.items()}
|
|
return {}
|
|
|
|
def remove_all(self) -> bool:
|
|
shutil.copy(self.filename, self.output_filename)
|
|
f = mutagen.File(self.output_filename)
|
|
try:
|
|
f.delete()
|
|
f.save()
|
|
except mutagen.MutagenError:
|
|
raise ValueError
|
|
return True
|
|
|
|
|
|
class MP3Parser(MutagenParser):
|
|
mimetypes = {'audio/mpeg', }
|
|
|
|
def get_meta(self) -> Dict[str, Union[str, Dict]]:
|
|
metadata: Dict[str, Union[str, Dict]] = dict()
|
|
meta = mutagen.File(self.filename).tags
|
|
if not meta:
|
|
return metadata
|
|
for key in meta:
|
|
if isinstance(key, tuple):
|
|
metadata[key[0]] = key[1]
|
|
continue
|
|
if not hasattr(meta[key], 'text'): # pragma: no cover
|
|
continue
|
|
metadata[key.rstrip(' \t\r\n\0')] = ', '.join(map(str, meta[key].text))
|
|
return metadata
|
|
|
|
|
|
class OGGParser(MutagenParser):
|
|
mimetypes = {'audio/ogg', }
|
|
|
|
|
|
class FLACParser(MutagenParser):
|
|
mimetypes = {'audio/flac', 'audio/x-flac'}
|
|
|
|
def remove_all(self) -> bool:
|
|
shutil.copy(self.filename, self.output_filename)
|
|
f = mutagen.File(self.output_filename)
|
|
f.clear_pictures()
|
|
f.delete()
|
|
f.save(deleteid3=True)
|
|
return True
|
|
|
|
def get_meta(self) -> Dict[str, Union[str, Dict]]:
|
|
meta = super().get_meta()
|
|
for num, picture in enumerate(mutagen.File(self.filename).pictures):
|
|
name = picture.desc if picture.desc else 'Cover %d' % num
|
|
extension = mimetypes.guess_extension(picture.mime)
|
|
if extension is None: # pragma: no cover
|
|
meta[name] = 'harmful data'
|
|
continue
|
|
|
|
_, fname = tempfile.mkstemp()
|
|
fname = fname + extension
|
|
with open(fname, 'wb') as f:
|
|
f.write(picture.data)
|
|
p, _ = parser_factory.get_parser(fname) # type: ignore
|
|
if p is None:
|
|
raise ValueError
|
|
p.sandbox = self.sandbox
|
|
# Mypy chokes on ternaries :/
|
|
meta[name] = p.get_meta() if p else 'harmful data' # type: ignore
|
|
os.remove(fname)
|
|
return meta
|
|
|
|
|
|
class WAVParser(video.AbstractFFmpegParser):
|
|
mimetypes = {'audio/x-wav', }
|
|
meta_allowlist = {'AvgBytesPerSec', 'BitsPerSample', 'Directory',
|
|
'Duration', 'Encoding', 'ExifToolVersion',
|
|
'FileAccessDate', 'FileInodeChangeDate',
|
|
'FileModifyDate', 'FileName', 'FilePermissions',
|
|
'FileSize', 'FileType', 'FileTypeExtension',
|
|
'MIMEType', 'NumChannels', 'SampleRate', 'SourceFile',
|
|
}
|
|
|
|
|
|
class AIFFParser(video.AbstractFFmpegParser):
|
|
mimetypes = {'audio/aiff', 'audio/x-aiff'}
|
|
meta_allowlist = {'AvgBytesPerSec', 'BitsPerSample', 'Directory',
|
|
'Duration', 'Encoding', 'ExifToolVersion',
|
|
'FileAccessDate', 'FileInodeChangeDate',
|
|
'FileModifyDate', 'FileName', 'FilePermissions',
|
|
'FileSize', 'FileType', 'FileTypeExtension',
|
|
'MIMEType', 'NumChannels', 'SampleRate', 'SourceFile',
|
|
'NumSampleFrames', 'SampleSize',
|
|
}
|