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)
这样装饰器就可以处理不确定个参数的函数了。