searxng/searx/engines/spotify.py

77 lines
2.1 KiB
Python
Raw Normal View History

"""
Spotify (Music)
@website https://spotify.com
@provide-api yes (https://developer.spotify.com/web-api/search-item/)
@using-api yes
@results JSON
@stable yes
@parse url, title, content, embedded
"""
2015-02-20 17:47:56 +00:00
from json import loads
2016-11-30 17:43:03 +00:00
from searx.url_utils import urlencode
2019-07-31 18:44:41 +00:00
import requests
import base64
2015-02-20 17:47:56 +00:00
# engine dependent config
categories = ['music']
paging = True
2019-07-31 18:44:41 +00:00
api_client_id = None
api_client_secret = None
2015-02-20 17:47:56 +00:00
# search-url
url = 'https://api.spotify.com/'
search_url = url + 'v1/search?{query}&type=track&offset={offset}'
embedded_url = '<iframe data-src="https://embed.spotify.com/?uri=spotify:track:{audioid}"\
width="300" height="80" frameborder="0" allowtransparency="true"></iframe>'
# do search-request
def request(query, params):
offset = (params['pageno'] - 1) * 20
2016-11-30 17:43:03 +00:00
params['url'] = search_url.format(query=urlencode({'q': query}), offset=offset)
2015-02-20 17:47:56 +00:00
2019-07-31 18:44:41 +00:00
r = requests.post(
'https://accounts.spotify.com/api/token',
data={'grant_type': 'client_credentials'},
headers={'Authorization': 'Basic ' + str(base64.b64encode(
(api_client_id + ":" + api_client_secret).encode('utf-8')
), 'utf-8')}
)
j = loads(r.text)
params['headers'] = {'Authorization': 'Bearer ' + j['access_token']}
2015-02-20 17:47:56 +00:00
return params
# get response from search-request
def response(resp):
results = []
search_res = loads(resp.text)
# parse results
for result in search_res.get('tracks', {}).get('items', {}):
if result['type'] == 'track':
title = result['name']
url = result['external_urls']['spotify']
content = u'{} - {} - {}'.format(
result['artists'][0]['name'],
result['album']['name'],
result['name'])
2015-02-20 17:47:56 +00:00
embedded = embedded_url.format(audioid=result['id'])
# append result
results.append({'url': url,
'title': title,
'embedded': embedded,
'content': content})
# return results
return results