class Example(QMainWindow):
def __init__(self):
super().__init__()
self.initUI()
def initUI(self):
self.textEdit = QTextEdit()
self.setCentralWidget(self.textEdit)
self.statusBar()
openFile = QAction(QIcon('open.png'), 'Open', self)
openFile.setShortcut('Ctrl+O')
openFile.setStatusTip('Open new File')
openFile.triggered.connect(self.showDialog)
menubar = self.menuBar()
fileMenu = menubar.addMenu('&File')
fileMenu.addAction(openFile)
self.setGeometry(300, 300, 550, 450)
self.setWindowTitle('File dialog')
self.show()
def showDialog(self):
home_dir = str(Path.home())
fname = QFileDialog.getOpenFileName(self, 'Open file', home_dir)
if fname[0]:
f = open(fname[0], 'r')
with f:
data = f.read()
self.textEdit.setText(data)
class FileDialogDemo(QWidget):
def __init__(self, parent=None):
super(FileDialogDemo, self).__init__(parent)
layout = QVBoxLayout()
self.button = QPushButton("加载图片")
self.button.clicked.connect(self.get_file)
layout.addWidget(self.button)
self.le = QLabel("")
layout.addWidget(self.le)
self.button1 = QPushButton("加载文本文件")
self.button1.clicked.connect(self.get_files)
layout.addWidget(self.button1)
self.contents = QTextEdit()
layout.addWidget(self.contents)
self.setLayout(layout)
self.setWindowTitle("File Dialog 实例")
def get_file(self):
fname, _ = QFileDialog.getOpenFileName(self, "Open file", 'C:\\', 'Image files (*.jpg &.gif)')
self.le.setPixmap(QPixmap(fname))
def get_files(self):
dialog = QFileDialog()
dialog.setFileMode(QFileDialog.FileMode.AnyFile)
dialog.setFilter(QDir.Filter.Files)
if dialog.exec():
filenames = dialog.selectedFiles()
f = open(filenames[0], 'r')
with f:
data = f.read()
self.contents.setText(data)
class MyApp(QWidget):
def __init__(self):
super().__init__()
self.setWindowTitle('my app')
self.setWindowIcon(QIcon('python.ico'))
self.resize(500, 350) # x, y
layout = QVBoxLayout()
self.setLayout(layout)
# widgets
self.inputField = QLineEdit()
button = QPushButton('&say hello')
button.clicked.connect(self.say_hello)
self.output = QTextEdit()
layout.addWidget(self.inputField)
layout.addWidget(button)
layout.addWidget(self.output)
def say_hello(self):
input_text = self.inputField.text()
self.output.setText(f'Hello {input_text}')
class DocumentWidget(QWidget):
def __init__(self, interactive_matching_widget):
super(DocumentWidget, self).__init__(interactive_matching_widget)
self.interactive_matching_widget = interactive_matching_widget
self.layout = QVBoxLayout()
self.layout.setContentsMargins(0, 0, 0, 0)
self.layout.setSpacing(10)
self.setLayout(self.layout)
self.document = None
self.current_nugget = None
self.base_formatted_text = ""
self.idx_mapper = {}
self.nuggets_in_order = []
self.text_edit = QTextEdit()
self.layout.addWidget(self.text_edit)
self.text_edit.setReadOnly(True)
self.text_edit.setFrameStyle(0)
self.text_edit.setFont(CODE_FONT)
self.text_edit.setHorizontalScrollBarPolicy(
Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
self.text_edit.setText("")
self.buttons_widget = QWidget()
self.buttons_widget_layout = QHBoxLayout(self.buttons_widget)
self.layout.addWidget(self.buttons_widget)
self.left_button = QPushButton("Skip Left")
self.left_button.setFont(BUTTON_FONT)
self.left_button.clicked.connect(self._left_button_clicked)
self.buttons_widget_layout.addWidget(self.left_button)
self.right_button = QPushButton("Skip Right")
self.right_button.setFont(BUTTON_FONT)
self.right_button.clicked.connect(self._right_button_clicked)
self.buttons_widget_layout.addWidget(self.right_button)
self.match_button = QPushButton("Confirm Match")
self.match_button.setFont(BUTTON_FONT)
self.match_button.clicked.connect(self._match_button_clicked)
self.buttons_widget_layout.addWidget(self.match_button)
self.no_match_button = QPushButton("There Is No Match")
self.no_match_button.setFont(BUTTON_FONT)
self.no_match_button.clicked.connect(self._no_match_button_clicked)
self.buttons_widget_layout.addWidget(self.no_match_button)
def _left_button_clicked(self):
idx = self.nuggets_in_order.index(self.current_nugget)
if idx > 0:
self.current_nugget = self.nuggets_in_order[idx - 1]
self._highlight_current_nugget()
def _right_button_clicked(self):
idx = self.nuggets_in_order.index(self.current_nugget)
if idx < len(self.nuggets_in_order) - 1:
self.current_nugget = self.nuggets_in_order[idx + 1]
self._highlight_current_nugget()
def _match_button_clicked(self):
self.interactive_matching_widget.main_window.give_feedback_task({
"message":
"is-match",
"nugget":
self.current_nugget
})
def _no_match_button_clicked(self):
self.interactive_matching_widget.main_window.give_feedback_task({
"message":
"no-match-in-document",
"nugget":
self.current_nugget
})
def _highlight_current_nugget(self):
mapped_start_char = self.idx_mapper[self.current_nugget.start_char]
mapped_end_char = self.idx_mapper[self.current_nugget.end_char]
formatted_text = (
f"{self.base_formatted_text[:mapped_start_char]}"
f"<span style='background-color: #FFFF00'><b>"
f"{self.base_formatted_text[mapped_start_char:mapped_end_char]}</span></b>"
f"{self.base_formatted_text[mapped_end_char:]}")
self.text_edit.setText("")
self.text_edit.textCursor().insertHtml(formatted_text)
def update_document(self, nugget):
self.document = nugget.document
self.current_nugget = nugget
self.nuggets_in_order = list(
sorted(self.document.nuggets, key=lambda x: x.start_char))
if self.nuggets_in_order != []:
self.idx_mapper = {}
char_list = []
end_chars = []
next_start_char = 0
next_nugget_idx = 0
for idx, char in enumerate(list(self.document.text)):
if idx == next_start_char:
if end_chars == []:
char_list += list("<b>")
end_chars.append(
self.nuggets_in_order[next_nugget_idx].end_char)
next_nugget_idx += 1
if next_nugget_idx < len(self.nuggets_in_order):
next_start_char = self.nuggets_in_order[
next_nugget_idx].start_char
else:
next_start_char = -1
while idx in end_chars:
end_chars.remove(idx)
if end_chars == []:
char_list += list("</b>")
self.idx_mapper[idx] = len(char_list)
char_list.append(char)
self.base_formatted_text = "".join(char_list)
else:
self.idx_mapper = {}
for idx in range(len(self.document.text)):
self.idx_mapper[idx] = idx
self.base_formatted_text = ""
self._highlight_current_nugget()
scroll_cursor = QTextCursor(self.text_edit.document())
scroll_cursor.setPosition(nugget.start_char)
self.text_edit.setTextCursor(scroll_cursor)
self.text_edit.ensureCursorVisible()
def enable_input(self):
self.left_button.setEnabled(True)
self.right_button.setEnabled(True)
self.match_button.setEnabled(True)
self.no_match_button.setEnabled(True)
def disable_input(self):
self.left_button.setDisabled(True)
self.right_button.setDisabled(True)
self.match_button.setDisabled(True)
self.no_match_button.setDisabled(True)
class NuggetListItemWidget(CustomScrollableListItem):
def __init__(self, nugget_list_widget):
super(NuggetListItemWidget, self).__init__(nugget_list_widget)
self.nugget_list_widget = nugget_list_widget
self.nugget = None
self.setFixedHeight(40)
self.setStyleSheet("background-color: white")
self.layout = QHBoxLayout(self)
self.layout.setContentsMargins(20, 0, 20, 0)
self.layout.setSpacing(10)
self.info_label = QLabel()
self.info_label.setFont(CODE_FONT_BOLD)
self.layout.addWidget(self.info_label)
self.left_split_label = QLabel("|")
self.left_split_label.setFont(CODE_FONT_BOLD)
self.layout.addWidget(self.left_split_label)
self.text_edit = QTextEdit()
self.text_edit.setReadOnly(True)
self.text_edit.setFrameStyle(0)
self.text_edit.setFont(CODE_FONT)
self.text_edit.setLineWrapMode(QTextEdit.LineWrapMode.FixedPixelWidth)
self.text_edit.setLineWrapColumnOrWidth(10000)
self.text_edit.setHorizontalScrollBarPolicy(
Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
self.text_edit.setVerticalScrollBarPolicy(
Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
self.text_edit.setFixedHeight(30)
self.text_edit.setText("")
self.layout.addWidget(self.text_edit)
self.right_split_label = QLabel("|")
self.right_split_label.setFont(CODE_FONT_BOLD)
self.layout.addWidget(self.right_split_label)
self.match_button = QPushButton()
self.match_button.setIcon(QIcon("aset_ui/resources/correct.svg"))
self.match_button.setFlat(True)
self.match_button.clicked.connect(self._match_button_clicked)
self.layout.addWidget(self.match_button)
self.fix_button = QPushButton()
self.fix_button.setIcon(QIcon("aset_ui/resources/incorrect.svg"))
self.fix_button.setFlat(True)
self.fix_button.clicked.connect(self._fix_button_clicked)
self.layout.addWidget(self.fix_button)
def update_item(self, item, params=None):
self.nugget = item
sentence = self.nugget[CachedContextSentenceSignal]["text"]
start_char = self.nugget[CachedContextSentenceSignal]["start_char"]
end_char = self.nugget[CachedContextSentenceSignal]["end_char"]
self.text_edit.setText("")
formatted_text = (
f"{' ' * (params - start_char)}{sentence[:start_char]}"
f"<span style='background-color: #FFFF00'><b>{sentence[start_char:end_char]}</b></span>"
f"{sentence[end_char:]}{' ' * 50}")
self.text_edit.textCursor().insertHtml(formatted_text)
scroll_cursor = QTextCursor(self.text_edit.document())
scroll_cursor.setPosition(params + 50)
self.text_edit.setTextCursor(scroll_cursor)
self.text_edit.ensureCursorVisible()
self.info_label.setText(
f"{str(round(self.nugget[CachedDistanceSignal], 2)).ljust(4)}")
def _match_button_clicked(self):
self.nugget_list_widget.interactive_matching_widget.main_window.give_feedback_task(
{
"message": "is-match",
"nugget": self.nugget
})
def _fix_button_clicked(self):
self.nugget_list_widget.interactive_matching_widget.get_document_feedback(
self.nugget)
def enable_input(self):
self.match_button.setEnabled(True)
self.fix_button.setEnabled(True)
def disable_input(self):
self.match_button.setDisabled(True)
self.fix_button.setDisabled(True)
class RenameDialog(QDialog):
out = pyqtSignal(object)
def __init__(self, parent=None):
super(RenameDialog, self).__init__(parent)
self.infos = []
self.min_width = 400
self.initUI()
self.update_text()
self.setStyleSheet(dialog_qss_style)
def set_values(self, infos=None):
self.infos = infos or []
self.update_text() # 更新界面
def initUI(self):
self.setWindowIcon(QIcon(SRC_DIR + "desc.ico"))
self.lb_name = QLabel()
self.lb_name.setText("文件夹名：")
self.lb_name.setAlignment(Qt.AlignmentFlag.AlignRight
| Qt.AlignmentFlag.AlignTrailing
| Qt.AlignmentFlag.AlignVCenter)
self.tx_name = QLineEdit()
self.lb_desc = QLabel()
self.tx_desc = QTextEdit()
self.lb_desc.setText("描 述：")
self.lb_desc.setAlignment(Qt.AlignmentFlag.AlignRight
| Qt.AlignmentFlag.AlignTrailing
| Qt.AlignmentFlag.AlignVCenter)
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Orientation.Horizontal)
self.buttonBox.setStandardButtons(
QDialogButtonBox.StandardButton.Ok
| QDialogButtonBox.StandardButton.Cancel)
self.buttonBox.button(QDialogButtonBox.StandardButton.Ok).setText("确定")
self.buttonBox.button(
QDialogButtonBox.StandardButton.Cancel).setText("取消")
self.grid = QGridLayout()
self.grid.setSpacing(10)
self.grid.addWidget(self.lb_name, 1, 0)
self.grid.addWidget(self.tx_name, 1, 1)
self.grid.addWidget(self.lb_desc, 2, 0)
self.grid.addWidget(self.tx_desc, 2, 1, 5, 1)
self.grid.addWidget(self.buttonBox, 7, 1, 1, 1)
self.setLayout(self.grid)
self.buttonBox.accepted.connect(self.btn_ok)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
def update_text(self):
self.tx_desc.setFocus()
num = len(self.infos)
if num == 1:
self.lb_name.setVisible(True)
self.tx_name.setVisible(True)
infos = self.infos[0]
self.buttonBox.button(
QDialogButtonBox.StandardButton.Ok).setToolTip("") # 去除新建文件夹影响
self.buttonBox.button(
QDialogButtonBox.StandardButton.Ok).setEnabled(
True) # 去除新建文件夹影响
self.setWindowTitle("修改文件夹名与描述")
self.tx_name.setText(str(infos.name))
if infos.desc:
self.tx_desc.setText(str(infos.desc))
self.tx_desc.setToolTip('原描述：' + str(infos.desc))
else:
self.tx_desc.setText("")
self.tx_desc.setToolTip('')
self.tx_desc.setPlaceholderText("无")
self.min_width = len(str(infos.name)) * 8
if infos.is_file:
self.setWindowTitle("修改文件描述")
self.tx_name.setFocusPolicy(Qt.FocusPolicy.NoFocus)
self.tx_name.setReadOnly(True)
else:
self.tx_name.setFocusPolicy(Qt.FocusPolicy.StrongFocus)
self.tx_name.setReadOnly(False)
self.tx_name.setFocus()
elif num > 1:
self.lb_name.setVisible(False)
self.tx_name.setVisible(False)
self.setWindowTitle(f"批量修改{num}个文件(夹)的描述")
self.tx_desc.setText('')
self.tx_desc.setPlaceholderText("建议160字数以内。")
else:
self.setWindowTitle("新建文件夹")
self.tx_name.setText("")
self.buttonBox.button(
QDialogButtonBox.StandardButton.Ok).setEnabled(False)
self.buttonBox.button(
QDialogButtonBox.StandardButton.Ok).setToolTip("请先输入文件名！")
self.tx_name.textChanged.connect(self.slot_new_ok_btn)
self.tx_name.setPlaceholderText("不支持空格，如有会被自动替换成 _")
self.tx_name.setFocusPolicy(Qt.FocusPolicy.StrongFocus)
self.tx_name.setReadOnly(False)
self.tx_desc.setPlaceholderText("可选项，建议160字数以内。")
self.tx_name.setFocus()
if self.min_width < 400:
self.min_width = 400
self.resize(self.min_width, 200)
def slot_new_ok_btn(self):
"""新建文件夹槽函数"""
self.buttonBox.button(
QDialogButtonBox.StandardButton.Ok).setEnabled(True)
self.buttonBox.button(
QDialogButtonBox.StandardButton.Ok).setToolTip("")
def btn_ok(self):
new_name = self.tx_name.text()
new_des = self.tx_desc.toPlainText()
info_len = len(self.infos)
if info_len == 0: # 在 work_id 新建文件夹
if new_name:
self.out.emit(("new", new_name, new_des))
elif info_len == 1:
if new_name != self.infos[0].name or new_des != self.infos[0].desc:
self.infos[0].new_des = new_des
self.infos[0].new_name = new_name
self.out.emit(("change", self.infos))
else:
if new_des:
for infos in self.infos:
infos.new_des = new_des
self.out.emit(("change", self.infos))
