装饰器@ 在阅读源码时经常遇到,所以有必要弄明白!
概括的讲,装饰器的作用就是为已经存在的对象添加额外的功能;
本质上,装饰器就是一个返回函数的高阶函数。其接受一个函数作为参数,并返回一个函数。 借助Python的@语法,把装饰器置于函数的定义处,实现功能扩展的目的;
举例
等价于
显然,被“@函数”修饰的函数不再是原来的函数,而是被替换成一个新的东西(取决于装饰器的返回值),即如果装饰器函数的返回值为普通变量,那么被修饰的函数名就变成了变量名;同样,如果装饰器返回的是一个函数的名称,那么被修饰的函数名依然表示一个函数!
1. 需要用装饰器的原因
先来看一个例子:
这个函数的功能是打印出一窜字符窜。如果想要测试执行这个函数用了多长时间,我们可以这样做:
这样能够很好的达到目的。但是想测试一个模块的所有函数的执行时间呢,就得把所有函数中都加入如上时间差的计算方法,这样不太现实。
为了不改变原来的函数,我们可以定义一个函数timeit,将f0()的引用传递给他,然后在timeit中调用f0并进行计时,这样我们就不用修改f0函数而达到目的了:
这样看上去逻辑没有问题,而且可以正常的工作。但却修改了调用部分的代码,原本是f0()调用,现在却成了timeit(f0),如果f0在很多处都被调用了,就需要在很多处修改代码。注:这也是接下来的代码里,在timeit()函数里需要再定义一个wrapper()函数的原因.
如果不改动调用的代码,也就意味着调用f0()需要产生timeit(f0)的效果。我们可以这样做,把timeit(f0)的返回值付给f0,然后直接调用f0(),就不用修改源代码了:
这样,我们需在定义f0以后和调用f0之前,加上f0=timeit(f0),就可以达到目的了。这就是修饰器,看起来像f0被timeit修饰了。
上面的代码,看似没法再精简了,python于是提供了一个特殊的语法来降低字符输入量:
在第12行的@timeit,效果和f0=timeit(f0)一样,而且看上去更有修饰器的感觉。
这就是Python中修饰器的原理。
参考 https://www.tuicool.com/articles/6Z3Mbuj
2. 装饰器使用方法
函数作为一个对象:① 可以被赋值给其他变量,可以作为返回值 ② 也可以被定义在另外一个函数内;
装饰器分类:有参/无参
① 无参:用于生成一个新的装饰器函数
② 有参:先处理这个参数,再生成一个新的装饰器函数,然后对其做装饰;
例
按照装饰器与函数各自是否有参,组合共4种:
① 无参的装饰器 ,包装无参的函数 —— 不需要针对参数进行处理和优化
等价于
② 无参的装饰器 ,包装有参的函数
错误如下(装饰器F并没有对目标函数f1起到任何作用):
改正如下:
等价于
③ 有参的装饰器 ,包装无参的函数
等价于
④ 有参的装饰器 ,包装有参的函数
等价于
另外,函数装饰器是可以多层嵌套的,如
上面程序的执行顺序是里到外,所以它等效于下面这行代码:
3. python内置的3个装饰器staticmethod, classmethod, property
一般来说,要使用某个类的方法,需要先实例化一个对象再调用方法。而使用@staticmethod或@classmethod,就可以不需要实例化,直接类名.方法名()来调用。这有利于组织代码,把某些应该属于某个类的函数给放到那个类里去,同时有利于命名空间的整洁。
既然@staticmethod和@classmethod都可以直接类名.方法名()来调用,那他们有什么区别呢
从它们的使用上来看,
@staticmethod不需要表示自身对象的self和自身类的cls参数,就跟使用函数一样。
@classmethod也不需要self参数,但第一个参数需要是表示自身类的cls参数。
如果在@staticmethod中要调用到这个类的一些属性方法,只能直接类名.属性名或类名.方法名。
而@classmethod因为持有cls参数,可以来调用类的属性,类的方法,实例化对象等,避免硬编码。