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