functools.wraps 等价于类装饰器

Question

当我们装饰函数时，我们使用functools.wraps 使装饰的函数看起来像原来的一样。

当我们想装饰班级时，有什么可以做的吗？

def some_class_decorator(cls_to_decorate):
    class Wrapper(cls_to_decorate):
        """Some Wrapper not important doc."""
        pass
    return Wrapper


@some_class_decorator
class MainClass:
    """MainClass important doc."""
    pass


help(MainClass)

输出：

class Wrapper(MainClass)
 |  Some Wrapper not important doc.
 |  
 |  Method resolution order:
 |      Wrapper
 |      MainClass
 |      builtins.object
 |  
 # ... no MainClass important doc below.

我尝试为基于functools.wraps 源代码的类装饰器编写等效包装，但我的实现不正确：

import functools


def wraps_cls(original_cls):
    def wrapper(wrapper_cls):
        """Update wrapper_cls to look like original_cls."""
        for attr in functools.WRAPPER_ASSIGNMENTS:
            try:
                value = getattr(original_cls, attr)
            except AttributeError:
                pass
            else:
                setattr(wrapper_cls, attr, value)
        return wrapper_cls
    return wrapper


def some_class_decorator(cls_to_decorate):
    @wraps_cls(cls_to_decorate)
    class Wrapper(cls_to_decorate):
        """Some Wrapper not important doc."""
        pass
    return Wrapper


@some_class_decorator
class MainClass:
    """MainClass important doc."""
    pass


help(MainClass)

输出：

class MainClass(MainClass)
 |  MainClass important doc.
 |  
 |  Method resolution order:
 |      MainClass
 |      MainClass
 |      builtins.object
 |  
 # ... MainClass doc is here but "Method resolution order" is broken.

有没有用未修饰的 MainClass 帮助输出完全替换修饰的 MainClass 帮助输出？

Answer 1

不，没有，假设你的装饰器真的像some_class_decorator 那样子类化了包装类。 help 的输出由 pydoc.Helper 类定义，对于类最终调用 pydoc.text.docclass，其中包含以下代码：

# List the mro, if non-trivial.
mro = deque(inspect.getmro(object))
if len(mro) > 2:
    push("Method resolution order:")
    for base in mro:
        push('    ' + makename(base))
    push('')

您可以看到它是硬编码的，以显示班级的真实 MRO。这是应该的。您上一个示例中显示的 MRO 没有“损坏”，它是正确的。通过使您的包装类从被包装类继承，您向继承层次结构中添加了一个附加类。向 MRO 展示忽略了这一点会产生误导，因为那里确实有一个类。在您的示例中，此包装器类不提供其自身的任何行为，但实际的包装器类会（或者您为什么要进行包装？），并且您想知道包装器的哪些行为类和包装类中的哪个。

如果您愿意，您可以制作一个装饰器，使用从原始名称派生的名称动态重命名包装类，因此 MRO 会在适当的位置显示类似 DecoratorWrapper_of_MainClass 的内容。这是否比仅仅拥有Wrapper 更具可读性还有待商榷。

Answer 2

哦，我想我现在明白你想要达到什么目的了。

您想使用类装饰器从“包装器”附加新方法。

这是一个工作示例：

class Wrapper(object):
    """Some Wrapper not important doc."""
    def another_method(self):
        """Another method."""
        print 'Another method'


def some_class_decorator(cls_to_decorate):
    return type(cls_to_decorate.__name__, cls_to_decorate.__bases__, dict
        (cls_to_decorate.__dict__, another_method=Wrapper.__dict__['another_method']))


class MainClass(object):
    """MainClass important doc."""
    def method(self):
        """A method."""
        print "A method"


help(MainClass)
_MainClass = some_class_decorator(MainClass)
help(_MainClass)
_MainClass().another_method()
MainClass().another_method()

这个例子创建了一个新的类而不修改旧的类。

但我猜你也可以只在旧类中注入给定的方法，并对其进行适当的修改。