【Python之旅】第四篇（一）：Python装饰器

jgfhrt

有时候拿到一个程序接口，需要对其进行扩展，但是又不能修改原来接口的源代码，这时候就需要使用装饰器了。    有下面一个小程序，假如是别人提供给我们的调用接口：

1 2 3 4 5

import time def sayHi():         time.sleep(1)         print 'Hello, I am xpleaf.'

    一般情况下，如果想要计算该程序的执行时间（因为有可能要对该接口进行某些性能上的测试），就需要把以上接口修改为下面这样，同时执行一下：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

程序代码： import time def sayHi():         start = time.time()         time.sleep(1)         print 'Hello, I am xpleaf.'         end = time.time()         print 'This project costs:',end - start sayHi() 执行情况： xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf. This project costs: 1.00269603729

    虽然这样做是达到了扩展该接口的目的，但当出现下面的限制时，也许就不那么适用了：

1.第一个条件仍然是需要对该接口函数进行扩展，如计算执行时间
2.第二个条件则是不能改变原来接口的源代码

    因此这样的话，上面的方法就不适用了，但是尽管如此，我们仍然可以使用下面的替代方法，把上面的程序修改一下：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15

import time def sayHi():         time.sleep(1)         print 'Hello, I am xpleaf.' def time_counter():         start = time.time()         sayHi()         end = time.time()         print 'The project costs:', end - start time_counter()

    执行情况如下：

1 2 3

xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf. The project costs: 1.00348091125

    尽管上面的方法还是可以暂时缓解一下情况，但当出现下面的需求时：

1.第一个条件仍然是需要对该接口函数进行扩展，如计算执行时间

2.第二个条件则是不能改变原来接口的源代码

3.第三个条件则是不能改变原来接口的调用方式

    显然第二个方法不满足第三个条件，因为我们是通过新定义一个函数来进行该操作的，已经改变了原来接口的调用方式，这时候就得使用Python装饰器了，把上面的代码修改为如下：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

import time def time_counter(func):     def wrapper():         start = time.time()         func()         end = time.time()         print 'The project costs:', end - start     return wrapper @time_counter def sayHi():     time.sleep(1)     print 'Hello, I am xpleaf.' sayHi()

    执行情况如下：

1 2 3

xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf. The project costs: 1.00221300125

    可以看到，满足了上面需要的三个条件，而且如果需要对多个函数接口进行相同的扩展时，只需要在定义调用接口的前面加上@function_name就可以了，上面的代码情况，其实又类似下面的情况：

1 2 3

xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf. The project costs: 1.00514888763

    执行情况如下：

1 2 3

xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf. The project costs: 1.00514888763

    当然就装饰器来说，肯定是要使用前面的方法了。