装饰器,其实就是一个函数,它可以让其他函数在不需要做任何代码变动的前提下额外增加功能,装饰器的返回是一个函数对象。
用一个例子来说明一下什么是函数对象:
从这里可以看到,直接在函数后面加bar(),就是调用,不加就是对象,函数对象有__name__的属性,可以拿到函数的名字:
可以看到:
如果说,我想要在bar执行的时候添加一个log日志,那么我可以通过装饰器来实现,比如说:
然后可以通过@语法糖,就可以得到我想要的效果:
执行的结果是:
这样执行的效果相当于:use_print(bar)()
接下来,我们继续往下走:如果装饰器本身包含参数呢?
通过@语法糖修饰某个方法后为:
执行的结果如下:(这里可以看到:只有在参数为符合条件的时候,才会执行特定的语句,否则就只会执行foo3自己的逻辑)
添加了参数之后,这样的执行效果相当于:
foo2=use_logging_2("ERROR")(foo2)
foo2()
完成到这样的程度,还有哪些问题吗?
我们可以尝试去获取一下foo2.__name__的值,可以看到在使用了装饰器之后,得到foo2.__name__的值为:
可见foo2的name属性发生了变化,需要把原始函数的__name__等属性复制到wrapper()函数中,否则,有些依赖函数签名的代码执行就会出错,
不需要编写wrapper.__name__ = func.__name__这样的代码,Python内置的functools.wraps就是干这个事的,所以,一个完整的decorator的写法如下,代码修改如下:
然后再去获取@use_logging_wraps()装饰的具体方法,之后获取其函数对象的__name__属性,就可以得到:
结果为:
备注:Python的log模块的使用: