2017-08-24 201 views
-1

我有一個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行剛剛被刪除):

Screenshot of QTableView with one row deleted

我做的任何似乎成功刪除行的更改,使應用程序退出「處理完成退出代碼11」,我找不到任何詳細的解釋(平臺特定,我在Mac OS和Python 3.6上使用PyCharm)

謝謝

回答

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 

簡單但很有效。