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

py3study

发布于 2020-01-13 11:32:18

2780

发布于 2020-01-13 11:32:18

文章被收录于专栏：python3

装饰器，顾名思义，就是给函数作装饰作用的，相当于只是给函数化了一个装，并没有给他添加额外的代码，但是却增加了新功能，就如同我们在冬天很冷的情况下，想要变暖和（添加新功能），但是我们不是生活在武侠小说里有内功，肿么办？我们可以穿一件羽绒服来保暖（这件羽绒服就相当于是装饰器了）装饰器的内部实现原理就不讲了，这里只是装饰器的5种形式，搞定了，其本就完全明白了装饰器的的有用法。 1. 定义普通不带参数的函数的装饰器 eg:要求在调用函数testfun前后输出‘上有天堂’和‘下有苏杭’

定义如下：

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. 定义普通带参数的函数加装饰器

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. 定义普通不带参数的函数的装饰器(装饰器要求带参数)

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. 定义普通带参数的函数的装饰器(装饰器要求带参数)

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,多了以下两行，请留意：

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