65 lines
2.7 KiB
Python
65 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
import difflib
|
|
|
|
def discogs_albumfind(data, artist, album):
|
|
albums = []
|
|
masters = []
|
|
# sort results by lowest release id (first version of a release)
|
|
releases = sorted(data.get('results',[]), key=lambda k: k['id'])
|
|
for item in releases:
|
|
masterid = item['master_id']
|
|
# we are not interested in multiple versions that belong to the same master release
|
|
if masterid not in masters:
|
|
masters.append(masterid)
|
|
albumdata = {}
|
|
albumdata['artist'] = item['title'].split(' - ',1)[0]
|
|
albumdata['album'] = item['title'].split(' - ',1)[1]
|
|
albumdata['artist_description'] = item['title'].split(' - ',1)[0]
|
|
albumdata['year'] = str(item.get('year', ''))
|
|
albumdata['label'] = item['label'][0]
|
|
albumdata['thumb'] = item['thumb']
|
|
albumdata['dcalbumid'] = item['id']
|
|
# discogs does not provide relevance, use our own
|
|
artistmatch = difflib.SequenceMatcher(None, artist.lower(), albumdata['artist'].lower()).ratio()
|
|
albummatch = difflib.SequenceMatcher(None, album.lower(), albumdata['album'].lower()).ratio()
|
|
if artistmatch > 0.90 and albummatch > 0.90:
|
|
score = round(((artistmatch + albummatch) / 2), 2)
|
|
albumdata['relevance'] = str(score)
|
|
albums.append(albumdata)
|
|
return albums
|
|
|
|
def discogs_albummain(data):
|
|
if data:
|
|
if 'main_release_url' in data:
|
|
url = data['main_release_url'].rsplit('/', 1)[1]
|
|
return url
|
|
|
|
def discogs_albumdetails(data):
|
|
albumdata = {}
|
|
albumdata['album'] = data['title']
|
|
if 'styles' in data:
|
|
albumdata['styles'] = ' / '.join(data['styles'])
|
|
albumdata['genres'] = ' / '.join(data['genres'])
|
|
albumdata['year'] = str(data['year'])
|
|
albumdata['label'] = data['labels'][0]['name']
|
|
artists = []
|
|
for artist in data['artists']:
|
|
artistdata = {}
|
|
artistdata['artist'] = artist['name']
|
|
artists.append(artistdata)
|
|
albumdata['artist'] = artists
|
|
albumdata['artist_description'] = data['artists_sort']
|
|
albumdata['rating'] = str(int((float(data['community']['rating']['average']) * 2) + 0.5))
|
|
albumdata['votes'] = str(data['community']['rating']['count'])
|
|
if 'images' in data:
|
|
thumbs = []
|
|
for thumb in data['images']:
|
|
thumbdata = {}
|
|
thumbdata['image'] = thumb['uri']
|
|
thumbdata['preview'] = thumb['uri150']
|
|
# not accurate: discogs can provide any art type, there is no indication if it is an album front cover (thumb)
|
|
thumbdata['aspect'] = 'thumb'
|
|
thumbs.append(thumbdata)
|
|
albumdata['thumb'] = thumbs
|
|
return albumdata
|