Skip to content

Commit 22cba75

Browse files
author
Umair Idris
committed
Suggest unexisting default filename (Implement matplotlib#3608)
1 parent c90469b commit 22cba75

File tree

1 file changed

+14
-3
lines changed

1 file changed

+14
-3
lines changed

lib/matplotlib/backend_bases.py

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2253,9 +2253,20 @@ def get_default_filename(self):
22532253
Return a string, which includes extension, suitable for use as
22542254
a default filename.
22552255
"""
2256-
default_filename = self.get_window_title() or 'image'
2257-
default_filename = default_filename.lower().replace(' ', '_')
2258-
return default_filename + '.' + self.get_default_filetype()
2256+
default_basename = self.get_window_title() or 'image'
2257+
default_basename = default_basename.lower().replace(' ', '_')
2258+
default_filetype = self.get_default_filetype()
2259+
default_filename = default_basename + '.' + default_filetype
2260+
2261+
dir_path = os.path.expanduser(rcParams.get('savefig.directory', ''))
2262+
2263+
i = 1
2264+
while os.path.isfile(os.path.join(dir_path, default_filename)): # file aleady exists
2265+
# attach numerical value to filename
2266+
default_filename = '{}-{}.{}'.format(default_basename, i, default_filetype)
2267+
i += 1
2268+
2269+
return default_filename
22592270

22602271
def switch_backends(self, FigureCanvasClass):
22612272
"""

0 commit comments

Comments
 (0)