python - 可以用装饰器劫持类定义吗？

Question

假设我有一个

class A:
    def __init__(self, *args):
        pass

我想要一个装饰器来复制 A 的定义并用新类扩展它。

def decorator(cls): # some decorator here
    # make a new class which inherits from A
    # return it while preserving the original A

这可能吗？ (PS:这是为了避免维护问题。)

Answer 1

当您使用装饰器语法调用函数时:

@my_decorator_function
class A:
    pass

装饰器函数的返回值将替换 A 的现有定义.因此，如果您希望它创建一个新类并“在保留原始 A 的同时返回它”，您将遇到一个棘手的挑战。你是什​​么return将取代 A , 所以你需要决定这是否应该是原来的 A或新类(class)。您可以将另一个放在其他地方。

例如，这个装饰器将替换 A有一个子类，子类将生成原始的 A类可用作名为 _orig 的类属性:

def class_decorator(cls):
    class SubClass(cls):
        _orig = cls
        # add other stuff here?
    return SubClass

您可以添加额外的逻辑来复制原始类的 __name__和 __doc__如果你愿意，可以进入新类(class)。您也可以扭转逻辑，并添加 SubClass作为 cls 的属性在返回未经修改的 cls 之前.