mirror of
https://github.com/beetbox/beets.git
synced 2025-12-09 01:54:08 +01:00
124 lines
3.8 KiB
Python
Executable file
124 lines
3.8 KiB
Python
Executable file
import logging
|
|
import imghdr
|
|
|
|
from beets.plugins import BeetsPlugin
|
|
from beets import mediafile
|
|
from beets import ui
|
|
from beets.util import syspath, normpath
|
|
|
|
log = logging.getLogger('beets')
|
|
|
|
def _embed(path, items):
|
|
"""Embed an image file, located at `path`, into each item.
|
|
"""
|
|
data = open(syspath(path), 'rb').read()
|
|
kindstr = imghdr.what(None, data)
|
|
if kindstr not in ('jpeg', 'png'):
|
|
log.error('A file of type %s is not allowed as coverart.' % kindstr)
|
|
return
|
|
|
|
# Add art to each file.
|
|
log.debug('Embedding album art.')
|
|
for item in items:
|
|
f = mediafile.MediaFile(syspath(item.path))
|
|
f.art = data
|
|
f.save()
|
|
|
|
options = {
|
|
'autoembed': True,
|
|
}
|
|
class EmbedCoverArtPlugin(BeetsPlugin):
|
|
"""Allows albumart to be embedded into the actual files."""
|
|
def configure(self, config):
|
|
options['autoembed'] = \
|
|
ui.config_val(config, 'embedart', 'autoembed', True, bool)
|
|
|
|
def commands(self):
|
|
# Embed command.
|
|
embed_cmd = ui.Subcommand('embedart',
|
|
help='embed an image file into file metadata')
|
|
def embed_func(lib, config, opts, args):
|
|
if not args:
|
|
raise ui.UserError('specify an image file')
|
|
imagepath = normpath(args.pop(0))
|
|
embed(lib, imagepath, ui.make_query(args))
|
|
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')
|
|
def extract_func(lib, config, opts, args):
|
|
outpath = normpath(opts.outpath or 'cover')
|
|
extract(lib, outpath, ui.make_query(args))
|
|
extract_cmd.func = extract_func
|
|
|
|
# Clear command.
|
|
clear_cmd = ui.Subcommand('clearart',
|
|
help='remove images from file metadata')
|
|
def clear_func(lib, config, opts, args):
|
|
clear(lib, ui.make_query(args))
|
|
clear_cmd.func = clear_func
|
|
|
|
return [embed_cmd, extract_cmd, clear_cmd]
|
|
|
|
# "embedart" command.
|
|
def embed(lib, imagepath, query):
|
|
albums = lib.albums(query)
|
|
for i_album in albums:
|
|
album = i_album
|
|
break
|
|
else:
|
|
log.error('No album matches query.')
|
|
return
|
|
|
|
log.info('Embedding album art into %s - %s.' % \
|
|
(album.albumartist, album.album))
|
|
_embed(imagepath, album.items())
|
|
|
|
# "extractart" command.
|
|
def extract(lib, outpath, query):
|
|
items = lib.items(query)
|
|
for i_item in items:
|
|
item = i_item
|
|
break
|
|
else:
|
|
log.error('No item matches query.')
|
|
return
|
|
|
|
# Extract the art.
|
|
mf = mediafile.MediaFile(syspath(item.path))
|
|
art = mf.art
|
|
if not art:
|
|
log.error('No album art present in %s - %s.' %
|
|
(item.artist, item.title))
|
|
return
|
|
|
|
# Add an extension to the filename.
|
|
ext = imghdr.what(None, h=art)
|
|
if not ext:
|
|
log.error('Unknown image type.')
|
|
return
|
|
outpath += '.' + ext
|
|
|
|
log.info('Extracting album art from: %s - %s\n'
|
|
'To: %s' % \
|
|
(item.artist, item.title, outpath))
|
|
with open(syspath(outpath), 'wb') as f:
|
|
f.write(art)
|
|
|
|
# "clearart" command.
|
|
def clear(lib, query):
|
|
log.info('Clearing album art from items:')
|
|
for item in lib.items(query):
|
|
log.info(u'%s - %s' % (item.artist, item.title))
|
|
mf = mediafile.MediaFile(syspath(item.path))
|
|
mf.art = None
|
|
mf.save()
|
|
|
|
# Automatically embed art into imported albums.
|
|
@EmbedCoverArtPlugin.listen('album_imported')
|
|
def album_imported(lib, album):
|
|
if album.artpath and options['autoembed']:
|
|
_embed(album.artpath, album.items())
|