PyQt5在QTableview中拖拽导致选中行消失

Posted 2023-03-25

【中文标题】PyQt5在QTableview中拖拽导致选中行消失

【英文标题】：PyQt5 dragging and dropping in QTableview causes selected row to disappear

【发布时间】：2021-12-12 20:43:24

【问题描述】：

继上一个问题here 之后，我希望通过将 removeRows 函数添加到我的模型来将删除行按键功能添加到 PyQt5 中的 qtableview 表中。但是，自从添加了这个功能后，它破坏了我的拖放功能，当拖放到 qtableview 表中的其他位置时，被拖动的行消失了。 无论如何我可以防止拖动的行消失吗？

nb：有趣的是，在选择垂直标题'列'时，拖放功能工作，但我热衷于找到拖动行选择的解决方案。

下面是我的代码，在模型中添加了 removeRows 函数，以及视图中的 keyPressEvent

from PyQt5.QtGui import QBrush
from PyQt5.QtWidgets import *
from PyQt5.QtCore import QAbstractTableModel, Qt, QModelIndex

class myModel(QAbstractTableModel):
    def __init__(self, data, parent=None, *args):
        super().__init__(parent, *args)
        self._data = data or []
        self._headers = ['Type', 'result', 'count']

    def rowCount(self, index=None):
        return len(self._data)

    def columnCount(self, index=None):
        return len(self._headers)

    def headerData(self, section, orientation, role=Qt.DisplayRole):
        if role == Qt.DisplayRole:
            if orientation == Qt.Horizontal:
                if section < 0 or section >= len(self._headers):
                    return ""
                else:
                    return self._headers[section]
            else:
                return ''
        return None

    def removeRows(self, position, rows, QModelIndex):
        self.beginRemoveRows(QModelIndex, position, position + rows - 1)
        for i in range(rows):
            del (self._data[position])
        self.endRemoveRows()
        self.layoutChanged.emit()
        return True

    def data(self, index, role=None):
        if role == Qt.TextAlignmentRole:
            return Qt.AlignHCenter
        if role == Qt.ForegroundRole:
            return QBrush(Qt.black)
        if role == Qt.BackgroundRole:
            if (self.index(index.row(), 0).data().startswith('second')):
                return QBrush(Qt.green)
            else:
                if (self.index(index.row(), 1).data()) == 'abc':
                    return QBrush(Qt.yellow)
                if (self.index(index.row(), 1).data()) == 'def':
                    return QBrush(Qt.blue)
                if (self.index(index.row(), 1).data()) == 'ghi':
                    return QBrush(Qt.magenta)
        if role in (Qt.DisplayRole, Qt.EditRole):
            return self._data[index.row()][index.column()]

    def flags(self, index: QModelIndex) -> Qt.ItemFlags:
        return Qt.ItemIsDropEnabled | Qt.ItemIsEnabled | Qt.ItemIsEditable | Qt.ItemIsSelectable | Qt.ItemIsDragEnabled

    def supportedDropActions(self) -> bool:
        return Qt.MoveAction | Qt.CopyAction


class myTableView(QTableView):
    def __init__(self, parent):
        super().__init__(parent)
        header = self.verticalHeader()
        header.setSectionsMovable(True)
        header.setSectionResizeMode(QHeaderView.Fixed)
        header.setFixedWidth(10)
        QShortcut('F7', self, self.getLogicalRows)
        QShortcut('F6', self, self.toggleVerticalHeader)
        QShortcut('Alt+Up', self, lambda: self.moveRow(True))
        QShortcut('Alt+Down', self, lambda: self.moveRow(False))
        self.setSelectionBehavior(self.SelectRows)
        self.setSelectionMode(self.SingleSelection)
        self.setDragDropMode(self.InternalMove)
        self.setDragDropOverwriteMode(False)

    def dropEvent(self, event):
        if (event.source() is not self or
            (event.dropAction() != Qt.MoveAction and
             self.dragDropMode() != QAbstractItemView.InternalMove)):
            super().dropEvent(event)
        selection = self.selectedIndexes()
        from_index = selection[0].row() if selection else -1
        to_index = self.indexAt(event.pos()).row()
        if (0 <= from_index < self.model().rowCount() and
            0 <= to_index < self.model().rowCount() and
            from_index != to_index):
            header = self.verticalHeader()
            from_index = header.visualIndex(from_index)
            to_index = header.visualIndex(to_index)
            header.moveSection(from_index, to_index)
            event.accept()
        super().dropEvent(event)

    def toggleVerticalHeader(self):
        self.verticalHeader().setHidden(self.verticalHeader().isVisible())

    def moveRow(self, up=True):
        selection = self.selectedIndexes()
        if selection:
            header = self.verticalHeader()
            row = header.visualIndex(selection[0].row())
            if up and row > 0:
                header.moveSection(row, row - 1)
            elif not up and row < header.count() - 1:
                header.moveSection(row, row + 1)

    def getLogicalRows(self):
        header = self.verticalHeader()
        for vrow in range(header.count()):
            lrow = header.logicalIndex(vrow)
            index = self.model().index(lrow, 0)
            print(index.data())

    def keyPressEvent(self, event):
        if event.key() == Qt.Key_Delete:
            index = self.currentIndex()
            try:
                self.model().removeRows(index.row(), 1, index)
            except IndexError:
                pass
        else:
            super().keyPressEvent(event)


