fixup! Implement the Correlator special suite

European-XFEL · Feb 17, 2022 · e685bbe · e685bbe
1 parent 48df603
commit e685bbe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/extra_foam/special_suite/correlator_w.py b/extra_foam/special_suite/correlator_w.py
@@ -999,6 +999,9 @@ def _saveContext(self):
             path = QFileDialog.getSaveFileName(self, "Save context", filter="Python (*.py)")[0]
 
         if len(path) > 0:
+            if not path.endswith(".py"):
+                path += ".py"
+
             with open(path, "w") as context_file:
                 context_file.write(self._editor.text())
                 context_file.flush()