summaryrefslogtreecommitdiffstats
path: root/addons/metadata.generic.albums/lib/discogs.py
blob: 448c3547645f57e2d4fce50d36eda3e099b2394b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# -*- 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