如何实现存储QPixmap的自定义类的Pickle序列化？

我明白你的痛点——QPixmap本身不支持Pickle，而你又想避免重复执行耗CPU的图像操作。其实你之前的思路方向是对的，只是不需要把QBuffer纳入序列化流程里，它只是个临时转换工具而已。

核心思路是：在序列化时，把QPixmap转换成可Pickle的QByteArray（仅保存这个字节数组，临时用的QBuffer直接丢弃）；反序列化时，再把QByteArray还原回QPixmap。

下面是具体的实现代码示例，假设你的自定义类名为CustomPixmapHolder，其中存储QPixmap的属性叫_processed_pixmap：

from PyQt5.QtGui import QPixmap, QByteArray, QBuffer
from PyQt5.QtCore import QIODevice
import pickle

class CustomPixmapHolder:
    def __init__(self, original_image_path):
        # 模拟耗时的图像处理操作
        self._processed_pixmap = self._heavy_image_processing(original_image_path)

    def _heavy_image_processing(self, image_path):
        # 这里是你的重型操作：缩放、滤镜等
        pixmap = QPixmap(image_path)
        return pixmap.scaled(800, 600)  # 示例缩放操作

    def __getstate__(self):
        # 复制当前对象的状态字典，排除不可序列化的QPixmap
        state = self.__dict__.copy()
        # 取出QPixmap并转换为QByteArray
        pixmap = state.pop("_processed_pixmap")
        byte_array = QByteArray()
        # 临时使用QBuffer写入数据，用完就丢，不纳入序列化
        buffer = QBuffer(byte_array)
        buffer.open(QIODevice.WriteOnly)
        # 用PNG格式保存（无损，避免画质损失，也可以选JPG等其他格式）
        pixmap.save(buffer, "PNG")
        # 将字节数组存入状态字典
        state["_pixmap_byte_data"] = byte_array
        return state

    def __setstate__(self, state):
        # 恢复对象基础状态
        self.__dict__.update(state)
        # 从字节数组还原QPixmap
        byte_array = state["_pixmap_byte_data"]
        restored_pixmap = QPixmap()
        restored_pixmap.loadFromData(byte_array)
        # 恢复原有的QPixmap属性
        self._processed_pixmap = restored_pixmap
        # 清理临时存储的字节数据，避免对象冗余
        del self._pixmap_byte_data

# 测试代码
if __name__ == "__main__":
    # 创建实例（执行耗时操作）
    holder = CustomPixmapHolder("test_image.jpg")
    # 序列化到文件
    with open("pixmap_holder.pkl", "wb") as f:
        pickle.dump(holder, f)
    # 反序列化
    with open("pixmap_holder.pkl", "rb") as f:
        restored_holder = pickle.load(f)
    # 验证还原后的QPixmap可用
    print(f"原尺寸：{holder._processed_pixmap.size()}")
    print(f"还原后尺寸：{restored_holder._processed_pixmap.size()}")

关键注意点： #

• 格式选择：示例用PNG格式保存，因为它是无损压缩，能完整保留图像数据；如果对文件大小有要求，也可以换成JPG，但会有画质损失，根据你的需求调整。
• 临时对象处理：QBuffer只是转换过程中的临时工具，我们并没有把它存入__getstate__返回的状态字典，所以完全不会涉及到它的序列化问题。
• 属性清理：反序列化后记得删除临时存储的字节数据，避免你的类实例多出不必要的属性。

这个方案既利用了QByteArray可Pickle的特性，又避开了QBuffer的序列化限制，完美解决跨会话复用处理后QPixmap的需求。

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