Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-16 05:34:47 +01:00
Code Issues Projects Releases Wiki Activity

Fix Beatport source weight

Browse source
This commit is contained in:
Johannes Baiter 2013-06-01 15:50:59 +02:00
parent af81b67c0c
commit bbd50c65cd
1 changed files with 18 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
beetsplug/beatport.py
View file

@ -27,10 +27,6 @@ from beets.plugins import BeetsPlugin
log = logging.getLogger('beets')
# Distance parameters.
BEATPORT_SOURCE_WEIGHT = config['beatport']['source_weight'].as_number()
SOURCE_WEIGHT = config['match']['weight']['source'].as_number()
class BeatportAPIError(Exception):
pass
@ -152,17 +148,33 @@ class BeatportTrack(BeatportObject):
class BeatportPlugin(BeetsPlugin):
def __init__(self):
super(BeatportPlugin, self).__init__()
self.config.add({
'source_weight': 0.5,
})
def album_distance(self, items, album_info, mapping):
"""Returns the beatport source weight and the maximum source weight
for albums.
"""
return BEATPORT_SOURCE_WEIGHT * SOURCE_WEIGHT, SOURCE_WEIGHT
if album_info.data_source == 'Beatport':
return self.config['source_weight'].as_number() * \
config['match']['weight']['source'].as_number(), \
config['match']['weight']['source'].as_number()
else:
return 0.0, 0.0
def track_distance(self, item, info):
"""Returns the beatport source weight and the maximum source weight
for individual tracks.
"""
return BEATPORT_SOURCE_WEIGHT * SOURCE_WEIGHT, SOURCE_WEIGHT
if info.data_source == 'Beatport':
return self.config['source_weight'].as_number() * \
config['match']['weight']['source'].as_number(), \
config['match']['weight']['source'].as_number()
else:
return 0.0, 0.0
def candidates(self, items, artist, release, va_likely):
"""Returns a list of AlbumInfo objects for beatport search results