class sample_data(QMainWindow):
    def __init__(self):
        super().__init__()
        tv = myTableView(self)
        tv.setModel(myModel([
            ["first", 'abc', 123],
            ["second"],
            ["third", 'def', 456],
            ["fourth", 'ghi', 789],
        ]))
        self.setCentralWidget(tv)
        tv.setSpan(1, 0, 1, 3)

if __name__ == '__main__':

    app = QApplication(['Test'])
    test = sample_data()
    test.setGeometry(600, 100, 350, 185)
    test.show()
    app.exec_()

【参考方案1】：

主要的“问题”是，默认情况下，QAbstractItemModel 的removeRows 不执行任何操作（并返回False）。

技术问题有点微妙。 项目视图中的拖动操作总是以startDrag() 开头，它创建一个QDrag 对象并调用它的exec()。当用户删除数据时，只要接受的删除操作是 MoveAction，该实现还会调用私有 clearOrRemove 函数，这最终会覆盖数据或删除行。

你使用了setDragDropOverwriteMode(False)，所以它会调用removeRows。您之前的代码可以正常工作，因为如前所述，默认实现什么都不做，但现在您重新实现了它，在这种情况下它实际上删除了行。

解决方案是在放置事件是移动时更改放置操作（这有点不直观，但由于操作已经执行，这应该不是问题）。使用IgnoreAction 将避免不必要的行为，因为在这种情况下将不再调用clearOrRemove：

    def dropEvent(self, event):
        # ...
        if (0 <= from_index < self.model().rowCount() and
            0 <= to_index < self.model().rowCount() and
            from_index != to_index):
            # ...
            header.moveSection(from_index, to_index)
            event.accept()
            event.setDropAction(Qt.IgnoreAction)
        super().dropEvent(event)

更新

当项目没有填满整个视口时，放置事件可能会发生在项目范围之外，从而导致来自indexAt() 的 QModelIndex 无效，不仅在放置到最后一行之外，而且在最后一列之外。 由于您只对垂直移动感兴趣，因此解决方案是从垂直标题中获取to_index，并最终在它仍然无效时将其设置为最后一行（-1）：

    def dropEvent(self, event):
        if (event.source() is not self or
            (event.dropAction() != Qt.MoveAction and
             self.dragDropMode() != QAbstractItemView.InternalMove)):
            super().dropEvent(event)
        selection = self.selectedIndexes()
        from_index = selection[0].row() if selection else -1

        globalPos = self.viewport().mapToGlobal(event.pos())
        header = self.verticalHeader()
        to_index = header.logicalIndexAt(header.mapFromGlobal(globalPos).y())
        if to_index < 0:
            to_index = header.logicalIndex(self.model().rowCount() - 1)

        if from_index != to_index:
            from_index = header.visualIndex(from_index)
            to_index = header.visualIndex(to_index)
            header.moveSection(from_index, to_index)
            event.accept()
            event.setDropAction(Qt.IgnoreAction)
        super().dropEvent(event)

【讨论】：

非常感谢@musicamante，我永远找不到那个解决方案，更好的是它是一个单行解决方案！

我注意到自从上面实施了您的修复程序，虽然它总体上有效，但我注意到如果您将一行拖放到最后一行之外，它也会导致该行消失，有没有办法轻松解决这个问题？

@MikG 查看更新。

再次感谢您的帮助@musicmante，您的更新解决方案完美运行

抱歉再次纠缠你@musicamante，不确定我是否应该提出第二个问题，但我现在注意到如果我将一行拖放到表格底部（比如第二个行移动到表的底部），然后我在选定的第一行上按删除键，第二行现在移回到它在表中的原始位置，尽管现在在第一行的位置，因为原来的顶行被删除了。我是否需要向视图添加新方法以根据可见行而不是逻辑行进行刷新？