关于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函数，因为内存地址已经被修改过了。

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

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