QTreeView 复选框

Question

我知道这个问题已经被问过很多次了，但我找不到相关的答案。

我想问一下，使用Qt自带的simpletreemodel教程，我该怎么添加复选框呢？

Answer 1

这里有一个完整的PyQt示例，使用了QStandardItemModel：

model = QStandardItemModel()
parent_item = model.invisibleRootItem()  # type: QStandardItem
for row in [
    (Qt.Unchecked, 'unchecked'),
    (Qt.PartiallyChecked, 'partially'),
    (Qt.Checked, 'checked')
]:
    checked, text = row
    check_item = QStandardItem('')
    check_item.setCheckable(True)
    check_item.setCheckState(checked)
    parent_item.appendRow([check_item, QStandardItem(text)])
treeview.setModel(model)

顺便说一下，这个也适用于任何C++应用程序。

Answer 2

我把上面的内容转换成了PyQt，主要是为了自己的需要，觉得可以分享给大家。

def data(self, index, role):
    if not index.isValid():
        return None

    item = index.internalPointer();

    if role == Qt.CheckStateRole and index.column() == self.check_col:
        return int( Qt.Checked if item.isChecked() else Qt.Unchecked )

    return super(TreeModel, self).data(index, role)


def flags(self, index):
    if not index.isValid():
        return None

    if index.column() == self.check_col:
        flags = Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
    else:
        flags = super(TreeModel, self).flags(index)

    return flags


def setData(self, index, value, role=Qt.EditRole):
    if index.column() == self.check_col:
        if role == Qt.EditRole:
            return False
        if role == Qt.CheckStateRole:
            item = self.getItem(index)
            item.setChecked(value)
            self.dataChanged.emit(index, index)
            return True

    return super(TreeModel, self).setData(index, value, role)

Answer 3

首先，你需要修改TreeItem，让它能够记录每个项目的选中状态：

private:
    ...
    bool checked;

还需要添加一个设置器和一个获取器：

bool isChecked() const { return checked; }
void setChecked( bool set ) { checked = set; }

接下来，模型也需要进行修改，这样视图才能知道哪些项目是被选中的：

QVariant TreeModel::data(const QModelIndex &index, int role) const
{
    if (!index.isValid())
        return QVariant();

    TreeItem *item = static_cast<TreeItem*>(index.internalPointer());

    if ( role == Qt::CheckStateRole && index.column() == 0 )
        return static_cast< int >( item->isChecked() ? Qt::Checked : Qt::Unchecked );

    if (role != Qt::DisplayRole)
        return QVariant();

    return item->data(index.column());
}

并且要修改模型的标志方法，让视图知道这个模型包含可以被选中的项目：

Qt::ItemFlags TreeModel::flags(const QModelIndex &index) const
{
    if (!index.isValid())
        return 0;

    Qt::ItemFlags flags = Qt::ItemIsEnabled | Qt::ItemIsSelectable;

    if ( index.column() == 0 )
        flags |= Qt::ItemIsUserCheckable;

    return flags;
}

我觉得这样就可以了。如果你想在用户勾选和取消勾选项目时更新TreeItem的选中状态，那么你需要在你的TreeModel中提供QAbstractItemModel::setData这个方法。