python 从MusicBrainz获取/处理JSON数据

Posted 2021-05-09

import requests
import json


BASE_URL = "http://musicbrainz.org/ws/2/"
ARTIST_URL = BASE_URL + "artist/"
query_type = {  "simple": {},
                "atr": {"inc": "aliases+tags+ratings"},
                "aliases": {"inc": "aliases"},
                "releases": {"inc": "releases"}}

def query_site(url, params, uid="", fmt="json"):
    params['fmt'] = fmt
    r = requests.get(url + uid, params=params)

    if r.status_code == 200:
        return r.json()
    else:
        r.raise_for_status()


def query_by_name(url, params, name):
    params['query'] = ''.format('artist:{0}', name)
    return query_site(url, params)


def pretty_print(data, indent=4):
    if type(data) == dict:
        print(json.dumps(data, indent=indent, sort_keys=True))
    else:
        print(data)


def main():
    results = query_by_name(ARTIST_URL, query_type['simple'], 'Nirvana')
    pretty_print(results)


if __name__ == '__main__':
    main()