使用QTreeView编辑字典

时间:2014-12-01 17:54:39

标签: python dictionary pyqt qtreeview

我有一个简化的代码示例,其中只用一个 Button 创建一个窗口。

enter image description here

按下它将弹出Qt对话框TestDialog,该对话框将Python字典作为参数。此词典显示在对话框内的可编辑QTreeView中。

enter image description here

更改某些值后,您可以单击确定取消以接受或放弃更改。关闭对话框后,我的目的是从主窗口中检索调用dialog.get_data()的修改后的字典,现在只返回原始的未修改字典。单击 Ok 按钮后,检索到的字典将打印到标准输出。

我的问题是,如何在修改树视图时修改字典?有没有办法自动附加要在每个项目修改上执行的功能?因此,在树视图中编辑时,例如float,那么相应的值将在字典中更新为float?

字典没有固定的大小,其上的类型可能会发生变化。类型列表是有限且已知的,对于此示例,可以缩减为{intstrfloatOther}。也可以假设父母不应该是可编辑的,并且子项只能在第二列中编辑,就像下面的示例中所示。

这是我的代码:

import sys
from PyQt4 import QtGui, QtCore, uic
from copy import deepcopy


class TestDialog(QtGui.QDialog):
    def __init__(self, data):

        super(TestDialog, self).__init__()

        self.data = deepcopy(data)

        # Layout
        btOk = QtGui.QPushButton("OK")
        btCancel = QtGui.QPushButton("Cancel")
        self.tree = QtGui.QTreeView()
        hbox = QtGui.QHBoxLayout()
        hbox.addStretch(1)
        hbox.addWidget(btOk)
        hbox.addWidget(btCancel)
        vbox = QtGui.QVBoxLayout()
        vbox.addLayout(hbox)
        vbox.addWidget(self.tree)
        self.setLayout(vbox)
        self.setGeometry(300, 300, 600, 400)

        # Button signals
        btCancel.clicked.connect(self.reject)
        btOk.clicked.connect(self.accept)

        # Tree view
        self.tree.setModel(QtGui.QStandardItemModel())
        self.tree.setAlternatingRowColors(True)
        self.tree.setSortingEnabled(True)
        self.tree.setHeaderHidden(False)
        self.tree.setSelectionBehavior(QtGui.QAbstractItemView.SelectItems)

        self.tree.model().setHorizontalHeaderLabels(['Parameter', 'Value'])

        for x in self.data:
            if not self.data[x]:
                continue
            parent = QtGui.QStandardItem(x)
            parent.setFlags(QtCore.Qt.NoItemFlags)
            for y in self.data[x]:
                value = self.data[x][y]
                child0 = QtGui.QStandardItem(y)
                child0.setFlags(QtCore.Qt.NoItemFlags |
                                QtCore.Qt.ItemIsEnabled)
                child1 = QtGui.QStandardItem(str(value))
                child1.setFlags(QtCore.Qt.ItemIsEnabled |
                                QtCore.Qt.ItemIsEditable |
                                ~ QtCore.Qt.ItemIsSelectable)
                parent.appendRow([child0, child1])
            self.tree.model().appendRow(parent)

        self.tree.expandAll()

    def get_data(self):
        return self.data

class Other(object):
    def __init__(self, x, y):
        self.x = x
        self.y = y
    def __repr__(self):
        return '(%s, %s)' % (self.x, self.y)

class Example(QtGui.QWidget):

    def __init__(self):

        super(Example, self).__init__()

        btn = QtGui.QPushButton('Button', self)
        btn.resize(btn.sizeHint())
        btn.clicked.connect(self.show_dialog)

        self.data = {}
        # This example will be hidden (has no parameter-value pair)
        self.data['example0'] = {}
        # A set example with an integer and a string parameters
        self.data['example1'] = {}
        self.data['example1']['int'] = 14
        self.data['example1']['str'] = 'asdf'
        # A set example with a float and other non-conventional type
        self.data['example2'] = {}
        self.data['example2']['float'] = 1.2
        self.data['example2']['other'] = Other(4, 8)

    def show_dialog(self):
        dialog = TestDialog(self.data)
        accepted = dialog.exec_()
        if not accepted:
            return
        self.data = deepcopy(dialog.get_data())
        print self.data


if __name__ == '__main__':

    app = QtGui.QApplication(sys.argv)
    ex = Example()
    ex.show()
    sys.exit(app.exec_())

1 个答案:

答案 0 :(得分:2)

您可以连接到模特的itemChanged信号:

    self.tree.model().itemChanged.connect(self.handleItemChanged)

处理程序看起来像这样:

    def handleItemChanged(self, item):
        parent = self.data[item.parent().text()]
        key = item.parent().child(item.row(), 0).text()
        parent[key] = type(parent[key])(item.text())

请注意,使用type转换值不一定适用于Other等自定义类。因此,您必须确保此类的构造函数可以转换字符串表示形式,或者在将字符串传递给构造函数之前将其解析为适当的参数。

另请注意,我并不打算在上面的示例代码中处理QString值。如果您使用Python 3,这不是问题,因为默认情况下它们会自动转换为Python字符串或从Python字符串转换。但是对于Python 2,您可以通过执行以下操作来切换此行为:

import sip
sip.setapi('QString', 2)
from PyQt4 import QtGui, QtCore, uic

有关详细信息,请参阅Selecting Incompatible APIs中的PyQt4 Docs