PyQt6 QSqlTableModel如何同时显示数据库数据与单元格复选框？

要让每个单元格同时展示数据库原始数据和复选框，核心是在重写data()方法时，正确处理DisplayRole和CheckStateRole两个角色：

1. 保留复选框的状态逻辑，确保控件正常显示；
2. 修复数据显示逻辑，替换固定返回值为数据库真实数据。

修改后的MyModel核心代码 #

from PyQt6.QtSql import QSqlTableModel
from PyQt6.QtCore import Qt

class MyModel(QSqlTableModel):
    def data(self, index, role=Qt.ItemDataRole.DisplayRole):
        # 处理复选框状态
        if role == Qt.ItemDataRole.CheckStateRole:
            # 这里根据业务逻辑返回选中状态，示例为固定未选中
            # 可替换为从数据库字段/本地存储读取的状态值
            return Qt.CheckState.Unchecked.value
        
        # 处理数据显示
        if role == Qt.ItemDataRole.DisplayRole:
            # 保留特定单元格的自定义内容（按需修改判断条件）
            if index.row() == 0 and index.column() == 0:
                return "特定单元格自定义内容"
            # 其他单元格调用父类方法获取数据库真实数据
            return super().data(index, role)
        
        # 其他角色默认交给父类处理
        return super().data(index, role)

关键说明 #

• 复选框逻辑：CheckStateRole的返回值决定复选框状态，可根据业务需求从数据库字段或本地状态存储中读取，示例仅为固定值参考。
• 数据显示修复：移除返回固定值“SOMETHING”的代码，调用super().data(index, role)即可直接获取QSqlTableModel从数据库加载的原始数据。
• 特定单元格处理：如果需要给部分单元格设置自定义内容，保留对应的条件判断即可，其余单元格自动展示数据库数据。

可选：让复选框支持交互 #

如果需要用户点击复选框切换状态，还需重写flags()和setData()方法：

def flags(self, index):
        # 开启单元格的勾选、编辑权限
        return super().flags(index) | Qt.ItemFlag.ItemIsUserCheckable | Qt.ItemFlag.ItemIsEnabled

    def setData(self, index, value, role=Qt.ItemDataRole.EditRole):
        if role == Qt.ItemDataRole.CheckStateRole:
            # 处理状态变更逻辑，例如写入数据库或更新本地存储
            print(f"行{index.row()}列{index.column()}状态变更为: {value}")
            self.dataChanged.emit(index, index)
            return True
        return super().setData(index, value, role)

内容的提问来源于stack exchange，提问作者Benny Lutin