Python get_graph_db_connection示例

示例#1

文件： graph_extractor.py 项目： rcbull/brasil.io

def _extract_network(query, path_key="p"):
    graph = nx.DiGraph()
    output = get_graph_db_connection().run(query)
    while output.forward():
        path = output.current()[path_key]
        nodes = path.nodes()
        rels = path.relationships()

        for node in nodes:
            labels = list(node.labels())
            graph.add_node(node.__name__, tipo=labels[0], labels=labels, **node.properties)

        for rel in rels:
            graph.add_edge(
                rel.start_node().__name__, rel.end_node().__name__, tipo_relacao=rel.type(), **rel.properties
            )

    return graph

示例#2

文件： build_company_groups_network.py 项目： snifferchess/brasil.io

 def __init__(self, *args, **kwargs):
     super(BaseCommand, self).__init__(*args, **kwargs)
     self.graph_db = get_graph_db_connection()

示例#3

文件： graph_extractor.py 项目： snifferchess/brasil.io

def selector():
    return NodeSelector(get_graph_db_connection())

示例#4

 def __init__(self, *args, **kwargs):
     super(BaseCommand, self).__init__(*args, **kwargs)
     self.open_transaction = None
     self.batch_size = 1000
     self.graph_db = get_graph_db_connection()
     self.company_names = {}