Get docstrings working.

2014-11-20 02:19:01 +01:00
parent ce5d428d22
commit 164518b993
4 changed files with 14 additions and 11 deletions
--- a/jedi/parser/representation.py
+++ b/jedi/parser/representation.py
@@ -543,9 +543,7 @@ class Scope(Simple, DocstringMixin):

    @property
    def statements(self):
-        return [s for c in self.children if is_node(c, 'simple_stmt')
-                for s in c.children if isinstance(s, (ExprStmt, Import,
-                                                      KeywordStatement))]
+        return self._search_in_scope((ExprStmt, Import, KeywordStatement))

    def is_scope(self):
        return True