如何编写load方法更新类属性？pickle序列化问题

问题原因 #

在实例方法中，self是方法的局部参数，直接执行self = pickle.load(...)只是修改了这个局部变量的指向，并不会改变调用该方法的原实例对象，因此属性无法得到更新。

解决方案 #

核心思路是：从Pickle文件加载出对象后，将其属性复制到当前实例，而非直接替换self变量。

方法1：通过__dict__批量更新属性（适用于大多数自定义类） #

大多数自定义类的实例属性会存储在__dict__字典中，直接用update方法可以批量同步属性：

import pickle

class YourClass:
    def save(self, filepath):
        # 原save方法保持不变，改用with语句自动管理文件
        with open(filepath + '.pickle', 'wb') as pickle_out:
            pickle.dump(self, pickle_out)
    
    def load(self, filepath):
        with open(filepath, 'rb') as infile:
            loaded_obj = pickle.load(infile)
        
        # 校验加载对象与当前实例类型一致，避免属性不匹配
        if not isinstance(loaded_obj, type(self)):
            raise TypeError(f"加载对象类型不匹配：期望{type(self)}，实际{type(loaded_obj)}")
        
        # 同步加载对象的所有属性到当前实例
        self.__dict__.update(loaded_obj.__dict__)

方法2：处理定义了__slots__的类 #

如果你的类使用__slots__限制实例属性（节省内存），实例不会生成__dict__，需要遍历__slots__中的属性逐个赋值：

import pickle

class YourClass:
    __slots__ = ['attr1', 'attr2']  # 假设类定义了这些属性
    
    def save(self, filepath):
        with open(filepath + '.pickle', 'wb') as pickle_out:
            pickle.dump(self, pickle_out)
    
    def load(self, filepath):
        with open(filepath, 'rb') as infile:
            loaded_obj = pickle.load(infile)
        
        if not isinstance(loaded_obj, type(self)):
            raise TypeError(f"加载对象类型不匹配：期望{type(self)}，实际{type(loaded_obj)}")
        
        # 遍历slots中的属性，逐个同步到当前实例
        for attr in self.__slots__:
            if hasattr(loaded_obj, attr):
                setattr(self, attr, getattr(loaded_obj, attr))

补充说明 #

• 使用with语句管理文件可以自动关闭文件，比手动调用close()更安全简洁。
• 添加类型检查能有效避免加载不同类对象导致的属性异常。

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