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