python--装饰器deorator--高级

python–装饰器deorator-- 高级

"""
如何优化你的装饰器
   1、wrapt是一个功能非常完善的包，用于实现各种你想到或者你没想到的装饰器。
   2、使用wrapt实现的装饰器你不需要担心之前inspect中遇到的所有问题，因为它都帮你处理了，甚至inspect.getsource(func)也准确无误
"""
"""
1、无参数
   注要pip
"""
import wrapt
"""
1、函数签名是固定的，必须是(wrapped, instance, args, kwargs)，
2、注意第二个参数instance是必须的，就算你不用它
3、args和kwargs也是固定的，注意前面没有星号。在装饰器内部调用原函数时才带星号。
"""
# without argument in decorator
@wrapt.decorator
def logging(wrapped, instance, args, kwargs):  # instance is must
    print("[DEBUG]: enter {}()".format(wrapped.__name__))
    return wrapped(*args, **kwargs)


@logging
def say(something): pass


"""
带参数
1"""


def logging(level):
    @wrapt.decorator
    def wrapper(wrapped, instance, args, kwargs):
        print("[{}]: enter {}()".format(level, wrapped.__name__))
        return wrapped(*args, **kwargs)

    return wrapper


@logging(level="INFO")
def do(work): pass