PySide2自定义QStandardItem拖拽后类型异常问题求助

嘿，这个问题我之前也碰到过——默认的QStandardItemModel在处理拖拽操作时，只会按照标准项的逻辑来序列化和重建数据，自然不会保留你的MyStandardItem类型。下面给你详细说下解决方案：

是不是必须自定义模型？ #

没错，你需要创建自定义的模型子类（继承QStandardItemModel）。因为默认模型在接收拖拽数据时，只会生成QStandardItem实例，根本不知道你的自定义子类存在。不过放心，实现起来只需要重写几个关键的MIME相关方法，代码量不大。

自定义模型需要重写哪些函数？ #

你需要重写三个核心方法：mimeTypes()、mimeData()和dropMimeData()，分别负责定义自定义MIME类型、序列化自定义项数据、以及解析数据重建自定义项。

1. 先完善你的MyStandardItem #

首先给你的自定义项加个实例属性保存图标路径（方便序列化），同时简化setCheckState的写法：

from PySide2.QtCore import Qt
from PySide2.QtGui import QStandardItem, QIcon, QFont

class MyStandardItem(QStandardItem):
    def __init__(self, text, font, icon_path='', value='', num=0, check_state=None):
        super().__init__()
        self.setDragEnabled(True)
        self.setDropEnabled(True)
        self.setText(text)
        self.setData({'value': (value, num)})
        self.setToolTip(str(self.data()['value']))
        self.setFont(font)
        self.icon_path = icon_path  # 保存图标路径，方便序列化
        self.setIcon(QIcon(icon_path))
        self.toggled = check_state
        if check_state is not None:
            self.setCheckable(True)
            self.setCheckState(check_state)
    
    def setCheckState(self, checkState):
        super().setCheckState(checkState)
        self.toggled = checkState  # 直接赋值，简化逻辑

2. 实现自定义模型MyStandardItemModel #

from PySide2.QtCore import QMimeData, Qt
from PySide2.QtGui import QStandardItemModel
import json

class MyStandardItemModel(QStandardItemModel):
    # 定义唯一的自定义MIME类型标识符
    CUSTOM_MIME_TYPE = "application/x-mystandarditem"

    def mimeTypes(self):
        # 扩展默认的MIME类型，添加我们的自定义类型
        mime_types = super().mimeTypes()
        mime_types.append(self.CUSTOM_MIME_TYPE)
        return mime_types

    def mimeData(self, indexes):
        mime_data = super().mimeData(indexes)
        # 处理单个选中项的拖拽（多个项的话可以循环扩展）
        if len(indexes) == 1:
            index = indexes[0]
            item = self.itemFromIndex(index)
            if isinstance(item, MyStandardItem):
                # 把自定义项的所有属性序列化为JSON格式
                font = item.font()
                item_data = {
                    "text": item.text(),
                    "font": {
                        "family": font.family(),
                        "pointSize": font.pointSize(),
                        "bold": font.bold(),
                        "italic": font.italic()
                    },
                    "icon_path": item.icon_path,
                    "value": item.data()['value'][0],
                    "num": item.data()['value'][1],
                    "check_state": item.checkState() if item.isCheckable() else None
                }
                # 转成字节数据存入MIME
                json_bytes = json.dumps(item_data).encode('utf-8')
                mime_data.setData(self.CUSTOM_MIME_TYPE, json_bytes)
        return mime_data

    def dropMimeData(self, data, action, row, column, parent):
        # 优先处理自定义MIME类型
        if data.hasFormat(self.CUSTOM_MIME_TYPE):
            try:
                # 解析JSON数据
                json_str = data.data(self.CUSTOM_MIME_TYPE).data().decode('utf-8')
                item_data = json.loads(json_str)
                
                # 重建字体对象
                font = QFont()
                font.setFamily(item_data['font']['family'])
                font.setPointSize(item_data['font']['pointSize'])
                font.setBold(item_data['font']['bold'])
                font.setItalic(item_data['font']['italic'])
                
                # 处理复选状态
                check_state = item_data['check_state']
                if check_state is not None:
                    check_state = Qt.CheckState(check_state)
                
                # 创建自定义项实例
                new_item = MyStandardItem(
                    text=item_data['text'],
                    font=font,
                    icon_path=item_data['icon_path'],
                    value=item_data['value'],
                    num=item_data['num'],
                    check_state=check_state
                )
                
                # 将新项添加到模型的对应位置
                parent_item = self.itemFromIndex(parent) if parent.isValid() else self.invisibleRootItem()
                insert_row = row if row != -1 else parent_item.rowCount()
                parent_item.insertRow(insert_row, new_item)
                return True
            except Exception as e:
                print(f"解析自定义拖拽数据出错: {e}")
        # 如果自定义处理失败， fallback到默认逻辑
        return super().dropMimeData(data, action, row, column, parent)

3. 最后在你的视图中使用自定义模型 #

确保你的MyTreeView使用的是MyStandardItemModel，而不是默认的QStandardItemModel：

from PySide2.QtWidgets import QTreeView

class MyTreeView(QTreeView):
    def __init__(self, parent=None):
        super().__init__(parent)
        # 设置自定义模型
        self.setModel(MyStandardItemModel())
        # 配置拖拽模式（根据你的需求调整）
        self.setDragDropMode(QTreeView.InternalMove)
        self.setSelectionMode(QTreeView.SingleSelection)

关键逻辑说明 #

• mimeTypes()：告诉Qt我们的模型支持自定义的MIME类型，这样拖拽数据会带上这个标识。
• mimeData()：当拖拽开始时，把MyStandardItem的所有必要属性（文本、字体、图标路径等）序列化为JSON，存入MIME数据中。
• dropMimeData()：当接收拖拽时，先检查是否有我们的自定义MIME数据，如果有就解析并重建MyStandardItem，而不是默认的QStandardItem。

这样处理后，拖拽后的项就会保持MyStandardItem类型了。

内容的提问来源于stack exchange，提问作者John Reds