hook for providing type of 'with' block variables; provide type for contextlib.closing

python/psi-api/src/com/jetbrains/python/psi/impl/PyTypeProvider.java

@@ -28,4 +28,7 @@ public interface PyTypeProvider {
 
   @Nullable
   PyType getIterationType(PyClass iterable);
+
+  @Nullable
+  PyType getContextManagerVariableType(PyClass contextManager, PyExpression withExpression, TypeEvalContext context);
 }

python/psi-api/src/com/jetbrains/python/psi/types/PyTypeProviderBase.java

@@ -96,6 +96,12 @@ public class PyTypeProviderBase implements PyTypeProvider {
     return null;
   }
 
+  @Nullable
+  @Override
+  public PyType getContextManagerVariableType(PyClass contextManager, PyExpression withExpression, TypeEvalContext context) {
+    return null;
+  }
+
   protected void registerSelfReturnType(String classQualifiedName, Collection<String> methods) {
     registerReturnType(classQualifiedName, methods, mySelfTypeCallback);
   }