【Python】装饰器

一、装饰器的作用

装饰器能够为已经存在的对象添加额外的功能。

二、什么是装饰器

装饰器本质是一个python函数，它可以让其他函数在不需要做任何代码变动的前提下增加额外功能，装饰器的返回值也是一个函数对象。

三、装饰器的应用场景

插入日志、性能测试、事务处理、缓存、权限校验等。

四、写一个原始装饰器

例如，我写一个描述我在吃饭的函数，但是我不想做饭也不想洗碗，就可以通过一个装饰器来为函数添加功能：

def eat():
    print("我正在吃饭")

def test1(func):
    def test2():
        print("帮你把饭做好")
        func()
        print("洗碗")
    return test2

eat = test1(eat)
eat()

test1是一个高阶函数，它把函数作为参数传给函数，在test1函数体中，又定义了一个输入所需信息的函数test2，test1的返回值是函数test2。

运行后返回：

需求实现了。eat函数只有吃饭功能，但是test1函数为eat函数添加了额外的功能，test1()就是一个装饰器。

五、改进装饰器

上面的装饰器test1中，我们将eat()作为参数传给test1()，为了方便使用，将传参这件事交给计算机来做，加入一个@符号+功能函数名即可：

@test1 # 装饰器
def eat():
    print("我在吃饭")

def test1(func): # 定义一个额外功能（装饰器）
    def test2():
        print("帮你把饭做好")
        func()
        print("洗碗")
    return test2

@test1 # 装饰器
def eat():
    print("我在吃饭")

eat()

 

六、函数名的修改

为eat()函数添加装饰器后，增加了额外功能，查看添加装饰器后的函数名：

print(eat.__name__)

eat函数名字已经被改变 。如果不希望函数名称改变，可以使用wraps函数来用原始函数名包装高阶函数名，wraps()其实也是一个装饰器。

from functools import wraps 

def test1(func): # 定义一个额外功能（装饰器）
    @wraps(func) # 使用func函数来包装test2
    def test2():
        print("帮你把饭做好")
        func()
        print("洗碗")
    return test2

@test1 # 装饰器
def eat():
    print("我在吃饭")

# eat()
print(eat.__name__)

wraps函数需要从包functools中导入，此时再查看使用了装饰器的eat函数，名称已经变成了eat。

七、为装饰器传参

当添加装饰器的函数接收参数时，需要为装饰器设置接收参数，在和函数对应的位置加入*args,**kwargs即可：

def test1(func):
    @wraps(func)
    def test2(*args,**kwargs):
        print("帮你把饭做好")
        func(*args,**kwargs)
        print("洗碗")

    return test2