oweditdomain: Indicate items with warnings with a yellow color

Hide/show warning on change

Author: ales-erjavec

Orange/widgets/data/oweditdomain.py

@@ -1577,6 +1577,9 @@ def variable_icon(var):
 #: (`List[Union[ReinterpretTransform, Transform]]`)
 TransformRole = Qt.UserRole + 42
 
+#: Any warnings applying to the transform (`list[tuple[Msg, str]]`)
+RestoreWarningRole = TransformRole + 1
+
 
 class VariableEditDelegate(QStyledItemDelegate):
     ReinterpretNames = {
@@ -1624,9 +1627,16 @@ def initStyleOption(self, option, index):
             option.font.setItalic(True)
 
         multiplicity = index.data(MultiplicityRole)
+        warnings = index.data(RestoreWarningRole)
+
+        def set_color(palette: QPalette, color):
+            palette.setBrush(QPalette.Text, QBrush(color))
+            palette.setBrush(QPalette.HighlightedText, QBrush(color))
+
         if isinstance(multiplicity, int) and multiplicity > 1:
-            option.palette.setBrush(QPalette.Text, QBrush(Qt.red))
-            option.palette.setBrush(QPalette.HighlightedText, QBrush(Qt.red))
+            set_color(option.palette, Qt.red)
+        elif warnings:
+            set_color(option.palette, Qt.yellow)
 
     def helpEvent(self, event: QHelpEvent, view: QAbstractItemView,
                   option: QStyleOptionViewItem, index: QModelIndex) -> bool:
@@ -2123,6 +2133,7 @@ def reset_selected(self):
         modified = []
         for ind in self.selected_var_indices():
             midx = model.index(ind)
+            model.setData(midx, None, RestoreWarningRole)
             if midx.data(TransformRole):
                 model.setData(midx, [], TransformRole)
                 var = midx.data(Qt.EditRole)
@@ -2133,6 +2144,7 @@ def reset_selected(self):
                               self._on_variable_changed):
                 self._editor.set_data(modified)
             self._invalidate()
+        self._update_restore_warnings()
 
     def reset_all(self):
         """Reset all variables to their original state."""
@@ -2141,8 +2153,10 @@ def reset_all(self):
             for i in range(model.rowCount()):
                 midx = model.index(i)
                 model.setData(midx, [], TransformRole)
+                model.setData(midx, None, RestoreWarningRole)
             self.open_editor()
             self._invalidate()
+            self._update_restore_warnings()
 
     def selected_var_indices(self):
         """Return the current selected variable indices."""
@@ -2199,7 +2213,6 @@ def _restore(self):
         model = self.variables_model
         hints = self._domain_change_hints
         first_key = None
-        msgs = []
         for i in range(model.rowCount()):
             midx = model.index(i, 0)
             coldesc = model.data(midx, Qt.EditRole)  # type: DataVector
@@ -2208,9 +2221,9 @@ def _restore(self):
                 key, tr = res
                 if tr:
                     self._store_transform(coldesc.vtype, tr, key)
-                    tr, msgs_ = self._sanitize_transform(coldesc.vtype, tr)
+                    tr, msgs = self._sanitize_transform(coldesc.vtype, tr)
                     model.setData(midx, tr, TransformRole)
-                    msgs.extend(msgs_)
+                    model.setData(midx, msgs, RestoreWarningRole)
                     if first_key is None:
                         first_key = key
         # Reduce the number of hints to MAX_HINTS, but keep all current hints
@@ -2219,9 +2232,7 @@ def _restore(self):
                 (key := next(iter(hints))) != first_key:
             del hints[key]  # pylint: disable=unsupported-delete-operation
 
-        # Show warnings for non-applicable transforms
-        for msg, names in groupby(msgs, key=itemgetter(0)):
-            msg(", ".join(map(itemgetter(1), names)))
+        self._update_restore_warnings()
 
         # Restore the current variable selection
         selected_rows = [i for i, vec in enumerate(model)
@@ -2230,6 +2241,19 @@ def _restore(self):
             selected_rows = [0]
         itemmodels.select_rows(self.variables_view, selected_rows)
 
+    def _update_restore_warnings(self):
+        def messages(midx):
+            msgs = model.data(midx, RestoreWarningRole)
+            return msgs or []
+        model = self.variables_model
+        msgs = chain.from_iterable(
+            messages(model.index(i)) for i in range(model.rowCount())
+        )
+        self.Warning.cat_mapping_does_not_apply.clear()
+        # Show warnings for non-applicable transforms
+        for msg, names in groupby(msgs, key=itemgetter(0)):
+            msg(", ".join(map(itemgetter(1), names)))
+
     def _on_selection_changed(self, _, deselected):
         # If the user deselected the last item, select it back with disabled
         # signals, so nothing happens
@@ -2278,8 +2302,10 @@ def _on_variable_changed(self):
                                             *editor.get_data()):
             midx = model.index(idx, 0)
             model.setData(midx, transform, TransformRole)
+            model.setData(midx, None, RestoreWarningRole)
             self._store_transform(var, transform)
         self._invalidate()
+        self._update_restore_warnings()
 
     def _store_transform(
             self, var: Variable, transform: Iterable[Transform], deconvar=None

Orange/widgets/data/tests/test_oweditdomain.py

@@ -40,6 +40,7 @@
 from Orange.widgets.data.owcolor import OWColor, ColorRole
 from Orange.widgets.tests.base import WidgetTest, GuiTest
 from Orange.widgets.tests.utils import contextMenu
+from Orange.widgets.utils.itemmodels import select_row
 from Orange.widgets.utils import colorpalettes
 from Orange.tests import test_filename, assert_array_nanequal
 
@@ -382,6 +383,10 @@ def restore(state):
         self.assertEqual(output.domain.class_var.values,
                          ("Iris-setosa", "Iris-versicolor", "Iris-virginica"))
 
+        select_row(w.variables_view, 4)
+        w.reset_selected()
+        self.assertFalse(w.Warning.cat_mapping_does_not_apply.is_shown())
+
         restore({viris: [("Rename", ("A")), ("NonexistantTransform", ("AA",))]})
         tr = model.data(model.index(4), TransformRole)
         self.assertEqual(tr, [Rename("A")])

Orange/widgets/utils/itemmodels.py

@@ -37,9 +37,11 @@
 __all__ = [
     "PyListModel", "VariableListModel", "PyListModelTooltip", "DomainModel",
     "AbstractSortTableModel", "PyTableModel", "TableModel",
-    "ModelActionsWidget", "ListSingleSelectionModel"
+    "ModelActionsWidget", "ListSingleSelectionModel",
+    "select_row", "select_rows", "signal_blocking"
 ]
 
+
 @contextmanager
 def signal_blocking(obj):
     blocked = obj.signalsBlocked()