装饰器,顾名思义,就是给函数作装饰作用的,相当于只是给函数化了一个装,并没有给他添加额外的代码,但是却增加了新功能,就如同我们在冬天很冷的情况下,想要变暖和(添加新功能),但是我们不是生活在武侠小说里有内功,肿么办?我们可以穿一件羽绒服来保暖(这件羽绒服就相当于是装饰器了) 装饰器的内部实现原理就不讲了,这里只是装饰器的5种形式,搞定了,其本就完全明白了装饰器的的有用法。 1. 定义普通不带参数的函数的装饰器 eg:要求在调用函数testfun前后输出‘上有天堂’和‘下有苏杭’
定义如下:
代码语言:javascript复制def basedeco(func):
def wrapper():
print('上有天堂')
output = func()
print('下有苏杭')
return output
return wrapper
@basedeco
def testfun():
'''这是一个测试函数'''
print('我就是我,不一样的烟火')
print(testfun.__doc__)
if __name__ == '__main__':
testfun()
2. 定义普通带参数的函数加装饰器
代码语言:javascript复制def basedeco(func):
def wrapper(*args, **kw):
print('上有天堂')
output = func(*args, **kw)
print('下有苏杭')
return output
return wrapper
@basedeco
def testfun(name):
'''这是一个测试函数'''
print(f'{name}说:我就是我,不一样的烟火')
print(testfun.__doc__)
if __name__ == '__main__':
testfun('Zhangguorong')
3. 定义普通不带参数的函数的装饰器(装饰器要求带参数)
代码语言:javascript复制def strongdeco(*dargs, **dkw)
def basedeco(func):
def wrapper():
if dargs[0] == 1:
print('当装饰器参数=1时,这句话是对的')
if dargs[0] == 0:
print('当装饰器参数=0时,这句话是错的')
print('上有天堂')
output = func()
print('下有苏杭')
return output
return wrapper
return basedeco
@strongdeco(1)
def testfun():
'''这是一个测试函数'''
print('我就是我,不一样的烟火')
print(testfun.__doc__)
if __name__ == '__main__':
testfun()
4. 定义普通带参数的函数的装饰器(装饰器要求带参数)
代码语言:javascript复制def strongdeco(*dargs, **dkw):
def basedeco(func):
def wrapper(*args, **kw):
if dargs[0] == 1:
print('当装饰器参数=1时,这句话是对的')
if dargs[0] == 0:
print('当装饰器参数=0时,这句话是错的')
print('上有天堂')
output = func(*args, **kw)
print('下有苏杭')
return output
return wrapper
return basedeco
@strongdeco(0)
def testfun(name):
'''这是一个测试'''
print(f'{name}说:我就是我,不一样的烟火')
print(testfun.__doc__)
if __name__ == '__main__':
testfun('Zhangguorong')
5. 在应用装饰器的时候,你会发现,testfun函数对象的一些基本属性在调用时失效了, 返回结果均为None.
如要让这些函数对象属性可用,可以应用functools.wraps,多了以下两行,请留意:
代码语言:javascript复制from functools import wraps
def strongdeco(*dargs, **dkw):
from functools import wraps
def basedeco(func):
@wraps(func)
def wrapper(*args, **kw):
if dargs[0] == 1:
print('当装饰器参数=1时,这句话是对的')
if dargs[0] == 0:
print('当装饰器参数=0时,这句话是错的')
print('上有天堂')
output = func(*args, **kw)
print('下有苏杭')
return output
return wrapper
return basedeco
@strongdeco(0)
def testfun(name):
'''这是一个测试'''
print(f'{name}:我就是我,不一样的烟火')
print(testfun.__doc__)
if __name__ == '__main__':
testfun('Zhangguorong')
至此,装饰器应该就是这些东东了吧。
具体的应用可参考我之前的一篇blog:
https://blog.csdn.net/lxy210781/article/details/80561334
python实例分享:工作中撸的一个小工具集,现在流行All_in_One
里面定义的以下这个装饰器:
def deco_cls_menus(*dargs, **dkw)