Python装饰器的理解

Python装饰器是什么

Python装饰器就是一个函数，它的产生是为了在不改变目标函数结构的基础上为其增加功能，返回值是增加功能后的目标函数。

一个简单的装饰器 -为程序添加简单的计时功能

import time def get_run_time(method): def wrapper(): start = time.time() method() end = time.time() return end - start return wrapper @get_run_time def helloworld(): print('helloworld') if __name__ == '__main__': spend = helloworld() print('spend time: {}'.format(spend))

运行结果：

... helloworld spend time: 1.50203704834e-05

目标函数带参数的情况

当然有时候目标函数(在上面的程序中为helloworld函数)是有参数的，下面做一些微小的改动

import time def get_run_time(method): def wrapper(*args, **kwargs): start = time.time() method(*args, **kwargs) end = time.time() return end - start return wrapper @get_run_time def helloworld(name): print('I\'m {}, helloworld'.format(name)) if __name__ == '__main__': spend = helloworld('wk') print('spend time: {}'.format(spend))

运行结果：

... I'm wk, helloworld spend time: 2.09808349609e-05

装饰器函数带参数的情况

既然装饰器是个函数，那它也是可以有参数的，此时只要将上述代码稍加改动即可

import time def get_run_time(decimal): def wrapper_of_wrapper(method): def wrapper(*args, **kwargs): start = time.time() method(*args, **kwargs) end = time.time() return round(end - start, decimal) return wrapper return wrapper_of_wrapper @get_run_time(decimal=16) def helloworld(name): print('I\'m {}, helloworld'.format(name)) if __name__ == '__main__': spend = helloworld('wk') print('spend time: {}'.format(spend)) # 下面这行先不用看，这是下一个标题的内容 print(helloworld.__name__)

增加了控制返回时间精度的功能，其实就是在原基础上多封了一层，最外层是新增的一层，目的是增加一个相对全局变量(能看懂就行了，能力有限，就不扯深的了)，在上述代码中是decimal

执行结果：

... I'm wk, helloworld spend time: 4.60147857666e-05 wrapper

@functools.wraps

装饰器返回的函数虽然结构与目标函数相似，但本质已经发生了变话，返回的是另一个函数，看上面程序中print(helloworld.__name__)的输出为wrapper，再看看functools.wraps的效果

import time import functools def get_run_time(decimal): def wrapper_of_wrapper(method): @functools.wraps(method) def wrapper(*args, **kwargs): start = time.time() method(*args, **kwargs) end = time.time() return round(end - start, decimal) return wrapper return wrapper_of_wrapper @get_run_time(decimal=16) def helloworld(name): print('I\'m {}, helloworld'.format(name)) if __name__ == '__main__': spend = helloworld('wk') print('spend time: {}'.format(spend)) # 对比一下，只增加了@functools.wraps print(helloworld.__name__)

运行结果：

... I'm wk, helloworld spend time: 2.09808349609e-05 helloworld