此文为个人总结的廖雪峰学习笔记,详细知识见廖雪峰。
装饰器用途:
要增强now()函数的功能,又不希望修改now()函数的定义,
这种在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator)
本质上,装饰器是一个接受函数作为参数的返回函数。
装饰器的常见定义方式(两种):
①
def decorator(func):
def wrapper(*arg,**kw):
print('call %s():' % func.__name__)
return func(*arg,**kw)
return wrapper
@log()
def f():
print(2)
这种定义方式实现的功能较为简单,不需额外自定义log文本,可以在不改变原函数定义前提下,给原函数增加执行前输出日志的功能。
调用方式是,直接输入原函数f()
即可。
②
def log(text):
def decorator(func):
def wrapper(*arg,**kw):
print(text)
return func()
return wrapper
return decorator
@log('abc')
def f():
print(2)
这种定义方式可以自定义log中的text文本,这时装饰器需要再套一层,见上面代码。调用也是直接敲f()
即可。
**上两种都可以看成原函数变成了wrapper(*arg,**kw)。
有人有疑问,下面也可实现自定义log中的text文本功能,为啥还需嵌套一层?
def log(text):
def wrapper(fun):
print('%s %s():' % (text, fun.__name__))
return (fun)
return wrapper
@log('excute')
def now():
print('2020-02-04')
now()
虽然也可以输出:
excute now():
2020-02-04
但是,这样定义的now()不能带参数,因为参数传不进wrapper里,所以若要带参数,还得按上面两种常见定义。