"""
1.什么是装饰器：
    装饰函数或者类，对函数或类进行功能的扩展
    函数装饰器就是一个闭包函数
    装饰器包含函数装饰器和类装饰器
    函数装饰器带参数实质就是在外面在包裹一个函数
    当有多个装饰器时先执行最里面的装饰器，最后执行最外面的（就近原则）
"""
import logging
import time

def runTime(func):
    def wrapper(*args,**kwargs):
        start_time = time.time()
        #执行原函数功能
        func(*args,**kwargs)
        time.sleep(1)
        end_time = time.time()
        const = end_time - start_time
        print(f"统计{func.__name__}函数耗时{const}秒")
    return wrapper


def testLog(func):
    def get_logger():
        logger = logging.getLogger("logger")
        logger.setLevel("DEBUG")
        if not logger.handlers:
            path = time.strftime("%Y_%m_%d_%H_%M_%S", time.localtime())
            fh = logging.FileHandler(path + ".log", encoding="utf-8")
            fmt = logging.Formatter("%(asctime)s %(filename)s:%(lineno)s %(message)s", datefmt="%Y/%m/%d %H:%M:%S")
            fh.setFormatter(fmt)
            logger.addHandler(fh)
        return logger

    def wrapper(*args, **kwargs):
        try:
            func(*args, **kwargs)
            get_logger().info("执行成功")
        except Exception as e:
            get_logger().error(str(e))

    return wrapper

@runTime
def welcome_vip():
    print("欢迎来到vip")

@runTime
def fun2(a,b):
    sum = a + b
    print(f"执行函数2,sum={sum}")

@runTime
def fun3(a,b,c):
    sum = a + b + c
    print(f"执行函数3,sum={sum}")

@testLog
def fun4(a,b,c):
    sum = a+ b + c
    print(f"执行函数4,sum={sum}")

if __name__ == '__main__':
    # welcome_vip()
    # fun2(1,2)
    # fun3(1,2,3)
    fun4(1,2)

"""类装饰器"""
class Demo:
    def __init__(self,func):
        self.func = func

    #实例()，自动执行__call__方法
    def __call__(self, *args, **kwargs):
        print("执行__call__方法")
        #执行被装饰的函数功能
        self.func()
        #扩展功能
        print(f"执行了{self.func.__name__}扩展功能")

@Demo
def test():
    print("执行test函数")

#原函数功能+扩展功能
# test = Demo(test)
# test()

def a(func):
    def wrapper():
        func()
        print("a")
    return wrapper

def b(func):
    def wrapper():
        func()
        print("b")
    return wrapper

def c(func):
    def wrapper():
        func()
        print("c")
    return wrapper

@c
@b
@a
def test():
    pass

test()
#结果a b c