python装饰器原理其实是闭包
闭包:
在计算机科学中,闭包(英语:Closure),又称词法闭包(Lexical Closure)或函数闭包(function closures),是引用了自由变量的函数。这个被引用的自由变量将和这个函数一同存在,即使已经离开了创造它的环境也不例外。
装饰器作用:
(1)可以在不对被装饰函数做任何修改的前提下,给被装饰函数附加上一些功能。使用@func对do_something函数进行装饰时,我们没有对do_something函数的代码做什么的改变,但是被装饰后的do_something函数却多了开始运行和结束运行的功能。
(2)不改变被装饰函数的调用方式。在被装饰前,我们通过do_something()调用这个函数,被装饰后,还是通过do_something()调用这个函数。
(3)代码更加精简。我们只是用@func装饰了do_something一个函数,但是如果有多个函数需要添加开始运行和结束运行的提示功能,如果不用装饰器,那么就需要对每一个函数进行修改,则工作量和需要修改的代码量……用了装饰器之后,只需要在需要添加这一功能的函数前面添加@func就可以了。
代码示例:
from functools import wraps
def log_handler(func):
@wraps(func)
def wrapper(*args,**kwargs):
# 获取被装饰函数的名称
print("call %s(): " % func.__name__)
# 获取被装饰函数的参数
print("args = {}".format(*args))
# 透传参数,并且一定要return,否则不生效
return func(*args,**kwargs)
# 返回处理装饰函数的引用
return wrapper
@log_handler
def do_some_things(env):
pass
# 带参数带装饰器
def log_with_param(text):
def decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
print('call %s():' % func.__name__)
print('args = {}'.format(*args))
print('log_param = {}'.format(text))
return func(*args, **kwargs)
return wrapper
return decorator
@log_with_param("param")
def test_with_param(p):
print(test_with_param.__name__)
if __name__ == '__main__':
do_some_things("run a programmer")
print("------------------ split line -------------------")
test_with_param("anythings")
输出如下: