discogs.py

from aiohttp import ClientError, ClientSession
import json
import logging
import re
import time
import urllib.parse

from datamodel import UnifiedData
import httpclient
import settings



DISCOGS_CONSUMER_KEY = settings.DiscogsConfig.DISCOGS_CONSUMER_KEY
DISCOGS_CONSUMER_SECRET = settings.DiscogsConfig.DISCOGS_CONSUMER_SECRET
DISCOGS_HEADERS = {
    "Authorization": "Discogs key={key},secret={secret}".format(
        key=DISCOGS_CONSUMER_KEY,
        secret=DISCOGS_CONSUMER_SECRET)
}
DISCOGS_ROOT_URL = 'https://api.discogs.com'
DISCOGS_OMITTED_PHRASES = [
    'EP', '-',
    'Original Mix'
]
_LOGGER = logging.getLogger(__name__)



class DiscogsData(UnifiedData):
    def __init__(self, image_url, label):
        self.image_url = image_url
        self.label = label


class Discogs(object):
    def __init__(self):
        self.enabled = settings.GlobalConfig.ENABLE_DISCOGS
        if self.enabled:
            self.last_update = 0

    async def enrich(self, data):
        self.last_update = time.time()
        if data and data.nowplaying_audio():
            resource_url = self._search(data.trackname, data.artist, data.album)
            if resource_url:
                self._get_data_from_resource(resource_url, data)

    def _search(self, trackname, artist, album):
        artist = '"{artist}"'.format(artist=artist)
        queryparts = ' '.join([trackname, artist, album])
        queryparts = queryparts.replace('(', '').replace(')', '')
        for p in DISCOGS_OMITTED_PHRASES:
            queryparts = queryparts.replace(p, '')
        queryparts = queryparts.split()
        query = ' '.join(queryparts)
        query = urllib.parse.quote_plus(query)
        url = '{root}/database/search?q={query}'.format(
            root=DISCOGS_ROOT_URL,
            query=query)

        requestobj = urllib.request.Request(url, headers=DISCOGS_HEADERS)
        obj = httpclient.get_json_from_url(requestobj)
        try:
            return obj['results'][0]['resource_url']
        except (KeyError, IndexError) as err:
            _LOGGER.error('Discogs search failed to get meaningful results for URL [{url}]: {err}'.format(
                url=url,
                err=err))
        return None

    def _get_data_from_resource(self, url, data):
        requestobj = urllib.request.Request(url, headers=DISCOGS_HEADERS)
        obj = httpclient.get_json_from_url(requestobj)
        
        try:
            image_url = obj['images'][0]['resource_url'] if 'images' in obj else None
            label = obj['labels'][0]['name'] if 'labels' in obj else None
            data.image_url = image_url
            data.label = label
        except (KeyError, IndexError) as err:
            _LOGGER.error('Discogs data fetch failed to get meaningful results for URL [{url}]: {err}'.format(
                url=url,
                err=err))