From 4ffc24a919118dc6d0d6b63294ef70a2104dc2ff Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Mon, 21 Sep 2015 15:33:59 +0200 Subject: [PATCH] Fix for loops. --- jedi/parser/tree.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/jedi/parser/tree.py b/jedi/parser/tree.py index b28b4cd9..79c549bb 100644 --- a/jedi/parser/tree.py +++ b/jedi/parser/tree.py @@ -982,6 +982,12 @@ class ForStmt(Flow): type = 'for_stmt' __slots__ = () + def nodes_to_execute(self, last_added=False): + # We shouldn't include the definitions. + for child in self.children[3:]: + for node_to_execute in child.nodes_to_execute(): + yield node_to_execute + class TryStmt(Flow): type = 'try_stmt'