FunctionExecution improvement.

2016-10-02 19:54:03 +02:00
parent c2873792eb
commit 2f1e9d634f
3 changed files with 16 additions and 11 deletions
--- a/jedi/parser/tree.py
+++ b/jedi/parser/tree.py
@@ -58,13 +58,10 @@ def search_ancestor(node, node_type_or_types):
    if not isinstance(node_type_or_types, (list, tuple)):
        node_type_or_types = (node_type_or_types,)

-    node = node.parent
-    while node.type not in node_type_or_types:
-        try:
-            node = node.parent
-        except AttributeError:
-            return None
-    return node
+    while True:
+        node = node.parent
+        if node is None or node.type in node_type_or_types:
+            return node


 class PositionModifier(object):