fixed wxagg ans gtkagg switch bug

jdh2358 · jdh2358 · commit cbb3c188386b · 2007-05-15T21:20:12.000Z
svn path=/trunk/matplotlib/; revision=3271
diff --git a/lib/matplotlib/backends/backend_gtkagg.py b/lib/matplotlib/backends/backend_gtkagg.py
@@ -114,6 +114,9 @@ def print_figure(self, filename, dpi=None, facecolor='w', edgecolor='w',
                                  orientation, **kwargs)
             except IOError, msg:
                 error_msg_gtk('Failed to save\nError message: %s'%(msg,), self)
+            except:
+                self.figure.set_canvas(self)
+                raise
 
         self.figure.set_canvas(self)
         if DEBUG: print 'FigureCanvasGTKAgg.print_figure done'
diff --git a/lib/matplotlib/backends/backend_wxagg.py b/lib/matplotlib/backends/backend_wxagg.py
@@ -102,9 +102,14 @@ def print_figure(self, filename, dpi=None, facecolor='w', edgecolor='w',
         """
         if dpi is None: dpi = matplotlib.rcParams['savefig.dpi']
         agg = self.switch_backends(FigureCanvasAgg)
-        agg.print_figure(filename, dpi, facecolor, edgecolor, orientation,
-                         **kwargs)
-        self.figure.set_canvas(self)
+        try:
+            agg.print_figure(filename, dpi, facecolor, edgecolor, orientation,
+                             **kwargs)
+        except:
+            self.figure.set_canvas(self)
+            raise
+        else:
+            self.figure.set_canvas(self)
 
     def _get_imagesave_wildcards(self):
         'return the wildcard string for the filesave dialog'
