PyQt5实现可拖拽QTreeWidget

简单实现了一个QTreeWidget重载，子类化即可查看示例。

主要功能包括：

1、右键菜单新建组

2、外部拖拽文件进组

3、内部节点组间拖动

4、TreeWidget里增加combo box和按钮示例

类实现代码

# coding=utf-8
import os.path
import typing

from PyQt5 import QtGui, QtCore
from PyQt5.QtGui import QColor, QBrush, QDrag, QFont
from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem, QMenu, QComboBox, QPushButton, QTreeWidgetItemIterator
from PyQt5.QtCore import Qt, QMimeData, QVariant, QPoint


class TreeItemMimeData(QMimeData):
    def __init__(self):
        self._format = []
        self._item: QTreeWidgetItem = None
        super().__init__()

    def set_drag_data(self, fmt: str, item: QTreeWidgetItem):
        self._format.append(fmt)
        self._item = item

    def get_drag_data(self):
        return self._item

    def formats(self):
        return self._format

    def retrieveData(self, mimetype: str, preferredType: 'QVariant.Type'):
        if mimetype == 'ItemMimeData':
            return self._item
        else:
            return QMimeData.retrieveData(mimetype, preferredType)


class CDragTreeWidget(QTreeWidget):
    def __init__(self, parent):
        super(CDragTreeWidget, self).__init__(parent)

        self._cut_item: QTreeWidgetItem = None
        self._vertical_scroll_value = 0

        # 允许内部元素拖动
        self.setDragEnabled(True)

        # 允许接受外部拖进
        self.setAcceptDrops(True)

        self._last_bg_colord_item: QTreeWidgetItem = None
        self._start_drag_pnt: QPoint = None

        self.itemChanged.connect(self._slot_item_changed)
        self.itemDoubleClicked.connect(self._slot_item_double_clicked)

        # 右键菜单
        self.setContextMenuPolicy(Qt.CustomContextMenu)
        self.customContextMenuRequested.connect(self._right_click_menu)

        self._add_top_level_item()
        self.setHeaderLabels(['1', '2', '3'])
        self.setColumnWidth(0, 200)
        self._index_cmb = 1
        self._index_btn = 2
        self._cut_item = None

    def startDrag(self, supportedActions: typing.Union[QtCore.Qt.DropActions, QtCore.Qt.DropAction]) -> None:
        item = self.currentItem()

        mime_data: TreeItemMimeData = TreeItemMimeData()
        mime_data.set_drag_data('ItemMimeData', item)

        drag: QDrag = QDrag(self)

        # 定义drag事件，自定义的Mimedata传递到dragEnterEvent, dragMoveEvent, dropEvent
        drag.setMimeData(mime_data)

        # 记录下垂直滚动条的值，如果拖动没有改变滚动条，则用这个值，否则用改变之后的
        self._vertical_scroll_value = self.verticalScrollBar().value()

        if drag.exec(Qt.MoveAction) == Qt.MoveAction:
            # 新的item已经克隆并添加之后，移除拖动之前的item
            self._remove_item(item)

            self._update_custom_control()

            # 垂直滚动条设置新的值
            self.verticalScrollBar().setValue(self._vertical_scroll_value)
            self._clear_bg_color()

    # 拖动开始时Event
    def dragEnterEvent(self, event: QtGui.QDragEnterEvent) -> None:
        # 内部item移动
        if event.source() == self:
            if event.mimeData().hasFormat('ItemMimeData'):
                event.setDropAction(Qt.M