add support for path type

Author: DvirDukhan

redisgraph/__init__.py

@@ -1,3 +1,4 @@
 from .node import Node
 from .edge import Edge
 from .graph import Graph
+from .path import Path

redisgraph/path.py

@@ -0,0 +1,27 @@
+
+class Path(object):
+
+    def __init__(self, nodes, relationships):
+        self.nodes = nodes
+        self.relationships = relationships
+
+    def nodes(self):
+        return self.nodes
+
+    def relationships(self):
+        return self.relationships
+
+    def get_node(self, index):
+        return self.nodes[index]
+
+    def get_relationship(self, index):
+        return self.relationships[index]
+
+    def first_node(self):
+        return self.nodes[0]
+
+    def last_node(self):
+        return self.nodes[-1]
+
+    def __eq__(self, other):
+        return self.nodes == other.nodes and self.relationships == other.relationships

redisgraph/query_result.py

@@ -1,5 +1,6 @@
 from .node import Node
 from .edge import Edge
+from .path import Path
 from prettytable import PrettyTable
 from redis import ResponseError
 
@@ -21,6 +22,7 @@ class ResultSetScalarTypes(object):
     VALUE_ARRAY = 6
     VALUE_EDGE = 7
     VALUE_NODE = 8
+    VALUE_PATH = 9
 
 class QueryResult(object):
     LABELS_ADDED = 'Labels added'
@@ -124,6 +126,12 @@ def parse_edge(self, cell):
         properties = self.parse_entity_properties(cell[4])
         return Edge(src_node_id, relation, dest_node_id, edge_id=edge_id, properties=properties)
 
+
+    def parse_path(self, cell):
+        nodes = self.parse_scalar(cell[0])
+        relationships = self.parse_scalar(cell[1])
+        return Path(nodes, relationships)
+
     def parse_scalar(self, cell):
         scalar_type = int(cell[0])
         value = cell[1]
@@ -162,6 +170,9 @@ def parse_scalar(self, cell):
         elif scalar_type == ResultSetScalarTypes.VALUE_EDGE:
             scalar = self.parse_edge(value)
 
+        elif scalar_type == ResultSetScalarTypes.VALUE_PATH:
+            scalar = self.parse_path(value)
+
         elif scalar_type == ResultSetScalarTypes.VALUE_UNKNOWN:
             print("Unknown scalar type\n")
 

test.py

@@ -65,5 +65,25 @@ def test_array_functions(self):
         # All done, remove graph.
         redis_graph.delete()
 
+
+    def test_path(self):
+        redis_graph = Graph('social', self.r)
+
+        query = """CREATE (a:L1)-[:R1 {value:1}]->(b:L1)-[:R1 {value:2}]->(c:L1)"""
+        redis_graph.query(query)
+        query = "MATCH p=(:L1)-[:R1]->(:L1) RETURN p"
+        node0 = Node(node_id=0, label="L1")
+        edge01 = Edge(0, "R1", 1, edge_id=0, properties={'value': 1})
+        node1 = Node(node_id=1, label="L1")
+        edge12 = Edge(1, "R1", 2, edge_id=1, properties={'value': 2})
+        node2 = Node(node_id=2, label="L1")
+
+        path01 = Path([node0, node1], [edge01])
+        path12 = Path([node1, node2], [edge12])
+        expected_results = [[path01], [path12]]
+        result = redis_graph.query(query)
+        self.assertEqual(expected_results, result.result_set)
+
+
 if __name__ == '__main__':
 	unittest.main()