add support for 'for-assignment' hints

2016-02-15 17:37:03 +01:00
parent 8b28678d19
commit 3a1b2e7104
4 changed files with 37 additions and 9 deletions
@@ -524,7 +524,10 @@ class BaseNode(Base):
        after the node, including the #
        """
        try:
-            whitespace = self.last_leaf().get_next().prefix
+            if self.isinstance(ForStmt):
+                whitespace = self.children[5].first_leaf().prefix
+            else:
+                whitespace = self.last_leaf().get_next().prefix
        except AttributeError:
            return None
        except ValueError: