Implement savefig.directory for gtk gui

mspacek · mspacek · commit 4372f763c659 · 2013-02-01T14:44:32.000-08:00
diff --git a/lib/matplotlib/backends/backend_gtk.py b/lib/matplotlib/backends/backend_gtk.py
@@ -732,6 +732,7 @@ def get_filechooser(self):
         fc = FileChooserDialog(
             title='Save the figure',
             parent=self.win,
+            path=os.path.expanduser(rcParams.get('savefig.directory', '')),
             filetypes=self.canvas.get_supported_filetypes(),
             default_filetype=self.canvas.get_default_filetype())
         fc.set_current_name(self.canvas.get_default_filename())
@@ -740,6 +741,13 @@ def get_filechooser(self):
     def save_figure(self, *args):
         fname, format = self.get_filechooser().get_filename_from_user()
         if fname:
+            startpath = os.path.expanduser(rcParams.get('savefig.directory', ''))
+            if startpath == '':
+                # explicitly missing key or empty str signals to use cwd
+                rcParams['savefig.directory'] = startpath
+            else:
+                # save dir for next time
+                rcParams['savefig.directory'] = os.path.dirname(str(fname))
             try:
                 self.canvas.print_figure(fname, format=format)
             except Exception as e:
