python装饰器的学习

最近在学习python的过程中看到一些代码中在开头频频出现@符号，例如下代码所示：

DEBUG = True

def print_args(funcA):
    def funcB(a, b):
        if DEBUG:
            print(a, b)
    return funcB

@print_args
def make_list(a, b):
    return [a, b]

make_list(1, 2)

这样的结构其实是python中的装饰器，装饰器的本质其实很简单：函数。

在python中函数实际上就是一个特殊的变量，函数可以作为另一个函数的参数，可以作为另一个函数的返回值，同时函数还可以使用外层空间的变量。

函数在这样的三个特性下，装饰器就是：以函数为参数，并返回一个函数的函数。换句话说，装饰器就是一个特殊的函数，函数A作为参数输入，装饰器经过处理返回函数B，这里装饰器的作用就是对输入函数进行加强和修改。

以上面的代码为例：print_args()就是一个装饰器，在代码中用@调用，funcA作为输入参数，funcB是装饰器的返回参数。funcA在这里只构建一个list，这时装饰器对funcA进行功能添加，其返回的funcB增加了打印该list的功能。

如果装饰器处理的函数参数个数不确定怎么办，这里也有解决的办法：

在函数中，参数分为位置参数和关键字参数，位置参数传递顺序不能变，关键字参数位置可以变。如下函数所示：

def function(a, b, c, d=2, e=True):
    pass

a、b、c是位置参数，传参顺序不能改变，d、e是关键字参数，传参顺序可以改变。

所以可以这样定义函数参数：

def function(*args，**kwargs):
    pass

args是元组：(a, b, c)，kwargs是字典：{"d": 2, "e": True}。

这样定义就可以传递未知个数参数。

那么装饰器可以这样写：

DEBUG = True

def print_args(funcA):
    def funcB(*pos_args, **keyword_args):
        if DEBUG:
            print(pos_args, keyword_args)
    return funcB

@print_args
def make_list(a, b):
    return [a, b]

@print_args
def add(a, b, c=3):
    return a + b + c

make_list(1, 2)
add(1, 2, 3)

这样装饰器就可以处理不确定个参数的函数了。