ENH: make MainLoopBase a context manager

matplotlib · tacaswell · Feb 21, 2015 · Feb 22, 2015 · Feb 25, 2015 · Feb 26, 2015
commit 072c47ac7a2ce792217ed189dfbfa5562013a948
diff --git a/lib/matplotlib/backend_bases.py b/lib/matplotlib/backend_bases.py
@@ -151,12 +151,14 @@ def get_registered_canvas_class(format):
 
 class MainLoopBase(object):
     """This gets used as a key maintaining the event loop.
+
     Backends should only need to override begin and end.
     It should not matter if this gets used as a singleton or not due to
     clever magic.
     """
     _instance_count = {}
     _running = False
+
     def __init__(self):
         MainLoopBase._instance_count.setdefault(self.__class__, 0)
         MainLoopBase._instance_count[self.__class__] += 1
@@ -177,6 +179,14 @@ def __del__(self):
                 not is_interactive() and MainLoopBase._running):
             self.end()
 
+    def __enter__(self):
+        pass
+
+    def __exit__(self, excp_type, excp_value, excp_traceback):
+        if not excp_type:
+            self.begin()
+            self.end()
+
 
 class ShowBase(object):
     """