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

posted @ 2022-02-07 10:48  我在路上回头看  阅读(94)  评论(0编辑  收藏  举报