highlight log

hanjinliu · Aug 17, 2021 · d420543 · d420543
1 parent 084c39b
commit d420543
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/impy/viewer/widgets/log.py b/impy/viewer/widgets/log.py
@@ -2,16 +2,24 @@
 import napari
 from qtpy.QtWidgets import QPlainTextEdit
 from qtpy.QtGui import QFont
+from qtpy.QtCore import Qt
+
+from .textedit import WordHighlighter
 
 class LoggerWidget(QPlainTextEdit):
     def __init__(self, viewer:"napari.Viewer"):
         super().__init__(viewer.window._qt_window)
         self.setReadOnly(True)
         self.setMaximumBlockCount(500)
         self.setFont(QFont("Consolas"))
+
+        self.highlighter = WordHighlighter(self.document())
+        self.highlighter.appendRule(r"[a-zA-Z]+Warning", fcolor=Qt.yellow)
+        self.highlighter.appendRule(r"[a-zA-Z]+Error", fcolor=Qt.red)
 
     def appendPlainText(self, text:str):
         super().appendPlainText(text)
+        self.highlighter.setDocument(self.document())
         self.verticalScrollBar().setValue(self.verticalScrollBar().maximum())
         return None