Fix _get_renderer for FigureCanvasBase.

anntzer · anntzer · commit 5f6f8550b4c3 · 2020-06-24T15:54:25.000+02:00
diff --git a/lib/matplotlib/backend_bases.py b/lib/matplotlib/backend_bases.py
@@ -1534,7 +1534,6 @@ def _get_renderer(figure, print_method=None):
 
     If you need a renderer without any active draw methods use
     renderer._draw_disabled to temporary patch them out at your call site.
-
     """
     # This is implemented by triggering a draw, then immediately jumping out of
     # Figure.draw() by raising an exception.
@@ -1545,9 +1544,13 @@ class Done(Exception):
     def _draw(renderer): raise Done(renderer)
 
     with cbook._setattr_cm(figure, draw=_draw):
+        orig_canvas = figure.canvas
         if print_method is None:
             fmt = figure.canvas.get_default_filetype()
-            print_method = getattr(figure.canvas, f"print_{fmt}")
+            # Even for a canvas' default output type, a canvas switch may be
+            # needed, e.g. for FigureCanvasBase.
+            print_method = getattr(
+                figure.canvas._get_output_canvas(None, fmt), f"print_{fmt}")
         try:
             print_method(io.BytesIO(), dpi=figure.dpi)
         except Done as exc:
@@ -1556,6 +1559,8 @@ def _draw(renderer): raise Done(renderer)
         else:
             raise RuntimeError(f"{print_method} did not call Figure.draw, so "
                                f"no renderer is available")
+        finally:
+            figure.canvas = orig_canvas
 
 
 def _is_non_interactive_terminal_ipython(ip):
