如何在 QTableWidget 中将拖放与排序结合起来？

在 QTableWidget 中，我希望能够：

• 按列对表格进行排序和
• 通过将行拖放到不同位置来更改行顺序

经过大量过时或令人困惑的代码片段之后，到目前为止，对我来说，最清晰的移动行解决方案是here。 排序可以简单地通过 setSortingEnabled(True/False) 启用/禁用。

拖放单独工作，排序单独工作，但不能一起工作。

我猜是什么时候删除表将再次排序。 所以，我想如果我在拖动过程中关闭排序应该会得到想要的结果，但事实并非如此。显然，我做错了什么。

我确定这一定是一个小细节，但我在这里遗漏了什么？看起来我需要去掉列标题中的排序箭头。

代码：（应该是复制粘贴运行）：

import sys
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QApplication, QWidget, QTableWidget, QVBoxLayout, QTableWidgetItem, QAbstractItemView
from PyQt5.QtGui import QDropEvent, QDragMoveEvent
from datetime import datetime
import random

class TableWidgetDragRows(QTableWidget):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.setSortingEnabled(True)    # enable sorting by default
        self.setDragEnabled(True)
        self.setAcceptDrops(True)
        self.viewport().setAcceptDrops(True)
        self.setDragDropOverwriteMode(False)
        self.setDropIndicatorShown(True)
        self.setSelectionMode(QAbstractItemView.ExtendedSelection)
        self.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.setDragDropMode(QAbstractItemView.InternalMove)

    def dragEvent(self, event: QDragMoveEvent):
        self.setSortingEnabled(False)   # disable sorting during dragging

    def dropEvent(self, event: QDropEvent):
        self.setSortingEnabled(False)   # disable sorting during dropping

        if not event.isAccepted() and event.source() == self:
            drop_row = self.drop_on(event)

            rows = sorted(set(item.row() for item in self.selectedItems()))
            rows_to_move = [[QTableWidgetItem(self.item(row_index, column_index)) for column_index in range(self.columnCount())]
                            for row_index in rows]
            for row_index in reversed(rows):
                self.removeRow(row_index)
                if row_index < drop_row:
                    drop_row -= 1

            for row_index, data in enumerate(rows_to_move):
                row_index += drop_row
                self.insertRow(row_index)
                for column_index, column_data in enumerate(data):
                    self.setItem(row_index, column_index, column_data)
            event.accept()

            for row_index in range(len(rows_to_move)):   # maybe can be done smarter
                for col in range(self.columnCount()):
                    self.item(drop_row + row_index, col).setSelected(True)

        super().dropEvent(event)
        self.setSortingEnabled(True)

    def drop_on(self, event):
        index = self.indexAt(event.pos())
        if not index.isValid():
            return self.rowCount()
        return index.row() + 1 if self.is_below(event.pos(), index) else index.row()

    def is_below(self, pos, index):
        rect = self.visualRect(index)
        margin = 2
        if pos.y() - rect.top() < margin:
            return False
        elif rect.bottom() - pos.y() < margin:
            return True
        # noinspection PyTypeChecker
        return rect.contains(pos, True) and not (int(self.model().flags(index)) & Qt.ItemIsDropEnabled) and pos.y() >= rect.center().y()

class MyWindow(QWidget):
    def __init__(self):
        super(MyWindow,self).__init__()
        self.setGeometry(100,100,600,300)
        self.layout = QVBoxLayout()
        self.setLayout(self.layout)
        
        self.tw = TableWidgetDragRows(self)
        self.layout.addWidget(self.tw)
        self.tw.setRowCount(5)
        self.tw.setColumnCount(3)
        #
        for row in range(self.tw.rowCount()):
            for col in range(self.tw.columnCount()):
                if col==0:
                    myValue = "".join([chr(random.randint(65,90)) for i in range(0,4)])
                else:
                    myValue = random.randint(0,100)
                twi = QTableWidgetItem()
                twi.setData(Qt.DisplayRole,myValue)
                self.tw.setItem(row, col, twi)
        self.show()
            
if __name__ == '__main__':
    app = QApplication(sys.argv)
    app.setStyle("Fusion")
    window = MyWindow()
    sys.exit(app.exec_())

结果：