mirror of
https://github.com/beetbox/beets.git
synced 2025-12-15 04:55:10 +01:00
fba874932c
This required the introduction of a track_distance method on plugins. We'll also need to add an album_distance method as well as a mechanism for extending the search routine (so we can search for albums in MusicBrainz even when they have no tags). This commit also adds the '-v' flag for printing debug logs (something we should do more of).
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
# This file is part of beets.
|
|
# Copyright 2010, 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.
|
|
|
|
"""Adds Last.fm acoustic fingerprinting support to the autotagger.
|
|
Requires the pylastfp library.
|
|
"""
|
|
|
|
from __future__ import with_statement
|
|
from beets.plugins import BeetsPlugin
|
|
from beets import autotag
|
|
import lastfp
|
|
import logging
|
|
|
|
log = logging.getLogger('beets')
|
|
API_KEY = '2dc3914abf35f0d9c92d97d8f8e42b43'
|
|
|
|
_match_cache = {}
|
|
def match(path, metadata=None):
|
|
"""Gets the metadata from Last.fm for the indicated track. Returns
|
|
a dictionary with these keys: rank, artist, artist_mbid, title,
|
|
track_mbid. May return None if fingerprinting or lookup fails.
|
|
Caches the result, so multiple calls may be made efficiently.
|
|
"""
|
|
if path in _match_cache:
|
|
return _match_cache[path]
|
|
|
|
# Actually perform fingerprinting and lookup.
|
|
try:
|
|
xml = lastfp.gst_match(API_KEY, path, metadata)
|
|
matches = lastfp.parse_metadata(xml)
|
|
except lastfp.FingerprintError:
|
|
# Fail silently and cache the failure.
|
|
matches = None
|
|
match = matches[0] if matches else None
|
|
|
|
_match_cache[path] = match
|
|
return match
|
|
|
|
class LastIdPlugin(BeetsPlugin):
|
|
def track_distance(self, item, info):
|
|
last_data = match(item.path)
|
|
if not last_data:
|
|
# Match failed.
|
|
return 0.0, 0.0
|
|
|
|
dist = autotag._ie_dist(last_data['title'],
|
|
info['title']) \
|
|
* autotag.TRACK_TITLE_WEIGHT
|
|
dist_max = autotag.TRACK_TITLE_WEIGHT
|
|
|
|
log.debug('Last data: %s; distance: %f' %
|
|
(str(last_data), dist/dist_max))
|
|
|
|
return dist, dist_max
|