在写模块时,由于可以把很多细节封装起来,很喜欢用到”装饰器(decorator)“这个玩意儿,但关于它的用法都是从网上抄的,每次用时,特别是写带参装饰器,都要去参考别人的”模板“。
今天干脆来探个究竟。
Hello
这里写了一个装饰器”upper“,将返回值转成大写
def upper(f):
def func():
resp = f()
return resp.upper()
return func
@upper
def hello():
return 'hello, world'
if __name__ == '__main__':
print(hello())
输出:
HELLO, WORLD
实际上,装饰器就是一个语法糖,它的真面目是:
def upper(f):
def func():
resp = f()
return resp.upper()
return func
def hello():
return 'hello, world'
hello = upper(hello) # 装饰器的真面目
if __name__ == '__main__':
print(hello())
多个装饰器
这里又写了一个装饰器”tail“,用于给函数的返回值添加一个小尾巴
def upper(f):
def func():
resp = f()
return resp.upper()
return func
def tail(f):
def func():
resp = f()
return resp + ',hahaha~'
return func
@tail
@upper
def hello():
return 'hello, world'
if __name__ == '__main__':
print(hello())
输出:
HELLO, WORLD,hahaha~
将两个装饰器的位置交换后,输出结果是不一样的,上述代码中,先进行了”upper“操作,再进行的”tail“操作,所以小尾巴并没有转成大写。
按照上一节中的示例,不难看出多个装饰器的原理:
hello = tail(upper(hello))
带参装饰器
如果想为函数添加自定义的小尾巴,就得需要带参装饰器了:
def tail(s):
def warpper(f):
def func():
resp = f()
return resp + s
return func
return warpper
@tail(', wow~')
def hello():
return 'hello, world'
if __name__ == '__main__':
print(hello())
按照装饰器的原理,将语法糖展开,它的真面目是这样的:
hello = tail(', wow~')(hello)
一步一步展开:
- tail(', wow~') 执行后,返回warpper函数
hello = warpper(hello)
- warpper(hello)执行后,返回了它的真面目,func,并赋值给了hello
hello = func
装饰带参函数
上述例子中,均是”hello world“,那如果想”hello,baby“呢,这就得hello函数带参数了
又把上面的代码改了改:
def tail(s):
def warpper(f):
def func(p):
resp = f(p)
return resp + s
return func
return warpper
@tail(', wow~')
def hello(p):
return 'hello, {}'.format(p)
if __name__ == '__main__':
print(hello('baby'))
输出:
hello, baby, wow~
上例中,只是简单地将func的声明改成了和hello的声明一致,
但这也只能用于明确知道被装饰函数的声明,不具有通用性。
例如,若将上例中的”tail“用于装饰下面不带参的goodnight函数,这就报错啦:
@tail(', wow~')
def goodnight():
return 'goodnight'
输出:
TypeError: func() missing 1 required positional argument: 'p'
为了解决这个问题,又修改下代码,改用”*args, **kwargs“来传递参数:
def tail(s):
def warpper(f):
def func(*args, **kwargs):
resp = f(*args, **kwargs)
return resp + s
return func
return warpper
@tail(', wow~')
def goodnight():
return 'goodnight'
if __name__ == '__main__':
print(goodnight())