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

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 还是新的类。你可以把另一个放在别的地方。

例如，这个装饰器会将A 替换为一个子类，并且该子类会将原来的A 类作为一个名为_orig 的类属性提供：

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

如果需要，您可以添加额外的逻辑来将原始类的 __name__ 和 __doc__ 复制到新类中。你也可以扭转逻辑，在返回未修改的cls之前添加SubClass作为cls的属性。

Answer 2

使用@decorator 不是唯一可能的语法。您可以将B = decorator(A) 放在类定义之后。

class A:
    ...

B = decorator(A)

现在您仍然有未修饰的A 的引用，并且您有一个修饰的版本B。

Answer 3

其他答案做得很好，但要清楚为什么你不想这样做：

def dec(cls):
    new_cls = type(cls.__name__, (cls,), {})
    return new_cls

@dec
class A():
    pass

现在检查方法解析顺序类A：

>>> A.__mro__
(<class '__main__.A'>, <class '__main__.A'>, <class 'object'>)
>>> classes = A.__mro__
>>> classes[0].__name__
'A'
>>> classes[1].__name__
'A'

两班As！它们是一样的吗？

>>> classes[0] is classes[1]
False

不;不同的。当前变量A当然是指向最低的一个：

>>> A is classes[0]
True

但是现在您已经失去了对父级的名称访问权限。这通常不是最佳的。

简而言之：几个月后，当您忘记自己所做的一切时，您正在为自己制造大量的困惑和模棱两可。做点别的。

如果你真的想要，这里有一个衍生出新子类的想法：

def add_babymaker(cls):
    '''Adds a method for making new child classes.'''
    def babymaker(name=None):
        '''Creates a new child class based on the parent class.'''
        name = name if name is not None else cls.__name__
        new_cls = type(name, (cls,), {})
        return new_cls
    cls.babymaker = babymaker
    return cls

@add_babymaker
class A():
    pass

B = A.babymaker('B')
C = A.babymaker('C')
ANew = A.babymaker()

Answer 4

我想我已经解决了。这不是一个好主意。

def make_variant(cls):
    suffix='VARIANT'
    new = type(cls.__name__+suffix, (cls, ), {})
    # new.__repr__ = lambda self: 'HELLO' # Just do whatever needed here
    assert cls.__name__ + suffix not in globals()
    globals()[cls.__name__+suffix] = new # Think twice about this line
    return cls

@make_variant
class A:
    def __init__(self):
        pass

print(AVARIANT(), A())