我有一個QTableView
類,和一個QSQLRelationalTableModel
類用於顯示SQLITE數據庫中的數據。Python - 從QSQLRelationalTableModel/QTableView刪除選定的行
我已經將這些自己分類,因爲我需要覆蓋PyQt類中的各種方法。
我想實現一個'刪除行'功能,然後將自動重繪TableView。
不幸的是我很茫然。我已經覆蓋模型中的data()
方法,因爲我需要修改某些列返回的數據。 當我刪除一行,這個data()
方法顯然沒有什麼可以返回該行,這會引發類型錯誤(期望返回缺少的行的數據)。 目前,我已經坐在一個嘗試/除了塊(我知道這是不好的做法),並返回「已刪除」,如果它不工作。
這裏是我的完整型號:
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行剛剛被刪除):
我做的任何似乎成功刪除行的更改,使應用程序退出「處理完成退出代碼11」,我找不到任何詳細的解釋(平臺特定,我在Mac OS和Python 3.6上使用PyCharm)
謝謝