Get with statements working.

2014-11-13 12:51:49 +01:00
parent 541b8872d0
commit f3c2b4fc33
2 changed files with 17 additions and 2 deletions
--- a/jedi/evaluate/finder.py
+++ b/jedi/evaluate/finder.py
@@ -299,6 +299,8 @@ class NameFinder(object):
                types += self._eval_param(typ)
            elif typ.isinstance(pr.ExprStmt):
                types += self._remove_statements(typ, name)
+            elif typ.isinstance(pr.WithStmt):
+                types += evaluator.eval_element(typ.node_from_name(name))
            elif isinstance(typ, pr.Import):
                types += imports.ImportWrapper(self._evaluator, name).follow()
            elif isinstance(typ, pr.TryStmt):