使用Python将保存的“ graphdata”重新加载到重载实例时遇到问题

时间:2018-12-31 11:27:43

标签: python redis redisgraph

我想使用redisgraph-py将保存的图形加载回Graph对象。如何使用存储为graphdata类型的redis-py加载数据?

我正在运行redisgraph docker映像,并且在键中看到该图没有问题。我是Redis的新手,正尝试仅使用r.get('random')来加载保存的图形,但是从顶部的答案here可以看出,我需要使用正确的方法来提取正确的值类型redis-py。

from io import BytesIO
from pprint import pprint
import numpy as np
import redis
from redisgraph import Node, Edge, Graph

def _array2bytes(arr):
    with BytesIO() as b:
        np.save(b, arr)
        return b.getvalue()

def _bytes2array(byts):
    return np.load(BytesIO(byts))

def _add_nodes(r, graph, n_nodes):
    nodes = []
    for k in range(n_nodes):
        n = Node(label="node")
        graph.add_node(n)
        nodes.append(n)
        _id = n.alias
        feat = np.random.rand(1,100,7,7)
        feat_bytes = _array2bytes(feat)
        r.set(_id, feat_bytes)
    return nodes

def _add_edges(nodes, graph, edge_prob):
    edges = []
    for k, node0 in enumerate(nodes):
        for kk, node1 in enumerate(nodes):
            if np.random.rand() < edge_prob:
                edge = Edge(node0, "adjacent_to", node1)
                graph.add_edge(edge)
                edges.append(edge)
    return edges

def _create_random_graph(r, graphname="random", n_nodes=1000, edge_prob=0.1):
    redis_graph = Graph(graphname, r)
    nodes = _add_nodes(r, redis_graph, n_nodes)
    edges = _add_edges(nodes, redis_graph, edge_prob)
    return redis_graph

def _save_graph():
    r = redis.Redis(host='localhost', port=6379)
    g = _create_random_graph(r)
    g.commit()
    return r

def _load_graph(r):
    # The graph is saved under the key 'random'
    print(r.type('random'))

def _main():
    r = _save_graph()
    _load_graph(r)

if __name__ == "__main__":
    _main()

我看到的是graphdata,但我不知道如何使用redis-py或redisgraph-py加载此类数据。在文档中也找不到任何内容。

1 个答案:

答案 0 :(得分:1)

当前redisgraph-py无法通过包含图形的Redis键重新创建Python图形对象。

要实现此目的,模块(redisgraph-py)必须通过发出查询来检索所有节点:

MATCH (n) RETURN n

并分别创建一个redisgraph-py Node对象,此外,它还必须发出第二个查询以检索所有关系

MATCH (n)-[r]->(m) RETURN n,r,m

并从每个对象创建一个redisgraph-py边缘对象。

我认为,如果我们采用这种方式,则可以实现类似于NetworkX的接口/框架。