python带参数的类装饰器

 

 

# -*- coding: utf-8 -*-
# author：baoshan
# 带参数的类装饰器（和不带参数的类装饰器有很大的不同）
# 类装饰器的实现，必须实现__call__和__init__两个内置函数。
# __init__：不再接收被装饰函数，而是接收传入参数；
# __call__：接收被装饰函数，实现装饰逻辑


class logger(object):
    def __init__(self, level='INFO'):
        self.level = level

    def __call__(self, func):
        def wrapper(*args, **kwargs):
            print("[{level}]: the function {func}() is running...".format(level=self.level, func=func.__name__))
            func(*args, **kwargs)
        return wrapper


@logger(level="WARNING")
def say(something):
    print("say {}!".format(something))


say("hello")

 

运行结果：

[WARNING]: the function say() is running...
say hello!

 

参考自：https://zhuanlan.zhihu.com/p/65968462

 

谢谢！