2017-08-24 202 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 

简单但很有效。