Auto add file type suffix if missing.

Godrigos · Oct 16, 2018 · 396f0ba · 396f0ba
1 parent fd5274b
commit 396f0ba
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/msaj.py b/msaj.py
@@ -181,7 +181,9 @@ def report(self):
     def save_file(self):
         options = QFileDialog.Options()
         self.filename = QFileDialog.getSaveFileName(self, "Save File", str(Path.home()),
-                                                    "NEXUS Files (*.nex *.nexus *.nxs)", options=options)[0] + '.nex'
+                                                    "NEXUS Files (*.nex *.nexus *.nxs)", options=options)[0]
+        if QFileInfo(self.filename).suffix() != ('.nex' and '.nexus') and basename(splitext(self.filename)[0]) != '':
+            self.filename += '.nex'
         return self.filename
 
     def msaj(self):