[new] FileExtensionDialog for load-picks (e.g. _correlatex.xml)

PyLoT.py

@@ -79,7 +79,7 @@ from pylot.core.io.location import create_creation_info, create_event
 from pylot.core.util.widgets import FilterOptionsDialog, NewEventDlg, \
     PylotCanvas, WaveformWidgetPG, PropertiesDlg, HelpForm, createAction, PickDlg, \
     ComparisonWidget, TuneAutopicker, PylotParaBox, AutoPickDlg, CanvasWidget, AutoPickWidget, \
-    CompareEventsWidget, ProgressBarWidget, AddMetadataWidget
+    CompareEventsWidget, ProgressBarWidget, AddMetadataWidget, FileExtensionDialog
 from pylot.core.util.array_map import Array_map
 from pylot.core.util.structure import DATASTRUCTURE
 from pylot.core.util.thread import Thread, Worker
@@ -941,7 +941,11 @@ class MainWindow(QMainWindow):
             return
         refresh = False
         events = self.project.eventlist
-        fext = '.xml'
+        fed = FileExtensionDialog()
+        if not fed.exec_():
+            return
+        fext = fed.file_extension.text()
+        #fext = '.xml'
         for event in events:
             path = event.path
             eventname = path.split('/')[-1]  # or event.pylot_id

pylot/core/util/widgets.py

@@ -1489,6 +1489,39 @@ class PylotCanvas(FigureCanvas):
         self.draw()
 
 
+class FileExtensionDialog(QtGui.QDialog):
+    def __init__(self, parent=None, default_extension='.xml'):
+        super(FileExtensionDialog, self).__init__(parent)
+        self.default_extension = default_extension
+        self.setButtons()
+        self.connectSignals()
+        self.setupUi()
+
+    def setupUi(self):
+        self.main_layout = QtGui.QVBoxLayout()
+        self.sub_layout = QtGui.QHBoxLayout()
+        #
+        self.setLayout(self.main_layout)
+        self.textLabel = QtGui.QLabel('Specify file extension: ')
+        self.file_extension = QtGui.QLineEdit(self.default_extension)
+
+        self.sub_layout.addWidget(self.textLabel)
+        self.sub_layout.addWidget(self.file_extension)
+
+        self.main_layout.addLayout(self.sub_layout)
+        self.main_layout.addWidget(self._buttonbox)
+
+    def setButtons(self):
+        self._buttonbox = QDialogButtonBox(QDialogButtonBox.Ok |
+                                           QDialogButtonBox.Cancel)
+
+    def connectSignals(self):
+        self._buttonbox.accepted.connect(self.accept)
+        self._buttonbox.rejected.connect(self.reject)
+
+
+
+
 class PhaseDefaults(QtGui.QDialog):
     def __init__(self, parent=None, nrow=10,
                  phase_defaults=['P', 'S'],