mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 08:39:17 +01:00
93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
#!/usr/bin/python
|
|
# coding=utf-8
|
|
|
|
# Base Python File (badfiles.py)
|
|
# Created: Tue 11 Aug 2015 10:46:34 PM CEST
|
|
# Version: 1.0
|
|
#
|
|
# This Python script was developped by François-Xavier Thomas.
|
|
# You are free to copy, adapt or modify it.
|
|
# If you do so, however, leave my name somewhere in the credits, I'd appreciate it ;)
|
|
#
|
|
# (ɔ) François-Xavier Thomas <fx.thomas@gmail.com>
|
|
|
|
from beets.plugins import BeetsPlugin
|
|
from beets.ui import Subcommand
|
|
from beets.util import displayable_path
|
|
from beets import ui
|
|
from subprocess import check_output, CalledProcessError, list2cmdline, STDOUT
|
|
import shlex
|
|
import os
|
|
import errno
|
|
import sys
|
|
|
|
class BadFiles(BeetsPlugin):
|
|
|
|
def run_command(self, cmd):
|
|
self._log.debug(u"running command: %s" % displayable_path(list2cmdline(cmd)))
|
|
try:
|
|
output = check_output(cmd, stderr=STDOUT)
|
|
return 0, [line for line in output.split("\n") if line]
|
|
except CalledProcessError as e:
|
|
return 1, [line for line in e.output.split("\n") if line]
|
|
except OSError as e:
|
|
if e.errno == errno.ENOENT:
|
|
ui.print_("Command '%s' does not exit. Is it installed?" % cmd[0])
|
|
sys.exit(1)
|
|
else:
|
|
raise
|
|
|
|
def check_mp3val(self, path):
|
|
errors, output = self.run_command(["mp3val", path])
|
|
if errors == 0:
|
|
output = [line for line in output if line.startswith("WARNING:")]
|
|
errors = sum(1 for line in output if line.startswith("WARNING:"))
|
|
return errors, output
|
|
|
|
def check_flac(self, path):
|
|
return self.run_command(["flac", "-wst", path])
|
|
|
|
def check_custom(self, command):
|
|
def checker(path):
|
|
cmd = shlex.split(command)
|
|
cmd.append(path)
|
|
return self.run_command(cmd)
|
|
return checker
|
|
|
|
def get_checker(self, ext):
|
|
ext = ext.lower()
|
|
command = self.config['commands'].get().get(ext)
|
|
if command:
|
|
return self.check_custom(command)
|
|
elif ext == "mp3":
|
|
return self.check_mp3val
|
|
elif ext == "flac":
|
|
return self.check_flac
|
|
|
|
def check_bad(self, lib, opts, args):
|
|
for item in lib.items(args):
|
|
|
|
# First check if the path exists. If not, should run 'beets update'
|
|
# to cleanup your library.
|
|
dpath = displayable_path(item.path)
|
|
self._log.debug(u"checking path: %s" % dpath)
|
|
if not os.path.exists(item.path):
|
|
ui.print_(u"%s: file does not exist" % dpath)
|
|
|
|
# Run the checker against the file if one is found
|
|
ext = os.path.splitext(item.path)[1][1:]
|
|
checker = self.get_checker(ext)
|
|
if not checker:
|
|
continue
|
|
errors, output = checker(item.path)
|
|
if errors == 0:
|
|
ui.print_(u"%s: ok" % dpath)
|
|
else:
|
|
ui.print_(u"%s: checker found %d errors or warnings" % (dpath, errors))
|
|
for line in output:
|
|
ui.print_(u" %s" % displayable_path(line))
|
|
|
|
def commands(self):
|
|
bad_command = Subcommand('bad', help='check for corrupt or missing files')
|
|
bad_command.func = self.check_bad
|
|
return [bad_command]
|