From de452f9c32a69732f931665b23ad32ead6e1240c Mon Sep 17 00:00:00 2001
From: Irony <892768447@qq.com>
Date: Thu, 28 May 2020 15:26:28 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=AB=98=E4=BA=AE=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 QTextEdit/HighlightText.py | 37 +++++++++++++++++++++++++++----------
 1 file changed, 27 insertions(+), 10 deletions(-)

diff --git a/QTextEdit/HighlightText.py b/QTextEdit/HighlightText.py
index d1c8c48..092554a 100644
--- a/QTextEdit/HighlightText.py
+++ b/QTextEdit/HighlightText.py
@@ -1,4 +1,5 @@
 import sys
+from PyQt5.QtCore import QRegExp
 from PyQt5.QtGui import QTextCharFormat, QTextDocument, QTextCursor
 from PyQt5.QtWidgets import (QApplication, QMainWindow, QTextEdit,
                              QToolBar, QLineEdit, QPushButton, QColorDialog, QHBoxLayout, QWidget)
@@ -22,30 +23,46 @@ class TextEdit(QMainWindow):
 
         tb = QToolBar(self)
         tb.addWidget(widget)
+        self.addToolBar(tb)
 
     def setText(self, text):
         self.textEdit.setPlainText(text)
 
-    def mergeFormatOnWordOrSelection(self, format):
-        cursor = self.textEdit.textCursor()
-        if not cursor.hasSelection():
-            cursor.select(QTextCursor.WordUnderCursor)
-        cursor.mergeCharFormat(format)
-        self.textEdit.mergeCurrentCharFormat(format)
-
     def highlight(self):
         text = self.findText.text()  # 输入框中的文字
         if not text:
             return
+
         col = QColorDialog.getColor(self.textEdit.textColor(), self)
         if not col.isValid():
             return
+
+        # 恢复默认的颜色
+        cursor = self.textEdit.textCursor()
+        cursor.select(QTextCursor.Document)
+        cursor.setCharFormat(QTextCharFormat())
+        cursor.clearSelection()
+        self.textEdit.setTextCursor(cursor)
+
+        # 文字颜色
         fmt = QTextCharFormat()
         fmt.setForeground(col)
-        # 先把光标移动到开头
+
+        # 正则
+        expression = QRegExp(text)
         self.textEdit.moveCursor(QTextCursor.Start)
-        while self.textEdit.find(text, QTextDocument.FindWholeWords):  # 查找所有文字
-            self.mergeFormatOnWordOrSelection(fmt)
+        cursor = self.textEdit.textCursor()
+
+        # 循环查找设置颜色
+        pos = 0
+        index = expression.indexIn(self.textEdit.toPlainText(), pos)
+        while index >= 0:
+            cursor.setPosition(index)
+            cursor.movePosition(QTextCursor.Right,
+                                QTextCursor.KeepAnchor, len(text))
+            cursor.mergeCharFormat(fmt)
+            pos = index + expression.matchedLength()
+            index = expression.indexIn(self.textEdit.toPlainText(), pos)
 
 
 if __name__ == '__main__':