装饰器可以简单理解为修饰,为函数添加额外的功能,但是又不用修改原来函数的代码,只需要在原来函数定义前面使用@语法糖即可。它经常用于插入日志等类似的场景,可以达到重复使用而又不修改主要函数代码的目的。
看第一个例子:一个函数
def printName():
print('LiHaiyu')
printName()
输出:LiHaiyu
看第二个例子:简单的装饰器函数
def mydecorator(f):
def wrapper(): # 定义修饰函数
print('Before calling the function')
f()
print('After calling the function')
return wrapper #注意返回函数形成闭包
@mydecorator
def printName():
print('LiHaiyu')
printName()
输出:
提醒:@mydecorator 必须置于被修饰函数定义之前。相当于将 printName 函数传递给装饰器函数,当调用 printName() 时相当于执行装饰器函数。
第三个例子:函数带参数
def mydecorator(f):
def wrapper(*args, **kwargs):
print('Before calling the function')
f(*args, **kwargs)
print('Before calling the function')
return wrapper
@mydecorator
def printName(name):
print(name)
printName("LiHaiyu")
输出:同上
提醒:*args 表示传递一个列表参数,**kwargs 表示传递一个字典参数。所以在 call function 的时候也应该传递参数
第四个例子:装饰器带参数
def mydecorator(msg = 'Message'):
def decorated(f):
def wrapper(*args,**kwargs):
print('The message is : '+ msg)
print('Before calling the function')
f(*args,**kwargs)
print('After calling the function')
return wrapper
return decorated
@mydecorator(msg = 'print name')
def printName(name):
print(name)
printName('LiHaiyu')
输出:
提醒:返回函数
参考链接:https://www.youtube.com/watch?v=mZ5IwFfqvz8