mirror of
https://github.com/rembo10/headphones.git
synced 2026-07-20 07:54:01 +01:00
update Beets
This commit is contained in:
AdeHub
+57
-50
@@ -17,74 +17,78 @@ automatically whenever tags are written.
|
||||
"""
|
||||
|
||||
|
||||
from beets.plugins import BeetsPlugin
|
||||
from beets import ui
|
||||
from beets import util
|
||||
from beets import config
|
||||
import mediafile
|
||||
import mutagen
|
||||
|
||||
from beets import config, ui, util
|
||||
from beets.plugins import BeetsPlugin
|
||||
|
||||
_MUTAGEN_FORMATS = {
|
||||
'asf': 'ASF',
|
||||
'apev2': 'APEv2File',
|
||||
'flac': 'FLAC',
|
||||
'id3': 'ID3FileType',
|
||||
'mp3': 'MP3',
|
||||
'mp4': 'MP4',
|
||||
'oggflac': 'OggFLAC',
|
||||
'oggspeex': 'OggSpeex',
|
||||
'oggtheora': 'OggTheora',
|
||||
'oggvorbis': 'OggVorbis',
|
||||
'oggopus': 'OggOpus',
|
||||
'trueaudio': 'TrueAudio',
|
||||
'wavpack': 'WavPack',
|
||||
'monkeysaudio': 'MonkeysAudio',
|
||||
'optimfrog': 'OptimFROG',
|
||||
"asf": "ASF",
|
||||
"apev2": "APEv2File",
|
||||
"flac": "FLAC",
|
||||
"id3": "ID3FileType",
|
||||
"mp3": "MP3",
|
||||
"mp4": "MP4",
|
||||
"oggflac": "OggFLAC",
|
||||
"oggspeex": "OggSpeex",
|
||||
"oggtheora": "OggTheora",
|
||||
"oggvorbis": "OggVorbis",
|
||||
"oggopus": "OggOpus",
|
||||
"trueaudio": "TrueAudio",
|
||||
"wavpack": "WavPack",
|
||||
"monkeysaudio": "MonkeysAudio",
|
||||
"optimfrog": "OptimFROG",
|
||||
}
|
||||
|
||||
|
||||
class ScrubPlugin(BeetsPlugin):
|
||||
"""Removes extraneous metadata from files' tags."""
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.config.add({
|
||||
'auto': True,
|
||||
})
|
||||
self.config.add(
|
||||
{
|
||||
"auto": True,
|
||||
}
|
||||
)
|
||||
|
||||
if self.config['auto']:
|
||||
if self.config["auto"]:
|
||||
self.register_listener("import_task_files", self.import_task_files)
|
||||
|
||||
def commands(self):
|
||||
def scrub_func(lib, opts, args):
|
||||
# Walk through matching files and remove tags.
|
||||
for item in lib.items(ui.decargs(args)):
|
||||
self._log.info('scrubbing: {0}',
|
||||
util.displayable_path(item.path))
|
||||
self._log.info(
|
||||
"scrubbing: {0}", util.displayable_path(item.path)
|
||||
)
|
||||
self._scrub_item(item, opts.write)
|
||||
|
||||
scrub_cmd = ui.Subcommand('scrub', help='clean audio tags')
|
||||
scrub_cmd = ui.Subcommand("scrub", help="clean audio tags")
|
||||
scrub_cmd.parser.add_option(
|
||||
'-W', '--nowrite', dest='write',
|
||||
action='store_false', default=True,
|
||||
help='leave tags empty')
|
||||
"-W",
|
||||
"--nowrite",
|
||||
dest="write",
|
||||
action="store_false",
|
||||
default=True,
|
||||
help="leave tags empty",
|
||||
)
|
||||
scrub_cmd.func = scrub_func
|
||||
|
||||
return [scrub_cmd]
|
||||
|
||||
@staticmethod
|
||||
def _mutagen_classes():
|
||||
"""Get a list of file type classes from the Mutagen module.
|
||||
"""
|
||||
"""Get a list of file type classes from the Mutagen module."""
|
||||
classes = []
|
||||
for modname, clsname in _MUTAGEN_FORMATS.items():
|
||||
mod = __import__(f'mutagen.{modname}',
|
||||
fromlist=[clsname])
|
||||
mod = __import__(f"mutagen.{modname}", fromlist=[clsname])
|
||||
classes.append(getattr(mod, clsname))
|
||||
return classes
|
||||
|
||||
def _scrub(self, path):
|
||||
"""Remove all tags from a file.
|
||||
"""
|
||||
"""Remove all tags from a file."""
|
||||
for cls in self._mutagen_classes():
|
||||
# Try opening the file with this type, but just skip in the
|
||||
# event of any error.
|
||||
@@ -106,21 +110,22 @@ class ScrubPlugin(BeetsPlugin):
|
||||
del f[tag]
|
||||
f.save()
|
||||
except (OSError, mutagen.MutagenError) as exc:
|
||||
self._log.error('could not scrub {0}: {1}',
|
||||
util.displayable_path(path), exc)
|
||||
self._log.error(
|
||||
"could not scrub {0}: {1}", util.displayable_path(path), exc
|
||||
)
|
||||
|
||||
def _scrub_item(self, item, restore=True):
|
||||
def _scrub_item(self, item, restore):
|
||||
"""Remove tags from an Item's associated file and, if `restore`
|
||||
is enabled, write the database's tags back to the file.
|
||||
"""
|
||||
# Get album art if we need to restore it.
|
||||
if restore:
|
||||
try:
|
||||
mf = mediafile.MediaFile(util.syspath(item.path),
|
||||
config['id3v23'].get(bool))
|
||||
mf = mediafile.MediaFile(
|
||||
util.syspath(item.path), config["id3v23"].get(bool)
|
||||
)
|
||||
except mediafile.UnreadableFileError as exc:
|
||||
self._log.error('could not open file to scrub: {0}',
|
||||
exc)
|
||||
self._log.error("could not open file to scrub: {0}", exc)
|
||||
return
|
||||
images = mf.images
|
||||
|
||||
@@ -129,21 +134,23 @@ class ScrubPlugin(BeetsPlugin):
|
||||
|
||||
# Restore tags, if enabled.
|
||||
if restore:
|
||||
self._log.debug('writing new tags after scrub')
|
||||
self._log.debug("writing new tags after scrub")
|
||||
item.try_write()
|
||||
if images:
|
||||
self._log.debug('restoring art')
|
||||
self._log.debug("restoring art")
|
||||
try:
|
||||
mf = mediafile.MediaFile(util.syspath(item.path),
|
||||
config['id3v23'].get(bool))
|
||||
mf = mediafile.MediaFile(
|
||||
util.syspath(item.path), config["id3v23"].get(bool)
|
||||
)
|
||||
mf.images = images
|
||||
mf.save()
|
||||
except mediafile.UnreadableFileError as exc:
|
||||
self._log.error('could not write tags: {0}', exc)
|
||||
self._log.error("could not write tags: {0}", exc)
|
||||
|
||||
def import_task_files(self, session, task):
|
||||
"""Automatically scrub imported files."""
|
||||
for item in task.imported_items():
|
||||
self._log.debug('auto-scrubbing {0}',
|
||||
util.displayable_path(item.path))
|
||||
self._scrub_item(item)
|
||||
self._log.debug(
|
||||
"auto-scrubbing {0}", util.displayable_path(item.path)
|
||||
)
|
||||
self._scrub_item(item, ui.should_write())
|
||||
|
||||
Reference in New Issue
Block a user