def search_result(self):
p = self.result_text_edit.palette()
p.setColor(QPalette.Highlight, QColor("#ee799f"))
self.result_text_edit.setPalette(p)
search_settings = QTextDocument.FindFlags()
mod = QApplication.keyboardModifiers()
if (mod & Qt.ShiftModifier) != 0:
search_settings |= QTextDocument.FindBackward
r = self.result_text_edit.find(self.search_line.text(),
search_settings)
if not r:
if (mod & Qt.ShiftModifier) != 0:
self.goto(QTextCursor.End)
else:
self.goto(QTextCursor.Start)
self.result_text_edit.find(self.search_line.text(),
search_settings)
if self.search_line.text() == '':
self.search_summary_label.hide()
return
current_position = self.result_text_edit.textCursor().position()
try:
current_index = self.search_positions.index(current_position)
except ValueError:
current_index = -1
self.search_summary_label.show()
self.search_summary_label.setText(
'%s/%s' % (current_index + 1, len(self.search_positions)))
def getFindFlags(self):
flags = QTextDocument.FindFlags()
if self.ui.buttonCase.isChecked():
flags = flags | QTextDocument.FindCaseSensitively
if self.ui.buttonWords.isChecked():
flags = flags | QTextDocument.FindWholeWords
return flags
def find_all(
text: str, document: QTextDocument, flags=QTextDocument.FindFlags()
) -> List[QTextCursor]:
"""
Finds all occurrences of `text` in `document`, in order.
:param text: Text to find.
:param document: Document to search.
:param flags: Conditions to set on the search: none or (whole word and/or match case)
:return: Ordered list of all found instances.
"""
cursor = QTextCursor(document) # default pos == 0
found: List[QTextCursor] = []
while True:
cursor = document.find(text, cursor, flags)
if cursor.isNull():
return found
else:
found.append(cursor)
def __init__(self, plain_text_edit: QPlainTextEdit, parent=None):
"""
Sets up the dialog.
:param plain_text_edit: Text Editor to operate on.
:param parent: QWidget parent
"""
super().__init__(parent=parent, f=Qt.Tool)
self.plain_text_edit = plain_text_edit
self.cursors_needed = True
self.find_flags = QTextDocument.FindFlags()
self.found_cursors: List[QTextCursor] = []
self.current_cursor = QTextCursor()
# UI
layout = QVBoxLayout()
find_and_replace_layout = QGridLayout()
layout.addLayout(
find_and_replace_layout
) # if QGL is sub-layout, add to parent layout before doing stuff.
self.find_line_edit = QLineEdit()
find_label = QLabel("&Find")
find_label.setBuddy(self.find_line_edit)
options_layout = QHBoxLayout()
self.match_case_check_box = QCheckBox("Match Case")
self.whole_word_check_box = QCheckBox("Whole Word")
options_layout.addWidget(self.match_case_check_box)
options_layout.addWidget(self.whole_word_check_box)
options_layout.addStretch()
self.found_info_label = QLabel()
self.replace_line_edit = QLineEdit()
replace_label = QLabel("&Replace")
replace_label.setBuddy(self.replace_line_edit)
find_and_replace_layout.addWidget(find_label, 0, 0)
find_and_replace_layout.addWidget(self.find_line_edit, 0, 1)
find_and_replace_layout.addLayout(options_layout, 1, 1)
find_and_replace_layout.setRowMinimumHeight(2, 20)
find_and_replace_layout.addWidget(self.found_info_label, 2, 1)
find_and_replace_layout.addWidget(replace_label, 3, 0)
find_and_replace_layout.addWidget(self.replace_line_edit, 3, 1)
self.btn_box = QDialogButtonBox(QDialogButtonBox.Close)
self.find_next_btn = QPushButton("Next")
self.find_next_btn.setEnabled(False)
self.find_prev_btn = QPushButton("Previous")
self.find_prev_btn.setEnabled(False)
self.replace_btn = QPushButton("Replace")
self.replace_btn.setEnabled(False)
self.replace_all_btn = QPushButton("Replace All")
self.replace_all_btn.setEnabled(False)
self.btn_box.addButton(self.replace_btn, QDialogButtonBox.ActionRole)
self.btn_box.addButton(self.replace_all_btn,
QDialogButtonBox.ActionRole)
self.btn_box.addButton(self.find_prev_btn, QDialogButtonBox.ActionRole)
self.btn_box.addButton(self.find_next_btn, QDialogButtonBox.ActionRole)
layout.addWidget(self.btn_box)
self.setLayout(layout)
# End UI
self.btn_box.rejected.connect(self.reject)
self.find_line_edit.textEdited.connect(self.handle_text_edited)
self.find_next_btn.clicked.connect(self.next)
self.find_prev_btn.clicked.connect(self.prev)
self.replace_btn.clicked.connect(self.replace)
self.replace_all_btn.clicked.connect(self.replace_all)
self.plain_text_edit.document().contentsChanged.connect(
self.set_cursors_needed_true)
self.whole_word_check_box.stateChanged.connect(
self.toggle_whole_word_flag)
self.match_case_check_box.stateChanged.connect(
self.toggle_match_case_flag)
