扩展 class 无法 pickle

Question

我设计了一种以编程方式继承 class 的方法，因此基本上扩展了 class。它工作得很好。但是，我需要腌制 class，这是不可能的，因为扩展的 class 是在函数中定义的。

这是扩展 class 的玩具示例：

import pickle

class A():
    pass

def extend_class(base_class):
    class B(base_class):
        def hello(self):
            print('Yo!')

    return B

extended_class = extend_class(A)

b = extended_class()
b.hello()

但是如果我这样做

pickle.dump(b, open('extended_class.pickle', 'w'))

这个returns:

AttributeError: Can't pickle local object 'extend_class.<locals>.B'

有什么解决方法吗？我不一定需要使用我的方式来扩展 class。任何其他方式都是可以接受的，只要我最后能腌制 class。

Answer 1

你可以在全局域中声明一个class，然后在工厂函数中声明它为全局域，并覆盖它。
它看起来有点 hack，可能有更好的方法（也许通过将 class 直接注入 class dict），但是，这里是：

import pickle

class A:
    pass

class _B:
    pass
B = _B

def extend_class(base_class):
    global B
    class B(base_class):
        def hello(self):
            print('Yo!')

    return B


if __name__ == '__main__':

    extended_class = extend_class(A)

    b = extended_class()
    b.hello()
    with open('tst.pickle', "wb") as f:
        pickle.dump(b, f)

扩展 class 无法 pickle

Extended class can't be pickled

python

oop

class

pickle

python-3.x