Проблема загрузки сохраненных `graphdata` обратно в экземпляр редисграфа с Python
Я хочу использовать redisgraph-py для загрузки сохраненного графика обратно в объект Graph. Как загрузить данные с помощью Redis-Py, которые хранятся в виде типа graphdata
?
Я запускаю образ докера redisgraph и без проблем вижу график среди клавиш. Я новичок в Redis и пытался просто использовать r.get('random')
загрузить сохраненный график, но в верхнем ответе я вижу, что мне нужно использовать правильный метод для извлечения правильного типа значения с помощью 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 ответ
В настоящее время redisgraph-py не может повторно создать объект графа Python из ключа Redis, содержащего граф.
Для реализации этого модуль (redisgraph-py) должен получить все узлы, выполнив запрос:
MATCH (n) RETURN n
И создайте для каждого объекта объект типа redisgraph-py, кроме того, он должен будет выполнить второй запрос, извлекая все отношения
MATCH (n)-[r]->(m) RETURN n,r,m
И создайте объект края redisgraph-py из каждого.
Я думаю, что если бы мы пошли по этому пути, имело бы смысл реализовать интерфейс / структуру, аналогичную NetworkX.