fix #75 : catch exceptions when saving or loading files and display a QMessageBox instead

Author: alixdamman

larray_editor/editor.py

@@ -539,23 +539,34 @@ def new(self):
             self.statusBar().showMessage("Viewer has been reset", 4000)
 
     def _open_file(self, filepath):
-        self._reset()
+        def _update_arrays(current_file_name, session):
+            self._reset()
+            self.set_current_file(current_file_name)
+            self._add_arrays(session)
+            self._listwidget.setCurrentRow(0)
+            self.unsaved_modifications = False
+
         session = Session()
         if '.csv' in filepath:
             filepath = [filepath]
         if isinstance(filepath, (list, tuple)):
-            session.load(None, filepath)
             dirname = os.path.dirname(filepath[0])
             basenames = [os.path.basename(fpath) for fpath in filepath]
-            self.set_current_file(dirname)
-            self.statusBar().showMessage("CSV files {} loaded".format(' ,'.join(basenames)), 4000)
+            try:
+                session.load(None, filepath)
+                _update_arrays(dirname, session)
+                self.statusBar().showMessage("CSV files {} loaded".format(' ,'.join(basenames)), 4000)
+            except Exception as e:
+                QMessageBox.critical(self, "Error", "Something went wrong during load of CSV files {}:\n{}"
+                                     .format(' ,'.join(basenames), e))
         else:
-            session.load(filepath)
-            self.set_current_file(filepath)
-            self.statusBar().showMessage("File {} loaded".format(os.path.basename(filepath)), 4000)
-        self._add_arrays(session)
-        self._listwidget.setCurrentRow(0)
-        self.unsaved_modifications = False
+            try:
+                session.load(filepath)
+                _update_arrays(filepath, session)
+                self.statusBar().showMessage("File {} loaded".format(os.path.basename(filepath), 4000))
+            except Exception as e:
+                QMessageBox.critical(self, "Error", "Something went wrong during load of file {}:\n{}"
+                                     .format(filepath, e))
 
     def open(self):
         if self._ask_to_save_if_unsaved_modifications():
@@ -583,11 +594,14 @@ def open_recent_file(self):
                     QMessageBox.warning(self, "Warning", "File {} could not be found".format(filepath))
 
     def _save_data(self, filepath):
-        session = Session({k: v for k, v in self.data.items() if self._display_in_grid(k, v)})
-        session.save(filepath)
-        self.set_current_file(filepath)
-        self.unsaved_modifications = False
-        self.statusBar().showMessage("Arrays saved in file {}".format(filepath), 4000)
+        try:
+            session = Session({k: v for k, v in self.data.items() if self._display_in_grid(k, v)})
+            session.save(filepath)
+            self.set_current_file(filepath)
+            self.unsaved_modifications = False
+            self.statusBar().showMessage("Arrays saved in file {}".format(filepath), 4000)
+        except Exception as e:
+            QMessageBox.critical(self, "Error", "Something went wrong during save in file {}:\n{}".format(filepath, e))
 
     def save(self):
         """