add support for 'with-assignment' hints

2016-02-15 17:52:21 +01:00
parent 3a1b2e7104
commit 641fb80773
4 changed files with 20 additions and 0 deletions
--- a/jedi/parser/tree.py
+++ b/jedi/parser/tree.py
@@ -526,6 +526,8 @@ class BaseNode(Base):
        try:
            if self.isinstance(ForStmt):
                whitespace = self.children[5].first_leaf().prefix
+            elif self.isinstance(WithStmt):
+                whitespace = self.children[3].first_leaf().prefix
            else:
                whitespace = self.last_leaf().get_next().prefix
        except AttributeError: