编程语言
首页 > 编程语言> > python装饰器保留原有函数名称和属性functools.wraps()

python装饰器保留原有函数名称和属性functools.wraps()

作者:互联网

# python装饰器在实现的时候,被装饰后的函数其实已经是另外一个函数了(函数名等函数属性会发生改变),为了不影响,python的functools包中提供了一个叫wraps的decorator来消除这样的副作用。写一个decorator的时候,最好在实现之前加上functools的wrap,它能保留原有的函数的名称和函数属性

# 不加wraps
def my_decorator(f):
    def wrapper(*args, **kwargs):
        print("ing")
        return f(*args, **kwargs)
    return wrapper

@my_decorator
def example():
    print("called example function")
print(example.__name__,example.__doc__)  # wrapper,decorator

# 加上wraps
import functools
def my_decorator(f):
    @functools.wraps(f) #这边加上wraps装饰器
    def wrapper(*args, **kwargs):
        print("ing")
        return f(*args, **kwargs)
    return wrapper

@my_decorator
def example():
    print("called example function")
print(example.__name__,example.__doc__)  # example,Docstring

标签:wraps,python,functools,def,print,example,decorator
来源: https://www.cnblogs.com/weiweivip666/p/15867196.html