简单装饰器

最新推荐文章于 2022-11-22 17:12:35 发布

nongcunqq

最新推荐文章于 2022-11-22 17:12:35 发布

阅读量90

点赞数 1

分类专栏： python

本文链接：https://blog.csdn.net/nongcunqq/article/details/112783771

版权

python 专栏收录该内容

339 篇文章 0 订阅

订阅专栏

通用装饰器模型
在这里插入图片描述

在这里插入图片描述

def use_logging(func):

    def wrapper(*args,**kwargs):
        print(f'{func.__name__} is running')
        return func(*args, **kwargs)
    return wrapper

@use_logging
def bar(a,b):
    print(a)
    print(b)
    print('this is bar')

bar('a', 'b')

装饰器设置参数

def use_logging(level):
    def decorator(func):
        def wrapper(*args,**kwargs):
            if level == 'warn':
                print(f'{func.__name__} is name')
                return func(*args,**kwargs)
        return wrapper
    return decorator

@use_logging(level='warn')
def foo(name='foo'):
    print('name',name)

foo()

在这里插入图片描述

保留函数的元信息

from functools import wraps

def use_logging(level):
    def decorator(func):
        @wraps(func)
        def wrapper(*args,**kwargs):
            if level == 'warn':
                print(f'{func.__name__} is name')
                return func(*args,**kwargs)
        return wrapper
    return decorator

@use_logging(level='warn')
def foo(name='foo'):
    'haha---------------->'
    print('name',name)

foo()
print(foo.__name__)
print(foo.__doc__)