python装饰器的学习

来源:互联网 发布:jsp调用java显示图片 编辑:程序博客网 时间:2024/06/04 07:45
最近在学习python的过程中看到一些代码中在开头频频出现@符号,例如下代码所示:
DEBUG = Truedef print_args(funcA):    def funcB(a, b):        if DEBUG:            print(a, b)    return funcB@print_argsdef 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 = Truedef print_args(funcA):    def funcB(*pos_args, **keyword_args):        if DEBUG:            print(pos_args, keyword_args)    return funcB@print_argsdef make_list(a, b):    return [a, b]@print_argsdef add(a, b, c=3):    return a + b + cmake_list(1, 2)add(1, 2, 3)

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

0 0
原创粉丝点击