如何修改含__getattr__的Python类使其支持pickle序列化？

首先得搞清楚问题出在哪：当pickle处理对象时，会尝试访问一些特殊方法（比如__reduce__、__getstate__这类双下划线包裹的方法），如果你的类定义了__getattr__，但没处理这些特殊属性的话，__getattr__会被触发，要么抛出异常，要么返回不符合预期的值，直接导致pickle序列化失败。

下面给你两种实用的解决方案，结合你的Foo类来演示：

方法一：在__getattr__中排除特殊属性 #

修改你的__getattr__方法，当访问以双下划线开头和结尾的特殊属性时，直接抛出AttributeError，让pickle走默认的序列化逻辑，不会被自定义逻辑干扰。

示例代码：

import pickle

class Foo(object):
    def __init__(self, dct):
        for key, value in dct.items():
            setattr(self, key, value)
    
    def __getattr__(self, name):
        # 重点：排除特殊方法，让pickle正常访问它们
        if name.startswith('__') and name.endswith('__'):
            raise AttributeError(f"'Foo' object has no attribute '{name}'")
        # 这里是你原来的__getattr__自定义逻辑，比如返回默认值或者报错
        raise AttributeError(f"'Foo' object has no attribute '{name}'")

# 测试序列化和反序列化
foo = Foo({'a': 1, 'b': 2})
pickled_data = pickle.dumps(foo)
restored_foo = pickle.loads(pickled_data)
print(restored_foo.a)  # 输出：1

方法二：实现__getstate__和__setstate__方法 #

手动控制序列化和反序列化的过程，告诉pickle要保存哪些数据，以及如何恢复数据。这种方法更灵活，适合复杂的类结构。

示例代码：

import pickle

class Foo(object):
    def __init__(self, dct):
        for key, value in dct.items():
            setattr(self, key, value)
    
    def __getattr__(self, name):
        # 你的自定义__getattr__逻辑
        raise AttributeError(f"'Foo' object has no attribute '{name}'")
    
    def __getstate__(self):
        # 返回要序列化的对象状态，这里直接返回实例的所有属性字典
        return self.__dict__
    
    def __setstate__(self, state):
        # 反序列化时，把保存的状态恢复到实例中
        self.__dict__.update(state)

# 测试
foo = Foo({'x': 10, 'y': 20})
pickled_data = pickle.dumps(foo)
restored_foo = pickle.loads(pickled_data)
print(restored_foo.y)  # 输出：20

额外说明 #

如果你的__getattr__是从某个内部存储（比如self._data字典）动态获取属性，那在__getstate__里要确保把这个内部存储也包含进去，比如：

def __getstate__(self):
    return {'_data': self._data}

def __setstate__(self, state):
    self._data = state['_data']

这样反序列化后，__getattr__才能正常工作。

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