如何扩展functools.wraps装饰器的功能

Question

我想创建一个新的装饰器来代替@wraps(f)，它可以做任何神奇的@wraps(f)会做的事情以及其他事情。我该怎么做？

具体来说，我有几个装饰器的形式是：

def decorator(f):
    @wraps(f)
    def wrapper(*args, **kwargs):
        # does some stuff in here
        return f(*args, **kwargs)
    if not hasattr(wrapper, '_first_line'):
        wrapper._first_line = inspect.getsourcelines(f)[1]
    return wrapper

我似乎应该能够创建一个像@wraps_with_first_line(f)这样的装饰器，它将完成@wraps(f)和if not hasattr(wrapper, '_first_line'): wrapper._first_line = inspect.getsourcelines(f)[1]正在做的所有工作。

Answer 1

您应该遵循添加指向封装函数的__wrapped__属性的良好做法，而不是添加该封装函数的单个属性。新版本的functools.wraps()会自动执行此操作，但如果您使用的是比3.2更旧版本的Python，则可以轻松地扩展wraps()以添加__wrapped__:

def my_wraps(wrapped, **kwargs):
    def decorator(wrapper):
        functools.update_wrapper(wrapper, wrapped, **kwargs)
        wrapper.__wrapped__ = wrapped
    return decorator

编辑：这里有一个函数，它从一个可能被多重修饰的函数中提取原始函数：

def orig_function(f):
    try:
        while True:
            f = f.__wrapped__
    except AttributeError:
        return f

Answer 2

如果您想要添加的还不是包装对象的属性，那么您可以使用以下方法：

def wraps_with_first_line(f):
    def wrap(wrapper):
        wrapper = wraps(f)(wrapper)
        if not hasattr(wrapper, '_first_line'):
            wrapper._first_line = inspect.getsourcelines(f)[1] 
        return wrapper
    return wrap

如果已经是包装对象的属性，请使用Sven的方法。