make_graph.py

def make_graph(query, depth, children):
    return _make_graph(query, depth, children, dict())

def _make_graph(query, depth, children, d):
    for i in range(children):
        rand_artist = get_random_rec(query)
        if rand_artist == None:
            break
        if depth == 0:
            d[query] = {rand_artist};
            return d; 
        if query in d.keys():
            d[query].add(rand_artist)
        else:
            d[query] = {rand_artist}
        d = _make_graph(rand_artist, depth-1, d)
    return d