VimPlug/parso
1
0
Fork 0
mirror of https://github.com/davidhalter/parso.git synced 2025-12-09 14:14:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix get_next_sibling on module, fixes #102

Browse Source
This commit is contained in:
Dave Halter
2020-02-21 15:54:55 +01:00
parent 6dd29c8efb
commit 53da7e8e6b
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
parso/tree.py
View File

@@ -45,8 +45,12 @@ class NodeOrLeaf(object):
Returns the node immediately following this node in this parent's Returns the node immediately following this node in this parent's
children list. If this node does not have a next sibling, it is None children list. If this node does not have a next sibling, it is None
""" """
parent = self.parent
if parent is None:
return None
# Can't use index(); we need to test by identity # Can't use index(); we need to test by identity
for i, child in enumerate(self.parent.children): for i, child in enumerate(parent.children):
if child is self: if child is self:
try: try:
return self.parent.children[i + 1] return self.parent.children[i + 1]
@@ -59,8 +63,12 @@ class NodeOrLeaf(object):
children list. If this node does not have a previous sibling, it is children list. If this node does not have a previous sibling, it is
None. None.
""" """
parent = self.parent
if parent is None:
return None
# Can't use index(); we need to test by identity # Can't use index(); we need to test by identity
for i, child in enumerate(self.parent.children): for i, child in enumerate(parent.children):
if child is self: if child is self:
if i == 0: if i == 0:
return None return None
@@ -71,6 +79,9 @@ class NodeOrLeaf(object):
Returns the previous leaf in the parser tree. Returns the previous leaf in the parser tree.
Returns `None` if this is the first element in the parser tree. Returns `None` if this is the first element in the parser tree.
""" """
if self.parent is None:
return None
node = self node = self
while True: while True:
c = node.parent.children c = node.parent.children
@@ -94,6 +105,9 @@ class NodeOrLeaf(object):
Returns the next leaf in the parser tree. Returns the next leaf in the parser tree.
Returns None if this is the last element in the parser tree. Returns None if this is the last element in the parser tree.
""" """
if self.parent is None:
return None
node = self node = self
while True: while True:
c = node.parent.children c = node.parent.children