sima/lib/simaecho.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Copyright (c) 2014 Jack Kaliko <kaliko@azylum.org>
   4 #
   5 #   This program is free software: you can redistribute it and/or modify
   6 #   it under the terms of the GNU General Public License as published by
   7 #   the Free Software Foundation, either version 3 of the License, or
   8 #   (at your option) any later version.
   9 #
  10 #   This program is distributed in the hope that it will be useful,
  11 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #   GNU General Public License for more details.
  14 #
  15 #   You should have received a copy of the GNU General Public License
  16 #   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 #
  18 #
  19
  20 """
  21 Consume EchoNest web service
  22 """
  23
  24 __version__ = '0.0.5'
  25 __author__ = 'Jack Kaliko'
  26
  27
  28
  29 from sima import ECH
  30 from sima.lib.meta import Artist
  31 from sima.lib.track import Track
  32 from sima.lib.http import HttpClient
  33 from sima.utils.utils import WSError, WSNotFound
  34 from sima.utils.utils import getws
  35 if len(ECH.get('apikey')) == 23:  # simple hack allowing imp.reload
  36     getws(ECH)
  37
  38
  39 def get_mbid(obj, foreign='foreign_ids'):
  40     if foreign in obj:
  41         for frgnid in obj.get(foreign):
  42             if frgnid.get('catalog') == 'musicbrainz':
  43                 return frgnid.get('foreign_id').split(':')[2]
  44     return None
  45
  46
  47 class SimaEch:
  48     """EchoNest http client
  49     """
  50     root_url = 'http://{host}/api/{version}'.format(**ECH)
  51     name = 'EchoNest'
  52     cache = False
  53     stats = {'etag':0,
  54              'ccontrol':0,
  55              'minrl':120,
  56              'total':0}
  57
  58     def __init__(self):
  59         self.http = HttpClient(cache=self.cache, stats=self.stats)
  60
  61     def _controls_answer(self, ans):
  62         """Controls answer.
  63         """
  64         status = ans.get('response').get('status')
  65         code = status.get('code')
  66         if code is 0:
  67             return True
  68         if code is 5:
  69             raise WSNotFound('Artist not found')
  70         raise WSError(status.get('message'))
  71
  72     def _forge_payload(self, artist, top=False):
  73         """Build payload
  74         """
  75         payload = {'api_key': ECH.get('apikey')}
  76         if not isinstance(artist, Artist):
  77             raise TypeError('"{0!r}" not an Artist object'.format(artist))
  78         if artist.mbid:
  79             payload.update(id='musicbrainz:artist:{0}'.format(artist.mbid))
  80         else:
  81             payload.update(name=artist.name)
  82         payload.update(bucket='id:musicbrainz')
  83         payload.update(results=100)
  84         if top:
  85             if artist.mbid:
  86                 aid = payload.pop('id')
  87                 payload.update(artist_id=aid)
  88             else:
  89                 name = payload.pop('name')
  90                 payload.update(artist=name)
  91             payload.update(results=100)
  92             payload.update(sort='song_hotttnesss-desc')
  93         # > hashing the URL into a cache key
  94         # return a sorted list of 2-tuple to have consistent cache
  95         return sorted(payload.items(), key=lambda param: param[0])
  96
  97     def get_similar(self, artist):
  98         """Fetch similar artists
  99
 100         param: artist Artist: Artist object to get similarities from
 101         """
 102         payload = self._forge_payload(artist)
 103         # Construct URL
 104         ressource = '{0}/artist/similar'.format(SimaEch.root_url)
 105         ans = self.http(ressource, payload)
 106         self._controls_answer(ans.json())  # pylint: disable=no-member
 107         for art in ans.json().get('response').get('artists'):  # pylint: disable=no-member
 108             mbid = get_mbid(art)
 109             yield Artist(mbid=mbid, name=art.get('name'))
 110
 111     def get_toptrack(self, artist):
 112         """Fetch artist top tracks
 113
 114         param: artist Artist: Artist object to get top tracks from
 115         """
 116         payload = self._forge_payload(artist, top=True)
 117         # Construct URL
 118         ressource = '{0}/song/search'.format(SimaEch.root_url)
 119         ans = self.http(ressource, payload)
 120         self._controls_answer(ans.json())  # pylint: disable=no-member
 121         titles = list()
 122         art = {'artist': artist.name,
 123                'musicbrainz_artistid': artist.mbid,}
 124         for song in ans.json().get('response').get('songs'):  # pylint: disable=no-member
 125             title = song.get('title')
 126             if not art.get('musicbrainz_artistid'):
 127                 art['musicbrainz_artistid'] = get_mbid(song, 'artist_foreign_ids')
 128             if title not in titles:
 129                 titles.append(title)
 130                 yield Track(title=title, **art)
 131
 132
 133 # VIM MODLINE
 134 # vim: ai ts=4 sw=4 sts=4 expandtab