[转] functools.wraps定义函数装饰器

Posted leokale-zz

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[转] functools.wraps定义函数装饰器相关的知识,希望对你有一定的参考价值。

转自:https://www.cnblogs.com/fcyworld/p/6239951.html

 

  装饰器(decorator)是干嘛的?

  对于受到封装的原函数来说,装饰器能够在那个函数执行前或者执行后分别运行一些代码,使得可以再装饰器里面访问并修改原函数的参数以及返回值,以实现约束定义、调试程序、注册函数等目标。装饰器一般返回一个包装器(wrapper),而functools.wraps就是装饰包装器的装饰器。

 

先来看一个不使用functools.wraps的装饰器例子:

def tracer(func):
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        print(%s(%r,%r)->%r%(func.__name__,args,kwargs,result))
        return result
    return wrapper

@tracer
def fibonacci(n):
    if n in (0,1):
        return n
    return (fibonacci(n-1)+fibonacci(n-2))


fibonacci(3)
print(fibonacci)
print(help:)
help(fibonacci)

输出结果:

技术图片

可以看到,装饰器完全可以正常工作。。。

但是,函数的名字变成装饰器中的包装器了!!!help内置函数也失效了

也就是说,原函数的属性失效了。

 

如果想要保留原函数的属性,就可以用到functools.wraps了:

from functools import wraps
def tracer(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        print(%s(%r,%r)->%r%(func.__name__,args,kwargs,result))
        return result
    return wrapper

@tracer
def fibonacci(n):
    if n in (0,1):
        return n
    return (fibonacci(n-1)+fibonacci(n-2))


fibonacci(3)
print(fibonacci)
print(help:)
help(fibonacci)

输出结果:

技术图片

以上是关于[转] functools.wraps定义函数装饰器的主要内容,如果未能解决你的问题,请参考以下文章

python functools.wraps装饰器模块

python functools.wraps装饰器模块

@functools.wraps(func)

functools.wraps 等价于类装饰器

学习笔记2-functools.wraps 装饰器

标准库functools.wraps的使用方法