Make ConditionalTypeBinder a context manager.

Now we can use `with binder: X` instead of `binder.push_frame(); X; binder.pop_frame()`
python · Apr 19, 2016 · 9b7b383 · 9b7b383
1 parent 49e3864
commit 9b7b383
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/mypy/checker.py b/mypy/checker.py
@@ -1,6 +1,7 @@
 """Mypy type checker."""
 
 import itertools
+import contextlib
 
 from typing import (
     Any, Dict, Set, List, cast, Tuple, TypeVar, Union, Optional, NamedTuple
@@ -295,6 +296,12 @@ def push_loop_frame(self):
     def pop_loop_frame(self):
         self.loop_frames.pop()
 
+    def __enter__(self) -> None:
+        self.push_frame()
+
+    def __exit__(self, *args: Any) -> None:
+        self.pop_frame()
+
 
 def meet_frames(*frames: Frame) -> Frame:
     answer = Frame()