Python - 从 QSQLRelationalTableModel / QTableView 中删除选定的行

Question

我有一个 QTableView 类和一个 QSQLRelationalTableModel 类，用于显示 SQLITE 数据库中的数据。

我自己对这些进行了子类化，因为我需要覆盖 PyQt 类中的各种方法。

我正在尝试实现一个“删除行”功能，该功能将自动重绘 TableView。

不幸的是，我不知所措。我已经覆盖了模型中的data() 方法，因为我需要修改为某些列返回的数据。 当我删除一行时，这个data() 方法显然不会为该行返回任何内容，这会引发类型错误（它期望为丢失的行返回数据）。 目前我把它放在一个 try/except 块中（我知道这是不好的做法），如果它不起作用则返回“Deleted”。

这是我的完整模型：

class PayRateTableModel(QtSql.QSqlRelationalTableModel):
    def __init__(self,master):
        QtSql.QSqlRelationalTableModel.__init__(self,db=master.db)
        self.header_labels = ['ID','From','To','Day Rate','Night Rate', 'Description']
        self.editable_columns = [1,2,3,4,5]
        self.setTable('payrates')
        self.setEditStrategy(QtSql.QSqlRelationalTableModel.OnRowChange)
        self.select()

    def columnCount(self,master,parent=QtCore.QModelIndex()):
        return 6

    def flags(self, index):
        flags=super(self.__class__, self).flags(index)
        if index.column() not in self.editable_columns:
            flags ^= QtCore.Qt.ItemIsEditable
        return flags

    def headerData(self, p_int, Qt_Orientation, role=QtCore.Qt.DisplayRole):
        if role == QtCore.Qt.DisplayRole and Qt_Orientation == QtCore.Qt.Horizontal:
            return self.header_labels[p_int]
        return QtCore.QAbstractTableModel.headerData(self,p_int, Qt_Orientation,role)

    def data(self, index, role):
        row=index.row()
        col=index.column()
        data = super(PayRateTableModel, self).data(index, role)
        if role==QtCore.Qt.TextAlignmentRole:
            return QtCore.Qt.AlignCenter
        elif role==QtCore.Qt.FontRole:
            f=QtGui.QFont
            return f
        elif role==QtCore.Qt.ForegroundRole and index.column() not in self.editable_columns:
            return QtGui.QColor(150,150,150)
        try:
            if (col in [1,2]) and (role==QtCore.Qt.DisplayRole) and data:
                data=super(PayRateTableModel, self).data(index,role)
                year=data[:4]
                month=data[5:7]
                day=data[8:10]
                data=QtCore.QDate(int(year),int(month),int(day)).toString("dd/MM/yyyy")
                return data
            elif col in [3,4] and role ==QtCore.Qt.DisplayRole:
                data=super(PayRateTableModel, self).data(index,role)
                return '%.4f' % data
        except:
            return "Deleted"

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

这是我的表格视图

class PayRateTableView(QtWidgets.QTableView):
    """ Display table of editable pay rates"""
    def __init__(self, master):
        QtWidgets.QTableView.__init__(self)
        self.hideColumn(0)
        self.setSortingEnabled(True)
        self.model = PayRateTableModel(master)
        self.sortmodel=DateSortProxyModel()
        self.sortmodel.setSourceModel(self.model)
        self.setModel(self.sortmodel)
        self.sortByColumn(1,QtCore.Qt.AscendingOrder)
        self.setSelectionBehavior(QtWidgets.QTableView.SelectRows)
        self.resizeRowsToContents()
        self.resizeColumnsToContents()
        self.setAlternatingRowColors(True)
        self.horizontalHeader().setSectionResizeMode(QtWidgets.QHeaderView.Stretch)
        self.setItemDelegateForColumn(1, CalendarDelegate(master))
        self.setItemDelegateForColumn(2, CalendarDelegate(master))
        self.setItemDelegateForColumn(3, PayRateDelegate(master))
        self.setItemDelegateForColumn(4, PayRateDelegate(master))
        self.vhead=self.verticalHeader()
        self.vhead.setVisible(False)
        self.vhead.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
        self.vhead.customContextMenuRequested.connect(self.contextMenuEvent)
    def sizeHintForRow(self, p_int):
        return 20

    def contextMenuEvent(self, event):
        index = self.currentIndex()
        menu=QtWidgets.QMenu(self)
        menu.popup(QtGui.QCursor.pos())
        deleteAction = menu.addAction("Delete Row")
        action = menu.exec_(self.mapToGlobal(event.pos()))
        if action == deleteAction:
            self.model.removeRow(index.row())
            self.model.submitAll()

这确实有效，但是虽然它删除了数据，但它在表格顶部留下了半行，如下所示（第 3 行刚刚被删除）：

我所做的任何更改似乎都成功删除了该行，使应用程序退出并显示“处理完成，退出代码 11”，我找不到任何详细的解释（特定于平台，我在 Mac OS 和 Python 上使用 PyCharm 3.6)

谢谢

Answer 1

事实证明，我似乎已经想通了。的种类。 虽然它本身不是一个直接的答案，但我将删除方法从上下文菜单更改为工具栏图标。 选择行，点击工具栏图标，调用下面的Class方法：

def deletePayRate(self):
    index = self.currentIndex()
    deleteconfirmation = QtWidgets.QMessageBox.critical(self.parent(), "Delete row", "Really delete the selected Pay Rate?", QtWidgets.QMessageBox.Yes, QtWidgets.QMessageBox.No)
    if deleteconfirmation == QtWidgets.QMessageBox.Yes:
        self.model.removeRow(index.row())
        self.model.submitAll()
        return
    else:
        return

简单但有效。