python - 如何将 Qt5 Tree 的引用返回到一组数据？

Question

我正在尝试使用 TreeView 或 TreeWidget 创建一个仅显示图标/字符串来表示路径的层次结构。

我想要一个信号/槽，以便双击某个项目会打开一个新窗口来编辑该路径的内容。目前，我可以通过索引查找显示名称，但看不到任何存储/链接隐藏数据的方法(例如引用唯一的folder_id或node_id的键)

常见的范例是将 key 添加到模型，然后从树显示中删除这些列吗？

需要访问隐藏属性的示例数据/原因。

• 类:存储库属性:ID、名称

• 类别:端点属性:ID、名称、Repository_ID、地址、方法等...

• 类:GenerateDiagramTree属性:类型、Mixed_ID、名称

我只想在 View 中看到名称，但我希望能够引用类型/ID 以确定双击时会发生什么。

Answer 1

当您想要为每个节点保存自定义信息时，您必须使用角色(最好 >= Qt::UserRole，因为角色可以由 Qt 内部使用)。例如，下面的代码中使用2个角色来存储2种信息。

如果使用QTreeWidget，则必须使用setData()方法，并传递角色和值来存储信息，要获取信息，必须使用data() 通过传递角色的方法。

from enum import Enum
import uuid

from PyQt5 import QtCore, QtWidgets


TypeRole = QtCore.Qt.UserRole + 1000
IdRole = QtCore.Qt.UserRole + 1001


class TypeItem(Enum):
    ROOT = 0
    CHILD = 1


class Dialog(QtWidgets.QDialog):
    def __init__(self, name, type_, id_, parent=None):
        super().__init__(parent)

        self.name_le = QtWidgets.QLineEdit(name)
        type_label = QtWidgets.QLabel(str(type_))
        self.id_le = QtWidgets.QLineEdit(id_)
        button_box = QtWidgets.QDialogButtonBox()
        button_box.setOrientation(QtCore.Qt.Horizontal)
        button_box.setStandardButtons(
            QtWidgets.QDialogButtonBox.Cancel | QtWidgets.QDialogButtonBox.Ok
        )
        lay = QtWidgets.QVBoxLayout(self)
        lay.addWidget(self.name_le)
        lay.addWidget(type_label)
        lay.addWidget(self.id_le)
        lay.addWidget(button_box)

        button_box.accepted.connect(self.accept)
        button_box.rejected.connect(self.reject)

    @property
    def name(self):
        return self.name_le.text()

    @property
    def id_(self):
        return self.id_le.text()


class MainWindow(QtWidgets.QMainWindow):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.view = QtWidgets.QTreeWidget()
        self.view.itemDoubleClicked.connect(self.onItemDoubleClicked)
        self.setCentralWidget(self.view)

        for i in range(3):
            root_it = QtWidgets.QTreeWidgetItem(["tlv-{}".format(i)])
            root_it.setData(0, TypeRole, TypeItem.ROOT)
            root_it.setData(0, IdRole, uuid.uuid4().hex)
            self.view.addTopLevelItem(root_it)
            for j in range(3):
                child_it = QtWidgets.QTreeWidgetItem(["it-{}{}".format(i, j)])
                child_it.setData(0, TypeRole, TypeItem.CHILD)
                child_it.setData(0, IdRole, uuid.uuid4().hex)
                root_it.addChild(child_it)

    @QtCore.pyqtSlot(QtWidgets.QTreeWidgetItem, int)
    def onItemDoubleClicked(self, item, column):
        name = item.text(column)
        type_ = item.data(column, TypeRole)
        id_ = item.data(column, IdRole)
        d = Dialog(name, type_, id_)
        if d.exec_() == QtWidgets.QDialog.Accepted:
            item.setText(column, d.name)
            item.setData(column, IdRole, d.id_)


if __name__ == "__main__":
    import sys

    app = QtWidgets.QApplication(sys.argv)
    w = MainWindow()
    w.show()
    sys.exit(app.exec_())

如果您使用 QTreeView，则必须在模型中实现该角色的处理，对于以下示例，使用 QStandardItemModel，其中每个 QStandardItem 都有 setData() 方法和 data() 来存储和检索信息。

from enum import Enum
import uuid

from PyQt5 import QtCore, QtGui, QtWidgets


TypeRole = QtCore.Qt.UserRole + 1000
IdRole = QtCore.Qt.UserRole + 1001


class TypeItem(Enum):
    ROOT = 0
    CHILD = 1


class Dialog(QtWidgets.QDialog):
    def __init__(self, name, type_, id_, parent=None):
        super().__init__(parent)

        self.name_le = QtWidgets.QLineEdit(name)
        type_label = QtWidgets.QLabel(str(type_))
        self.id_le = QtWidgets.QLineEdit(id_)
        button_box = QtWidgets.QDialogButtonBox()
        button_box.setOrientation(QtCore.Qt.Horizontal)
        button_box.setStandardButtons(
            QtWidgets.QDialogButtonBox.Cancel | QtWidgets.QDialogButtonBox.Ok
        )
        lay = QtWidgets.QVBoxLayout(self)
        lay.addWidget(self.name_le)
        lay.addWidget(type_label)
        lay.addWidget(self.id_le)
        lay.addWidget(button_box)

        button_box.accepted.connect(self.accept)
        button_box.rejected.connect(self.reject)

    @property
    def name(self):
        return self.name_le.text()

    @property
    def id_(self):
        return self.id_le.text()


class MainWindow(QtWidgets.QMainWindow):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.model = QtGui.QStandardItemModel(self)
        self.view = QtWidgets.QTreeView()
        self.view.setModel(self.model)
        self.view.doubleClicked.connect(self.onDoubleClicked)
        self.setCentralWidget(self.view)

        for i in range(3):
            root_it = QtGui.QStandardItem("tlv-{}".format(i))
            root_it.setData(TypeItem.ROOT, TypeRole)
            root_it.setData(uuid.uuid4().hex, IdRole)
            self.model.appendRow(root_it)
            for j in range(3):
                child_it = QtGui.QStandardItem("it-{}{}".format(i, j))
                child_it.setData(TypeItem.CHILD, TypeRole)
                child_it.setData(uuid.uuid4().hex, IdRole)
                root_it.appendRow(child_it)

    @QtCore.pyqtSlot(QtCore.QModelIndex)
    def onDoubleClicked(self, index):
        item = self.model.itemFromIndex(index)
        name = item.text()
        type_ = item.data(TypeRole)
        id_ = item.data(IdRole)
        d = Dialog(name, type_, id_)
        if d.exec_() == QtWidgets.QDialog.Accepted:
            item.setText(d.name)
            item.setData(d.id_, IdRole)


if __name__ == "__main__":
    import sys

    app = QtWidgets.QApplication(sys.argv)
    w = MainWindow()
    w.show()
    sys.exit(app.exec_())