Catch errors in the timeline export

UmSenhorQualquer · Mar 14, 2019 · 59869d4 · 59869d4
1 parent 332f3e4
commit 59869d4
Showing 1 changed file with 43 additions and 33 deletions.
diff --git a/pyforms_gui/controls/control_event_timeline/control_eventtimeline.py b/pyforms_gui/controls/control_event_timeline/control_eventtimeline.py
@@ -487,41 +487,51 @@ def __open_import_win_evt(self):
     def __export(self):
         """Export annotations to a file."""
 
-        if conf.PYFORMS_DIALOGS_OPTIONS:
-            filename, ffilter = QFileDialog.getSaveFileName(parent=self,
-                 caption="Export annotations file",
-                 directory="untitled.csv",
-                 filter="CSV Files (*.csv);;CSV Matrix Files (*.csv)",
-                 options=conf.PYFORMS_DIALOGS_OPTIONS)
-        else:
-            filename, ffilter = QFileDialog.getSaveFileName(parent=self,
-                                                            caption="Export annotations file",
-                                                            directory="untitled.csv",
-                                                            filter="CSV Files (*.csv);;CSV Matrix Files (*.csv)")
-
-        filename = str(filename)
-        ffilter = str(ffilter)
-        if filename != "":
-            with open(filename, 'w') as csvfile:
-                spamwriter = csv.writer(csvfile, dialect='excel')
-                if ffilter == 'CSV Files (*.csv)':
-                    self._time.export_events_to_csvwriter(spamwriter)
-                elif ffilter == 'CSV Matrix Files (*.csv)':
-                    self._time.exportmatrix_events_to_csvwriter(spamwriter)
+        try:
+
+            if conf.PYFORMS_DIALOGS_OPTIONS:
+                filename, ffilter = QFileDialog.getSaveFileName(parent=self,
+                     caption="Export annotations file",
+                     directory="untitled.csv",
+                     filter="CSV Files (*.csv);;CSV Matrix Files (*.csv)",
+                     options=conf.PYFORMS_DIALOGS_OPTIONS)
+            else:
+                filename, ffilter = QFileDialog.getSaveFileName(parent=self,
+                                                                caption="Export annotations file",
+                                                                directory="untitled.csv",
+                                                                filter="CSV Files (*.csv);;CSV Matrix Files (*.csv)")
+
+            filename = str(filename)
+            ffilter = str(ffilter)
+            if filename != "":
+                with open(filename, 'w') as csvfile:
+                    spamwriter = csv.writer(csvfile, dialect='excel')
+                    if ffilter == 'CSV Files (*.csv)':
+                        self._time.export_events_to_csvwriter(spamwriter)
+                    elif ffilter == 'CSV Matrix Files (*.csv)':
+                        self._time.exportmatrix_events_to_csvwriter(spamwriter)
+
+        except Exception as e:
+            m = QMessageBox(QMessageBox.Critical, 'Error', str(e))
+            m.exec_()
 
     def __export_2_csv_matrix(self):
-        QMessageBox.warning(
-            self, "Important!", 'Please note that this file cannot be imported after.')
-
-        filename, _ = QFileDialog.getSaveFileName(parent=self,
-                                               caption="Export matrix file",
-                                               directory="untitled.csv",
-                                               filter="CSV Files (*.csv)",
-                                               options=QFileDialog.DontUseNativeDialog)
-        if filename != "":
-            with open(filename, 'w') as csvfile:
-                spamwriter = csv.writer(csvfile, dialect='excel')
-                self._time.exportmatrix_events_to_csvwriter(spamwriter)
+        try:
+            QMessageBox.warning(
+                self, "Important!", 'Please note that this file cannot be imported after.')
+
+            filename, _ = QFileDialog.getSaveFileName(parent=self,
+                                                   caption="Export matrix file",
+                                                   directory="untitled.csv",
+                                                   filter="CSV Files (*.csv)",
+                                                   options=QFileDialog.DontUseNativeDialog)
+            if filename != "":
+                with open(filename, 'w') as csvfile:
+                    spamwriter = csv.writer(csvfile, dialect='excel')
+                    self._time.exportmatrix_events_to_csvwriter(spamwriter)
+        except Exception as e:
+            m = QMessageBox(QMessageBox.Critical, 'Error', str(e))
+            m.exec_()
 
     def __remove_current_track_evt(self):
         reply = QMessageBox.question(self, 'Confirm',