36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import json
|
|
import logging
|
|
|
|
import requests
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
VALID_METRICS = ['mfccs', 'mfccsw', 'gfccs', 'gfccsw', 'key', 'bpm', 'onsetrate', 'moods',
|
|
'instruments', 'dortmund', 'rosamerica', 'tzanetakis']
|
|
|
|
|
|
class EndpointError(Exception):
|
|
pass
|
|
|
|
|
|
def get_similar_tracks_mbids_from_mbid(mbid, annoy_similarity):
|
|
|
|
if annoy_similarity not in VALID_METRICS:
|
|
raise AttributeError("Metric %s is not valid. Must be one of : " + print(VALID_METRICS))
|
|
|
|
headers = {'Content-Type': 'application/json'}
|
|
endpoint = "acousticbrainz.org/api/v1/similarity"
|
|
similar_tracks_mbids = []
|
|
similar_tracks = requests.get(
|
|
'https://{endpoint}/{annoy_similarity}/?recording_ids={mbid}&remove_dups&n_neighbours=1000'
|
|
.format(endpoint=endpoint, annoy_similarity=annoy_similarity, mbid=mbid), headers=headers
|
|
)
|
|
if similar_tracks.status_code != 200:
|
|
logger.warning("Error while querying {endpoint!r} : {similar_tracks.content!r}")
|
|
raise EndpointError
|
|
|
|
j = json.loads(similar_tracks.content)
|
|
for tracks in j['{mbid}'.format(mbid=mbid)]['0']:
|
|
similar_tracks_mbids.append(tracks['recording_mbid'])
|
|
return similar_tracks_mbids
|