mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 08:39:17 +01:00
291 lines
11 KiB
Python
291 lines
11 KiB
Python
# This file is part of beets.
|
|
# Copyright 2015, Adrian Sampson.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be
|
|
# included in all copies or substantial portions of the Software.
|
|
|
|
"""Allows beets to embed album art into file metadata."""
|
|
from __future__ import (division, absolute_import, print_function,
|
|
unicode_literals)
|
|
|
|
import os.path
|
|
import imghdr
|
|
import subprocess
|
|
import platform
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
from beets.plugins import BeetsPlugin
|
|
from beets import mediafile
|
|
from beets import ui
|
|
from beets.ui import decargs
|
|
from beets.util import syspath, normpath, displayable_path
|
|
from beets.util.artresizer import ArtResizer
|
|
from beets import config
|
|
|
|
|
|
class EmbedCoverArtPlugin(BeetsPlugin):
|
|
"""Allows albumart to be embedded into the actual files.
|
|
"""
|
|
def __init__(self):
|
|
super(EmbedCoverArtPlugin, self).__init__()
|
|
self.config.add({
|
|
'maxwidth': 0,
|
|
'auto': True,
|
|
'compare_threshold': 0,
|
|
'ifempty': False,
|
|
})
|
|
|
|
if self.config['maxwidth'].get(int) and not ArtResizer.shared.local:
|
|
self.config['maxwidth'] = 0
|
|
self._log.warning(u"ImageMagick or PIL not found; "
|
|
u"'maxwidth' option ignored")
|
|
if self.config['compare_threshold'].get(int) and not \
|
|
ArtResizer.shared.can_compare:
|
|
self.config['compare_threshold'] = 0
|
|
self._log.warning(u"ImageMagick 6.8.7 or higher not installed; "
|
|
u"'compare_threshold' option ignored")
|
|
|
|
self.register_listener('album_imported', self.album_imported)
|
|
|
|
def commands(self):
|
|
# Embed command.
|
|
embed_cmd = ui.Subcommand(
|
|
'embedart', help='embed image files into file metadata'
|
|
)
|
|
embed_cmd.parser.add_option(
|
|
'-f', '--file', metavar='PATH', help='the image file to embed'
|
|
)
|
|
maxwidth = self.config['maxwidth'].get(int)
|
|
compare_threshold = self.config['compare_threshold'].get(int)
|
|
ifempty = self.config['ifempty'].get(bool)
|
|
|
|
def embed_func(lib, opts, args):
|
|
if opts.file:
|
|
imagepath = normpath(opts.file)
|
|
if not os.path.isfile(syspath(imagepath)):
|
|
raise ui.UserError(u'image file {0} not found'.format(
|
|
displayable_path(imagepath)
|
|
))
|
|
for item in lib.items(decargs(args)):
|
|
self.embed_item(item, imagepath, maxwidth, None,
|
|
compare_threshold, ifempty)
|
|
else:
|
|
for album in lib.albums(decargs(args)):
|
|
self.embed_album(album, maxwidth)
|
|
|
|
embed_cmd.func = embed_func
|
|
|
|
# Extract command.
|
|
extract_cmd = ui.Subcommand('extractart',
|
|
help='extract an image from file metadata')
|
|
extract_cmd.parser.add_option('-o', dest='outpath',
|
|
help='image output file')
|
|
extract_cmd.parser.add_option('-a', dest='albums', action='store_true',
|
|
help='extract the art of all matching '
|
|
'albums')
|
|
|
|
def extract_func(lib, opts, args):
|
|
outpath = opts.outpath or config['art_filename'].get()
|
|
if opts.albums:
|
|
if opts.outpath and '/' in opts.outpath.replace('\\', '/'):
|
|
self._log.error(u"When using -a, only specify a filename "
|
|
u"instead of a full path for -o")
|
|
return
|
|
for album in lib.albums(decargs(args)):
|
|
artpath = normpath(os.path.join(album.path, outpath))
|
|
self.extract_first(artpath, album.items())
|
|
else:
|
|
outpath = normpath(outpath)
|
|
self.extract_first(outpath, lib.items(decargs(args)))
|
|
extract_cmd.func = extract_func
|
|
|
|
# Clear command.
|
|
clear_cmd = ui.Subcommand('clearart',
|
|
help='remove images from file metadata')
|
|
|
|
def clear_func(lib, opts, args):
|
|
self.clear(lib, decargs(args))
|
|
clear_cmd.func = clear_func
|
|
|
|
return [embed_cmd, extract_cmd, clear_cmd]
|
|
|
|
def album_imported(self, lib, album):
|
|
"""Automatically embed art into imported albums.
|
|
"""
|
|
if album.artpath and self.config['auto']:
|
|
max_width = self.config['maxwidth'].get(int)
|
|
self.embed_album(album, max_width, True)
|
|
|
|
def embed_item(self, item, imagepath, maxwidth=None, itempath=None,
|
|
compare_threshold=0, ifempty=False, as_album=False):
|
|
"""Embed an image into the item's media file.
|
|
"""
|
|
if compare_threshold:
|
|
if not self.check_art_similarity(item, imagepath,
|
|
compare_threshold):
|
|
self._log.info(u'Image not similar; skipping.')
|
|
return
|
|
if ifempty and self.get_art(item):
|
|
self._log.info(u'media file already contained art')
|
|
return
|
|
if maxwidth and not as_album:
|
|
imagepath = self.resize_image(imagepath, maxwidth)
|
|
|
|
try:
|
|
self._log.debug(u'embedding {0}', displayable_path(imagepath))
|
|
item['images'] = [self._mediafile_image(imagepath, maxwidth)]
|
|
except IOError as exc:
|
|
self._log.warning(u'could not read image file: {0}', exc)
|
|
else:
|
|
# We don't want to store the image in the database.
|
|
item.try_write(itempath)
|
|
del item['images']
|
|
|
|
def embed_album(self, album, maxwidth=None, quiet=False):
|
|
"""Embed album art into all of the album's items.
|
|
"""
|
|
imagepath = album.artpath
|
|
if not imagepath:
|
|
self._log.info(u'No album art present for {0}', album)
|
|
return
|
|
if not os.path.isfile(syspath(imagepath)):
|
|
self._log.info(u'Album art not found at {0} for {1}',
|
|
displayable_path(imagepath), album)
|
|
return
|
|
if maxwidth:
|
|
imagepath = self.resize_image(imagepath, maxwidth)
|
|
|
|
self._log.info(u'Embedding album art into {0}', album)
|
|
|
|
for item in album.items():
|
|
thresh = self.config['compare_threshold'].get(int)
|
|
ifempty = self.config['ifempty'].get(bool)
|
|
self.embed_item(item, imagepath, maxwidth, None,
|
|
thresh, ifempty, as_album=True)
|
|
|
|
def resize_image(self, imagepath, maxwidth):
|
|
"""Returns path to an image resized to maxwidth.
|
|
"""
|
|
self._log.debug(u'Resizing album art to {0} pixels wide', maxwidth)
|
|
imagepath = ArtResizer.shared.resize(maxwidth, syspath(imagepath))
|
|
return imagepath
|
|
|
|
def check_art_similarity(self, item, imagepath, compare_threshold):
|
|
"""A boolean indicating if an image is similar to embedded item art.
|
|
"""
|
|
with NamedTemporaryFile(delete=True) as f:
|
|
art = self.extract(f.name, item)
|
|
|
|
if art:
|
|
is_windows = platform.system() == "Windows"
|
|
|
|
# Converting images to grayscale tends to minimize the weight
|
|
# of colors in the diff score.
|
|
convert_proc = subprocess.Popen(
|
|
['convert', syspath(imagepath), syspath(art),
|
|
'-colorspace', 'gray', 'MIFF:-'],
|
|
stdout=subprocess.PIPE,
|
|
close_fds=not is_windows,
|
|
)
|
|
compare_proc = subprocess.Popen(
|
|
['compare', '-metric', 'PHASH', '-', 'null:'],
|
|
stdin=convert_proc.stdout,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
close_fds=not is_windows,
|
|
)
|
|
convert_proc.stdout.close()
|
|
|
|
stdout, stderr = compare_proc.communicate()
|
|
if compare_proc.returncode:
|
|
if compare_proc.returncode != 1:
|
|
self._log.debug(u'IM phashes compare failed for {0}, '
|
|
u'{1}', displayable_path(imagepath),
|
|
displayable_path(art))
|
|
return
|
|
out_str = stderr
|
|
else:
|
|
out_str = stdout
|
|
|
|
try:
|
|
phash_diff = float(out_str)
|
|
except ValueError:
|
|
self._log.debug(u'IM output is not a number: {0!r}',
|
|
out_str)
|
|
return
|
|
|
|
self._log.debug(u'compare PHASH score is {0}', phash_diff)
|
|
return phash_diff <= compare_threshold
|
|
|
|
return True
|
|
|
|
def _mediafile_image(self, image_path, maxwidth=None):
|
|
"""Return a `mediafile.Image` object for the path.
|
|
"""
|
|
|
|
with open(syspath(image_path), 'rb') as f:
|
|
data = f.read()
|
|
return mediafile.Image(data, type=mediafile.ImageType.front)
|
|
|
|
def get_art(self, item):
|
|
# Extract the art.
|
|
try:
|
|
mf = mediafile.MediaFile(syspath(item.path))
|
|
except mediafile.UnreadableFileError as exc:
|
|
self._log.warning(u'Could not extract art from {0}: {1}',
|
|
displayable_path(item.path), exc)
|
|
return
|
|
|
|
return mf.art
|
|
|
|
# 'extractart' command.
|
|
def extract(self, outpath, item):
|
|
art = self.get_art(item)
|
|
|
|
if not art:
|
|
self._log.info(u'No album art present in {0}, skipping.', item)
|
|
return
|
|
|
|
# Add an extension to the filename.
|
|
ext = imghdr.what(None, h=art)
|
|
if not ext:
|
|
self._log.warning(u'Unknown image type in {0}.',
|
|
displayable_path(item.path))
|
|
return
|
|
outpath += '.' + ext
|
|
|
|
self._log.info(u'Extracting album art from: {0} to: {1}',
|
|
item, displayable_path(outpath))
|
|
with open(syspath(outpath), 'wb') as f:
|
|
f.write(art)
|
|
return outpath
|
|
|
|
def extract_first(self, outpath, items):
|
|
for item in items:
|
|
if self.extract(outpath, item):
|
|
return outpath
|
|
|
|
# 'clearart' command.
|
|
def clear(self, lib, query):
|
|
id3v23 = config['id3v23'].get(bool)
|
|
|
|
items = lib.items(query)
|
|
self._log.info(u'Clearing album art from {0} items', len(items))
|
|
for item in items:
|
|
self._log.debug(u'Clearing art for {0}', item)
|
|
try:
|
|
mf = mediafile.MediaFile(syspath(item.path), id3v23)
|
|
except mediafile.UnreadableFileError as exc:
|
|
self._log.warning(u'Could not read file {0}: {1}',
|
|
displayable_path(item.path), exc)
|
|
else:
|
|
del mf.art
|
|
mf.save()
|