- Fix not capturing error from visualization lint

- enable linting of visualization table

Close #134

Author: PaulJonasJost

src/petab_gui/controllers/mother_controller.py

@@ -836,9 +836,10 @@ def new_file(self):
     def check_model(self):
         """Check the consistency of the model. And log the results."""
         capture_handler = CaptureLogHandler()
-        logger = logging.getLogger("petab.v1.lint")  # Target the specific
-        # logger
-        logger.addHandler(capture_handler)
+        logger_lint = logging.getLogger("petab.v1.lint")
+        logger_vis = logging.getLogger("petab.v1.visualize.lint")
+        logger_lint.addHandler(capture_handler)
+        logger_vis.addHandler(capture_handler)
 
         try:
             # Run the consistency check
@@ -867,7 +868,8 @@ def check_model(self):
             self.logger.log_message(msg, color="red")
         finally:
             # Always remove the capture handler
-            logger.removeHandler(capture_handler)
+            logger_lint.removeHandler(capture_handler)
+            logger_vis.removeHandler(capture_handler)
 
     def unsaved_changes_change(self, unsaved_changes: bool):
         self.unsaved_changes = unsaved_changes
@@ -1150,3 +1152,7 @@ def _show_help_welcome(self):
         msg.exec()
         if dont.isChecked():
             settings.setValue("help_mode/welcome_disabled", True)
+
+    def get_current_problem(self):
+        """Get the current PEtab problem from the model."""
+        return self.model.current_petab_problem

src/petab_gui/controllers/table_controllers.py

@@ -71,7 +71,7 @@ def __init__(
         self.undo_stack = undo_stack
         self.model.undo_stack = undo_stack
         self.check_petab_lint_mode = True
-        if model.table_type in ["simulation", "visualization"]:
+        if model.table_type in ["simulation"]:
             self.check_petab_lint_mode = False
         self.mother_controller = mother_controller
         self.view.table_view.setModel(self.proxy_model)
@@ -1315,3 +1315,14 @@ def __init__(
             undo_stack=undo_stack,
             mother_controller=mother_controller,
         )
+
+    @linter_wrapper(additional_error_check=True)
+    def check_petab_lint(
+        self,
+        row_data: pd.DataFrame = None,
+        row_name: str = None,
+        col_name: str = None,
+    ):
+        """Check a number of rows of the model with petablint."""
+        problem = self.mother_controller.get_current_problem()
+        return petab.visualize.validate_visualization_df(problem)