使用 QSqlTableModel 在 QListView 中的 CheckBox答案

【问题标题】：CheckBox in QListView using QSqlTableModel使用 QSqlTableModel 在 QListView 中的 CheckBox
【发布时间】：2018-06-19 23:19:25
【问题描述】：

我正在学习 PyQt5 框架并尝试构建项目的 QListView，每个项目前面都有一个复选框。我读过的大多数示例都没有显示如何使用 QSqlTableModel 完成此操作。有人可以指出我正确的方向吗？

db = QSqlDatabase.addDatabase("QSQLITE")
db.setDatabaseName("./Data/mydbase.db")
db.open()
model = QSqlTableModel()
model.setTable("labdata")
model.setSort(1,0)
model.select()
while model.canFetchMore():
    model.fetchMore()
self.ui.lst_LabData.setModel(model)
self.ui.lst_LabData.setModelColumn(1)
self.ui.lst_LabData.show()

【问题讨论】：

QListView 中显示的字段为布尔值，如果是，则希望在选中 QCheckBox 时更改它。
选中复选框后，我确实需要采取措施。视图中显示的该字段不是布尔值。
选中复选框后，我想在另一个小部件中显示记录的其他字段。不过，还没有走到那一步。只是试图获取列表视图上的复选框
我不明白这个问题。将行发送到哪里？
我想显示从 SQL 表中读取的名称列表，名称前面有一个复选框。选中该框后，我想在另一个小部件中显示该记录中的其他信息。

标签： python pyqt pyqt5 qsqltablemodel

【解决方案1】：

一个可能的选项是启用标志Qt::ItemIsUserCheckable，并在角色为Qt::CheckStateRole 时覆盖data() 和setData() 方法，以存储我们创建字典的数据。

class TableModel(QSqlTableModel):
    def __init__(self, *args, **kwargs):
        QSqlTableModel.__init__(self, *args, **kwargs)
        self.checkeable_data = {}

    def flags(self, index):
        fl = QSqlTableModel.flags(self, index)
        if index.column() == 1:
            fl |= Qt.ItemIsUserCheckable
        return fl

    def data(self, index, role=Qt.DisplayRole):
        if role == Qt.CheckStateRole and (
            self.flags(index) & Qt.ItemIsUserCheckable != Qt.NoItemFlags
        ):
            if index.row() not in self.checkeable_data.keys():
                self.setData(index, Qt.Unchecked, Qt.CheckStateRole)
            return self.checkeable_data[index.row()]
        else:
            return QSqlTableModel.data(self, index, role)

    def setData(self, index, value, role=Qt.EditRole):
        if role == Qt.CheckStateRole and (
            self.flags(index) & Qt.ItemIsUserCheckable != Qt.NoItemFlags
        ):
            self.checkeable_data[index.row()] = value
            self.dataChanged.emit(index, index, (role,))
            return True
        return QSqlTableModel.setData(self, index, value, role)

例子：

def createConnection():
    db = QSqlDatabase.addDatabase("QSQLITE");
    db.setDatabaseName(":memory:");
    if not db.open():
        print("Cannot open database"),
        print("Unable to establish a database connection.\n"
                        "This example needs SQLite support. Please read "
                        "the Qt SQL driver documentation for information how "
                        "to build it.\n\n"
                        "Click Cancel to exit.")
        return False

    query = QSqlQuery()
    query.exec_("create table person (id INTEGER PRIMARY KEY AUTOINCREMENT, "
               "firstname VARCHAR(20), lastname VARCHAR(20))");
    query.exec_("insert into person(firstname, lastname) values('Danny', 'Young')");
    query.exec_("insert into person(firstname, lastname) values('Christine', 'Holand')");
    query.exec_("insert into person(firstname, lastname) values('Lars', 'Gordon')");
    query.exec_("insert into person(firstname, lastname) values('Roberto', 'Robitaille')");
    query.exec_("insert into person(firstname, lastname) values('Maria', 'Papadopoulos')");

    return True

class Widget(QMainWindow):
    def __init__(self):
        QMainWindow.__init__(self)
        lv = QListView(self)
        self.setCentralWidget(lv)
        model = TableModel(self)
        model.setTable("person")
        model.select()
        lv.setModel(model)
        lv.setModelColumn(1)

if __name__ == '__main__':
    app = QApplication(sys.argv)
    if not createConnection():
        sys.exit(-1)
    ex = Widget()
    ex.show()
    sys.exit(app.exec_())

【讨论】：

正是我想要的。非常感谢。