前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >python3:装饰器的5大用法总结

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

作者头像
py3study
发布2020-01-13 11:32:18
2780
发布2020-01-13 11:32:18
举报
文章被收录于专栏:python3

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

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019/07/23 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • python实例分享:工作中撸的一个小工具集,现在流行All_in_One
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档