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
@@ -154,6 +154,13 @@ def find_type_from_comment_hint_for(evaluator, node, name):
        _find_type_from_comment_hint(evaluator, node, node.children[1], name)


+def find_type_from_comment_hint_with(evaluator, node, name):
+    assert len(node.children[1].children) == 3, \
+        "Can only be here when children[1] is 'foo() as f'"
+    return _find_type_from_comment_hint(
+        evaluator, node, node.children[1].children[2], name)
+
+
 def find_type_from_comment_hint_assign(evaluator, node, name):
    return \
        _find_type_from_comment_hint(evaluator, node, node.children[0], name)