关于python装饰器的一些体会总结。

首先说下装饰器的目的,在不改变原函数的情况下给函数添加功能。

装饰器的三个要求:

  1. 不能修改被装饰的函数的源代码
  2. 不能修改被装饰的函数的调用方式
  3. 满足1、2的情况下给程序增添功能

假设给下列函数添加个计算运行时间的功能,原函数

然后

看似实现了功能,可是有个问题,此时只能用time_count(test),无法进行time = time_count(test)调用,那就是改变了原函数的调用方式。为啥?函数调用的类型取决于他的返回值,这里没有return。

改进下

说下过程,其实就是,现在内存中申请了个time_count方法放那暂时不用,test函数是作为参数传到装饰器里的,然后调用deco_timer时候返回的是time_count在内存中的地址,原test函数也是个在内存中的地址,吧test这个指针指向time_count,这时候test就不是原来的test了,而是现在的time_count,调用test()就相当于调用time_count(),

上图

带参数的装饰器咋整?

参数放到形参fn里就好了。

如果原函数有返回值咋整?

例如这样的test

其实原来的test函数就是现在那个形参fn在的位置,装饰之后的test指向的是现在的time_count

用res接收一下原test函数的返回值,然后return出去。

顺便提一句:装饰器的要求有两个,在不能修改原函数代码的基础上增加功能,而且不能修改调用方式。

其实功能部分的修改就是time_count,已经完成了。外层函数deco_timer的功能仅仅是把形参fn(原test函数)的指针(内存地址)指向内层的time_count。这样再调用test函数的时候,实际上调用的是time_count函数,因为内存地址已经被修改过了。

最后一个问题,肿么用一个装饰器对不同的函数进行不同的装饰?

简单,给装饰器一个参数用来标记就好。外面再套一层= =,上图

猜你喜欢

转载自blog.csdn.net/houmenghu/article/details/84720493