装饰器

最新推荐文章于 2024-03-11 14:12:11 发布

原创最新推荐文章于 2024-03-11 14:12:11 发布 · 305 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

9 篇文章

订阅专栏

本文深入探讨Python装饰器的使用，包括无参数和带参数的装饰器实现，以及如何使用functools.wraps保持被装饰函数的元信息。通过具体示例，展示了装饰器在函数调用前后的操作，以及如何根据装饰器参数改变行为。

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

定义普通带参数的函数加装饰器

. 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')

定义普通不带参数的函数的装饰器(装饰器要求带参数)

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.youkuaiyun.com/lxy210781/article/details/80561334

 

python实例分享：工作中撸的一个小工具集，现在流行All_in_One
里面定义的以下这个装饰器：

def deco_cls_menus(*dargs, **dkw)