Skip to content

fixed path nodes and edges functions #68

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Mar 12, 2020
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 13 additions & 13 deletions redisgraph/path.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,52 +5,52 @@ class Path(object):

def __init__(self, nodes, edges):
assert(isinstance(nodes, list) and isinstance(edges, list))
self.nodes = nodes
self.edges = edges
self._nodes = nodes
self._edges = edges
self.append_type = Node

@classmethod
def new_empty_path(cls):
return cls([], [])

def nodes(self):
return self.nodes
return self._nodes

def edges(self):
return self.edges
return self._edges

def get_node(self, index):
return self.nodes[index]
return self._nodes[index]

def get_relationship(self, index):
return self.edges[index]
return self._edges[index]

def first_node(self):
return self.nodes[0]
return self._nodes[0]

def last_node(self):
return self.nodes[-1]
return self._nodes[-1]

def edge_count(self):
return len(self.edges)
return len(self._edges)

def nodes_count(self):
return len(self.nodes)
return len(self._nodes)

def add_node(self, node):
assert(type(node) == self.append_type)
self.nodes.append(node)
self._nodes.append(node)
self.append_type = Edge
return self

def add_edge(self, edge):
assert(type(edge) == self.append_type)
self.edges.append(edge)
self._edges.append(edge)
self.append_type = Node
return self

def __eq__(self, other):
return self.nodes == other.nodes and self.edges == other.edges
return self.nodes() == other.nodes() and self.edges() == other.edges()

def __str__(self):
res = "<"
Expand Down