PyQt6基于QSqlTableModel替换QTableView状态列0/1为指定文本方案咨询

方案选择 #

• 无需从零重写QAbstractTableModel：QSqlTableModel本身已经封装了完善的数据库交互逻辑，从零实现反而需要额外处理数据查询、更新同步等逻辑，性价比极低
• 推荐两种实现方案：
  1. 直接继承QSqlTableModel重写data()方法，仅修改Status列的展示逻辑，原有代码改动最小
  2. 用你已经在尝试的QIdentityProxyModel代理模型，逻辑更解耦，不会侵入原有模型的数据库交互逻辑

方案1：继承QSqlTableModel实现（改动最小） #

你只需要自定义一个继承自QSqlTableModel的类，重写data方法处理Status列的展示映射即可，示例代码如下：

from PyQt5 import QtCore, QtSql

class CustomWorkerModel(QtSql.QSqlTableModel):
    def data(self, index, role=QtCore.Qt.ItemDataRole.DisplayRole):
        # 仅处理第4列（Status列）的展示角色
        if index.column() == 4 and role == QtCore.Qt.ItemDataRole.DisplayRole:
            # 拿原始存储值，避免展示值干扰
            raw_val = super().data(index, QtCore.Qt.ItemDataRole.EditRole)
            if str(raw_val) in ("1", "true", "enabled"):
                return "Working"
            elif str(raw_val) in ("0", "false", "disabled"):
                return "Not working"
        # 其余场景全部走原有逻辑
        return super().data(index, role)

原有代码仅需要修改模型初始化部分即可，其余逻辑完全不用改：

# 把原来的QtSql.QSqlTableModel换成自定义类
model = CustomWorkerModel()
model.setTable('worker')
model.select()
# 表头设置、视图绑定逻辑不变

方案2：修正你的代理模型，解决DataMapper无法保存问题 #

你写的代理模型本身思路是对的，问题出在代码错误、缺失编辑逻辑，以及DataMapper绑定方式不对：

修正后的代理模型代码 #

class MyProxyModel(QtCore.QIdentityProxyModel):
    def data(self, index, role=QtCore.Qt.ItemDataRole.DisplayRole):
        if index.column() == 4 and role == QtCore.Qt.ItemDataRole.DisplayRole:
            raw_val = super().data(index, QtCore.Qt.ItemDataRole.EditRole)
            raw_str = str(raw_val).lower()
            if raw_str in ("1", "true", "enabled"):
                return "Working"
            elif raw_str in ("0", "false", "disabled"):
                return "Not working"
        return super().data(index, role)

    # 必须重写setData，把用户输入的展示文本转成数据库可识别的0/1值
    def setData(self, index, value, role=QtCore.Qt.ItemDataRole.EditRole):
        if index.column() == 4 and role == QtCore.Qt.ItemDataRole.EditRole:
            input_str = str(value).lower()
            if input_str in ("working", "works", "1", "true", "enabled"):
                return super().setData(index, 1, role)
            elif input_str in ("not working", "doesn't work", "0", "false", "disabled"):
                return super().setData(index, 0, role)
        return super().setData(index, value, role)

DataMapper无法保存的修复要点 #

• 你原来的代码有两处低级错误：一是column == 4::多写了一个冒号，二是desabled拼写错误应为disabled，这两个问题会直接导致逻辑异常
• QDataMapper必须绑定代理模型，不能直接绑定原始的QSqlTableModel
• 确保QSqlTableModel设置了正确的编辑策略，比如model.setEditStrategy(QtSql.QSqlTableModel.OnRowChange)，或者提交数据前手动调用model.submitAll()同步到数据库

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