Fix exception when using Gtk+ file chooser to save a file and the user

mdboom · mdboom · commit e6f8434c2a98 · 2007-07-08T16:38:25.000Z
provides no extension.

svn path=/trunk/matplotlib/; revision=3484
diff --git a/lib/matplotlib/backends/backend_gtk.py b/lib/matplotlib/backends/backend_gtk.py
@@ -1067,13 +1067,15 @@ def __init__ (self,
             for item in IMAGE_FORMAT:
                 cbox.append_text (item)
             cbox.set_active (IMAGE_FORMAT.index (IMAGE_FORMAT_DEFAULT))
+            self.ext = IMAGE_FORMAT_DEFAULT
 
             def cb_cbox_changed (cbox, data=None):
                 """File extension changed"""
                 head, filename = os.path.split(self.get_filename())
                 root, ext = os.path.splitext(filename)
                 ext = ext[1:]
                 new_ext = IMAGE_FORMAT[cbox.get_active()]
+                self.ext = new_ext
 
                 if ext in IMAGE_FORMAT:
                     filename = filename.replace(ext, new_ext)
@@ -1092,11 +1094,10 @@ def get_filename_from_user (self):
                 if self.run() != gtk.RESPONSE_OK:
                     break
                 filename = self.get_filename()
-                menu_ext  = IMAGE_FORMAT[self.cbox.get_active()]
                 root, ext = os.path.splitext (filename)
                 ext = ext[1:]
                 if ext == '':
-                    ext = menu_ext
+                    ext = self.ext
                     filename += '.' + ext
 
                 if ext in IMAGE_FORMAT:
@@ -1106,7 +1107,7 @@ def get_filename_from_user (self):
                     error_msg_gtk ('Image format "%s" is not supported' % ext,
                                    parent=self)
                     self.set_current_name (os.path.split(root)[1] + '.' +
-                                           menu_ext)
+                                           self.ext)
 
             self.hide()
             return filename
