diff --git a/beetsplug/discogs.py b/beetsplug/discogs.py index 951c5d593..8d7e59ce7 100644 --- a/beetsplug/discogs.py +++ b/beetsplug/discogs.py @@ -28,10 +28,6 @@ import time log = logging.getLogger('beets') -# Distance parameters. -DISCOGS_SOURCE_WEIGHT = config['discogs']['source_weight'].as_number() -SOURCE_WEIGHT = config['match']['weight']['source'].as_number() - # Silence spurious INFO log lines generated by urllib3. urllib3_logger = logging.getLogger('requests.packages.urllib3') urllib3_logger.setLevel(logging.CRITICAL) @@ -44,7 +40,12 @@ class DiscogsPlugin(BeetsPlugin): def album_distance(self, items, album_info, mapping): """Returns the discogs source weight and the maximum source weight. """ - return DISCOGS_SOURCE_WEIGHT * SOURCE_WEIGHT, SOURCE_WEIGHT + if album_info.data_source == 'Discogs': + return config['discogs']['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, album, va_likely): """Returns a list of AlbumInfo objects for discogs search results