python3:装饰器的5大用法总结

2020-01-13 11:32:18 浏览数 (1)

装饰器,顾名思义,就是给函数作装饰作用的,相当于只是给函数化了一个装,并没有给他添加额外的代码,但是却增加了新功能,就如同我们在冬天很冷的情况下,想要变暖和(添加新功能),但是我们不是生活在武侠小说里有内功,肿么办?我们可以穿一件羽绒服来保暖(这件羽绒服就相当于是装饰器了) 装饰器的内部实现原理就不讲了,这里只是装饰器的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)

0 人点赞