在 PyQt5 中实现一个带有加号功能的 QTreeWidget
,可以通过自定义一个 QTreeWidgetItem
并在其上添加加号按钮来实现,用户点击加号时会在当前节点下添加子节点。下面是一个实现示例:
代码示例
import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QTreeWidget, QTreeWidgetItem, QVBoxLayout, QPushButton, QWidget
from PyQt5.QtCore import Qt
class TreeWidget(QTreeWidget):
def __init__(self):
super().__init__()
self.setColumnCount(1)
self.setHeaderLabels(['Node'])
def add_tree_item(self, parent_item=None):
# 创建新的子项
new_item = QTreeWidgetItem(parent_item if parent_item else self)
new_item.setText(0, 'New Item')
# 在新项下添加加号
add_button = QPushButton('+')
add_button.clicked.connect(lambda: self.add_tree_item(new_item))
self.setItemWidget(new_item, 0, add_button)
new_item.setExpanded(True)
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.setWindowTitle("QTreeWidget Example")
self.setGeometry(100, 100, 400, 300)
self.tree_widget = TreeWidget()
root_item = QTreeWidgetItem(self.tree_widget)
root_item.setText(0, 'Root')
self.tree_widget.add_tree_item(root_item)
layout = QVBoxLayout()
layout.addWidget(self.tree_widget)
container = QWidget()
container.setLayout(layout)
self.setCentralWidget(container)
if __name__ == "__main__":
app = QApplication(sys.argv)
window = MainWindow()
window.show()
sys.exit(app.exec_())
解释:
-
TreeWidget 类:继承
QTreeWidget
,定义了add_tree_item
方法,该方法用来向树中添加新的QTreeWidgetItem
,并在该项上添加一个加号按钮。点击加号会触发该项的事件,向该项添加子项。 -
MainWindow 类:主窗口包含了一个
QTreeWidget
(即TreeWidget
),并且通过调用add_tree_item
方法在树上添加子项。 -
QPushButton:用来作为“加号”按钮。当点击时,会添加一个子节点。