python查询neo4j的数据以字典的方式返回数据

在使用python操作neo4j的时候，如果查询的数据比较多，结构比较复杂的时候，返回的数据量会比较大，而且信息比较多，并且不唯一。所以写了该方法，用于查询比较复杂的数据。

 

def query_graph(query):
    graph = get_graph()
    data = graph.run(query).data()

    node_ids = []
    new_nodes = []
    new_links = []

    for a in data:
        for tk,tv in a.items():
            nodes = tv.nodes
            relations = tv.relationships
            for n in nodes:
                if n.identity in node_ids:
                    continue
                obj = {}
                obj["id"] = n.identity
                obj["label"] = []
                if n.labels is not None:
                    for la in n.labels:
                        obj["label"].append(la)
                for k,v in n.items():
                    obj[k] = v
                node_ids.append(n.identity)
                new_nodes.append(obj)
            for r in relations:
                if r.identity in node_ids:
                    continue
                li = {}
                li["id"] = r.identity
                if r.types() is not None:
                    li["label"] = []
                    for la in r.types():
                        li["label"].append(la)
                li["source"] = r.start_node.identity
                li["target"] = r.end_node.identity
                for k,v in r.items():
                    li[k] = v
                node_ids.append(r.identity)
                new_links.append(li)
    result = {}
    result["nodes"] = new_nodes
    result["links"] = new_links            
    return result