Python 装饰器入门

一、*args 和 **kwargs

• 在作为形参时，*args 将‘多余的’位置参数打包成元组，存储在 args 中；**kwargs 将‘多余的’关键字参数打包成字典，存储在 kwargs 中。
• 作为实参时，它们的作用就是分解，*args 将元组分解成位置参数，并传递给函数；**kwargs 将字典分解成关键字参数，并传递给函数。
• 一个栗子：

def index(x, y):
    print(x, y)

def wrapper(*args, **kwargs):
    index(args, kwargs)

wrapper(1, 2, 3, 4, 5, a=1, b=2)

输出结果：

(1, 2, 3, 4, 5) {‘a’: 1, ‘b’: 2}

这里我们可以看出来，位置参数和关键字参数都分别被打包成了元组和字典。

• 另一个栗子：

def index_1(x, y):
    print('x:', x)
    print('y:', y)

def index_2(a, b):
    print('a:', a)
    print('b:', b)

def wrapper(*args, **kwargs):
    index_1(*args)  # index_1(*(1, 2)) 即相当于 index_1(1, 2)
    index_2(**kwargs)  # index_2(**{'a': 1, 'b': 2}) 即相当于 index_2(a=4, b=5)

wrapper(1, 2, a=4, b=5)

输出结果：

x: 1
y: 2
a: 4
b: 5

二、装饰器

利用装饰器，可以在不修改已有函数的情况下向已有函数中注入代码，使其具备新的功能。一个装饰器可以为多个函数注入代码，一个函数也可以使用多个装饰器装饰。

三、代码示例

def deco1(func):
    def inner1(*args, **kwargs):
        print('deco1 begin')
        func(*args, **kwargs)
        print('deco1 end')
    return inner1

def deco2(func):
    def inner2(*args, **kwargs):
        print('deco2 begin')
        func(*args, **kwargs)
        print('deco2 end')
    return inner2

@deco1
def f1(a, b):
    print('a+b=', a+b)

@deco1
@deco2
def f2(a, b, c):
    print('a+b+c=', a+b+c)

if __name__ == '__main__':
    f1(3, 5)
    f2(1, 3, 5)

输出结果：

deco1 begin
a+b= 8
deco1 end
deco1 begin
deco2 begin
a+b+c= 9
deco2 end
deco1 end

四、代码解释

• 上述代码中定义了两个装饰器，可见其实装饰器实际上就是闭包。
• 在装饰器外层函数的形参列表中只有一个形参，是用来接收所要装饰的函数的函数名。
• 将装饰器内存函数的形参列表写为 *args, **kwargs，表示要装饰的函数可以具有任意形式的形参列表；对应地，调用要装饰的函数时也要将实参列表写成这种形式。
• 在要装饰的函数前写上 @装饰器名，即可将装饰器中的代码注入该函数中。
• 当一个函数前面有多个 @装饰器名时，将按照从后至前的顺序进行装饰。