From d6f98352a11039c1eaba771ae0725ce48171a286 Mon Sep 17 00:00:00 2001 From: Pieter Cailliau Date: Sat, 15 Dec 2018 11:15:49 +0200 Subject: [PATCH] Adding merge functionality --- redisgraph/client.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/redisgraph/client.py b/redisgraph/client.py index 2bdb7d0..d299967 100644 --- a/redisgraph/client.py +++ b/redisgraph/client.py @@ -117,12 +117,15 @@ def add_edge(self, edge): assert self.nodes[edge.src_node.alias] is not None and self.nodes[edge.dest_node.alias] is not None self.edges.append(edge) - def commit(self): + def commit(self, merge=False): """ Create entire graph. """ + if merge: + query = 'MERGE ' + else: + query = 'CREATE ' - query = 'CREATE ' for _, node in self.nodes.items(): query += str(node) + ',' @@ -134,6 +137,12 @@ def commit(self): query = query[:-1] return self.query(query) + + def merge(self): + """ + Merge entire graph. + """ + return self.commit(True) def query(self, q): """