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