Python - 从QSQLRelationalTableModel / QTableView

Question

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

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

我正在尝试实施“删除行”＃39;然后将自动重绘TableView的函数。

不幸的是我不知所措。我已经覆盖了模型中的data()方法，因为我需要修改为某些列返回的数据。 当我删除一行时，这个data()方法显然没有任何东西可以返回该行，这会抛出类型错误（它期望返回缺失行的数据）。 目前我让它坐在一个尝试/除了块（我知道这是一个不好的做法），并返回＆＃34;删除＆＃34;如果它不起作用。

这是我完整的模型：

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)

这是我的TableView

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行刚被删除）：

我做的任何变化似乎都成功删除了行，让应用程序退出＆＃34;已完成退出代码11＆＃34;我找不到任何详细的解释（特定于平台，我在Mac OS和Python 3.6上使用PyCharm）

由于

Answer 1

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

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

简单但有效。