1 # -*- coding: utf-8 -*-
3 # Copyright (c) 2014 Jack Kaliko <kaliko@azylum.org>
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.
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.
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/>.
21 Consume EchoNest web service
25 __author__ = 'Jack Kaliko'
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
40 """EchoNest http client
42 root_url = 'http://{host}/api/{version}'.format(**ECH)
52 self.http = HttpClient(cache=self.cache, stats=self.stats)
54 def _controls_answer(self, ans):
57 status = ans.get('response').get('status')
58 code = status.get('code')
62 raise WSNotFound('Artist not found')
63 raise WSError(status.get('message'))
65 def _forge_payload(self, artist, top=False):
68 payload = {'api_key': ECH.get('apikey')}
69 if not isinstance(artist, Artist):
70 raise TypeError('"{0!r}" not an Artist object'.format(artist))
73 id='musicbrainz:artist:{0}'.format(artist.mbid))
75 payload.update(name=artist.name)
76 payload.update(bucket='id:musicbrainz')
77 payload.update(results=100)
80 aid = payload.pop('id')
81 payload.update(artist_id=aid)
83 name = payload.pop('name')
84 payload.update(artist=name)
85 payload.update(results=100)
86 payload.update(sort='song_hotttnesss-desc')
87 # > hashing the URL into a cache key
88 # return a sorted list of 2-tuple to have consistent cache
89 return sorted(payload.items(), key=lambda param: param[0])
91 def get_similar(self, artist=None):
92 """Fetch similar artists
94 payload = self._forge_payload(artist)
96 ressource = '{0}/artist/similar'.format(SimaEch.root_url)
97 ans = self.http(ressource, payload)
98 self._controls_answer(ans.json())
99 for art in ans.json().get('response').get('artists'):
101 if 'foreign_ids' in art:
102 for frgnid in art.get('foreign_ids'):
103 if frgnid.get('catalog') == 'musicbrainz':
104 mbid = frgnid.get('foreign_id'
105 ).lstrip('musicbrainz:artist:')
106 yield Artist(mbid=mbid, name=art.get('name'))
108 def get_toptrack(self, artist=None):
109 """Fetch artist top tracks
111 payload = self._forge_payload(artist, top=True)
113 ressource = '{0}/song/search'.format(SimaEch.root_url)
114 ans = self.http(ressource, payload)
115 self._controls_answer(ans.json())
118 'artist': artist.name,
119 'musicbrainz_artistid': artist.mbid,
121 for song in ans.json().get('response').get('songs'):
122 title = song.get('title')
123 if title not in titles:
125 yield Track(title=title, **art)
129 # vim: ai ts=4 sw=4 sts=4 expandtab