check extension on save

git-svn-id: http://gnuradio.org/svn/gnuradio/trunk@10097 221aa14e-8319-0410-a670-987f0aec2ac5

diff --git a/grc/src/gui/FileDialogs.py b/grc/src/gui/FileDialogs.py
index efd3fecfed59f1c827e0aa15867dd1cd8154f776..7c10d9844b89e3c8654ba20e3179173f71aef10f 100644 (file)
--- a/grc/src/gui/FileDialogs.py
+++ b/grc/src/gui/FileDialogs.py
@@ -119,12 +119,12 @@ class FileDialog(FileDialogHelper):
                #############################################
                if self.type in (SAVE_FLOW_GRAPH, SAVE_IMAGE):
                        filename = self.get_filename()
-                       for extension, filter in (
-                               (Preferences.file_extension(), get_flow_graph_files_filter()),
-                               (IMAGE_FILE_EXTENSION, get_image_files_filter()),
-                       ): #append the missing file extension if the filter matches
-                               if filename[len(filename)-len(extension):] != extension \
-                                       and filter == self.get_filter(): filename += extension
+                       extension = {
+                               SAVE_FLOW_GRAPH: Preferences.file_extension(),
+                               SAVE_IMAGE: IMAGE_FILE_EXTENSION,
+                       }[self.type]
+                       #append the missing file extension if the filter matches
+                       if path.splitext(filename)[1].lower() != extension: filename += extension
                        self.set_current_name(path.basename(filename)) #show the filename with extension
                        if path.exists(filename): #ask the user to confirm overwrite
                                if MessageDialogHelper(