本文共 1948 字,大约阅读时间需要 6 分钟。
有时候拿到一个程序接口,需要对其进行扩展,但是又不能修改原来接口的源代码,这时候就需要使用装饰器了。
有下面一个小程序,假如是别人提供给我们的调用接口:
import timedef sayHi(): time.sleep(1) print 'Hello, I am xpleaf.'
一般情况下,如果想要计算该程序的执行时间(因为有可能要对该接口进行某些性能上的测试),就需要把以上接口修改为下面这样,同时执行一下:
程序代码:import timedef sayHi(): start = time.time() time.sleep(1) print 'Hello, I am xpleaf.' end = time.time() print 'This project costs:',end - startsayHi()执行情况:xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf.This project costs: 1.00269603729
虽然这样做是达到了扩展该接口的目的,但当出现下面的限制时,也许就不那么适用了:
1.第一个条件仍然是需要对该接口函数进行扩展,如计算执行时间
2.第二个条件则是不能改变原来接口的源代码
因此这样的话,上面的方法就不适用了,但是尽管如此,我们仍然可以使用下面的替代方法,把上面的程序修改一下:
import timedef sayHi(): time.sleep(1) print 'Hello, I am xpleaf.'def time_counter(): start = time.time() sayHi() end = time.time() print 'The project costs:', end - starttime_counter()
执行情况如下:
xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf.The project costs: 1.00348091125
尽管上面的方法还是可以暂时缓解一下情况,但当出现下面的需求时:
1.第一个条件仍然是需要对该接口函数进行扩展,如计算执行时间
2.第二个条件则是不能改变原来接口的源代码
3.第三个条件则是不能改变原来接口的调用方式
显然第二个方法不满足第三个条件,因为我们是通过新定义一个函数来进行该操作的,已经改变了原来接口的调用方式,这时候就得使用Python装饰器了,把上面的代码修改为如下:
import timedef time_counter(func): def wrapper(): start = time.time() func() end = time.time() print 'The project costs:', end - start return wrapper@time_counterdef sayHi(): time.sleep(1) print 'Hello, I am xpleaf.'sayHi()
执行情况如下:
xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf.The project costs: 1.00221300125
可以看到,满足了上面需要的三个条件,而且如果需要对多个函数接口进行相同的扩展时,只需要在定义调用接口的前面加上@function_name就可以了,上面的代码情况,其实又类似下面的情况:
xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf.The project costs: 1.00514888763
执行情况如下:
xpleaf@xpleaf-machine:/mnt/hgfs/Python/day4/blog$ python wraper.py Hello, I am xpleaf.The project costs: 1.00514888763
当然就装饰器来说,肯定是要使用前面的方法了。
转载地址:http://vbyao.baihongyu.com/