tidalf/plugin.audio.qobuz

# -*- coding: UTF-8 -*-
'''
    qobuz.node.recommendation
    ~~~~~~~~~~~~~~~~~~~~~~~~~

    :part_of: kodi-qobuz
    :copyright: (c) 2012-2018 by Joachim Basmaison, Cyril Leclerc
    :license: GPLv3, see LICENSE for more details.
'''
from qobuz.api import api
from qobuz.debug import getLogger
from qobuz.gui.util import lang, getImage
from qobuz.node import getNode, Flag
from qobuz.node.inode import INode

logger = getLogger(__name__)

RECOS_TYPE_IDS = {
    1: 'new-releases',
    2: 'press-awards',
    3: 'best-sellers',
    4: 'editor-picks',
    5: 'most-featured',
    6: 'most-streamed'
}

RECOS_TYPES = {
    1: lang(30086),
    2: lang(30085),
    3: lang(30087),
    4: lang(30088),
    5: lang(30103),
    6: lang(30192)
}

RECOS_GENRES = {
    2: lang(30095),
    10: lang(30097),
    6: lang(30092),
    59: lang(30100),
    73: lang(30105),
    80: lang(30091),
    64: lang(30106),
    91: lang(30096),
    94: lang(30094),
    112: lang(30089),
    127: lang(30104),
    123: lang(30107),
    'null': 'All',
}


class Node_recommendation(INode):
    def __init__(self, parent=None, parameters=None, data=None):
        parameters = {} if parameters is None else parameters
        super(Node_recommendation, self).__init__(
            parent=parent, parameters=parameters, data=data)
        self.nt = Flag.RECOMMENDATION
        self.genre_id = self.get_parameter('genre-id', default=None)
        self.genre_type = self.get_parameter('genre-type', default=None)
        self.set_label(lang(30084))
        self.image = getImage('album')
        self.content_type = 'albums'

    def make_url(self, **ka):
        if self.genre_type is not None:
            ka['genre-type'] = self.genre_type
        if self.genre_id is not None:
            ka['genre-id'] = self.genre_id
        return super(Node_recommendation, self).make_url(**ka)

    def myid(self):
        if self.genre_id is None or self.genre_type is None:
            return None
        return str(self.genre_type) + '-' + str(self.genre_id)

    def fetch(self, options=None):
        if self.genre_type is None or self.genre_id is None:
            return {}
        return api.get('/album/getFeatured',
                       type=RECOS_TYPE_IDS[int(self.genre_type)],
                       genre_id=self.genre_id,
                       limit=self.limit,
                       offset=self.offset)

    def __populate_type(self, options=None):
        '''Populate type, we don't have genre_type nor genre_id
        '''
        for genre_type_id in RECOS_TYPE_IDS:
            node = getNode(Flag.RECOMMENDATION, {'genre-type': genre_type_id})
            node.label2 = self.label
            node.label = RECOS_TYPES[genre_type_id]
            self.add_child(node)
        return True

    def __populate_genre(self, options=None):
        '''Populate genre, we have genre_type but no genre_id
        '''
        for genre_id in RECOS_GENRES:
            node = getNode(
                Flag.RECOMMENDATION,
                parameters={
                    'genre-type': self.genre_type,
                    'genre-id': genre_id
                })
            node.label = RECOS_GENRES[genre_id]
            node.label2 = RECOS_TYPES[int(self.genre_type)]
            self.add_child(node)
        return True

    def __populate_type_genre(self, options=None):
        '''Populate album selected by genre_type and genre_id
        '''
        self.content_type = 'albums'
        if self.data is None:
            return False
        if 'albums' not in self.data:
            logger.warn('Recommendation data doesn\'t contain <albums> key')
            return False
        if self.data['albums'] is None or 'items' not in self.data['albums']:
            logger.warn('Recommendation data[\'albums\'] doesn\'t contain items')
            return False
        for product in self.data['albums']['items']:
            self.add_child(getNode(Flag.ALBUM, data=product))
        return True

    def populate(self, options=None):
        '''We are populating our node based on genre_type and genre_id
        '''
        if not self.genre_type:
            return self.__populate_type(options)
        elif not self.genre_id:
            return self.__populate_genre(options)
        return self.__populate_type_genre(options)