Python之装饰器

# 装饰器功能说明
"""
    python中的装饰器(decorator)一般采用语法糖的形式，是一种语法格式。
        比如：@classmethod，@staticmethod，@property，@xxx.setter，@wraps()，@func_name等都是python中的装饰器。
    装饰器，装饰的对象是函数或者方法。各种装饰器的作用都是一样的：改变被装饰函数或者方法的功能，性质。
"""
import time


def how_much_time(func):
    print("how_much_time函数开始运行")

    def inner():
        t_start = time.time()
        func()
        t_end = time.time()
        print(f"一共花费了{t_end - t_start:.6f}秒时间")

    return inner


# @how_much_time等价于sleep_1s = how_much_time(sleep_1s)
@how_much_time
def sleep_1s():
    """
    how_much_time函数开始运行
    1秒结束了
    一共花费了1.007465秒时间
    """
    time.sleep(1)
    print("%d秒结束了" % (1,))


sleep_1s()  # 执行


def mylog(func):
    print("mylog函数开始运行")

    def inner_2():
        print("start")
        func()
        print("end")

    return inner_2


# 多个装饰器demo1，等价于mylog(how_much_time(sleep_more))
@mylog
@how_much_time
def sleep_more():
    """
    how_much_time函数开始运行
    mylog函数开始运行
    start
    2秒结束了
    一共花费了2.003144秒时间
    end
    """
    time.sleep(2)
    print("%d秒结束了" % (2,))


sleep_more()


def add(func):
    def inner():
        x = func()
        x = x + 1
        print("add方法执行,x:", x)
        return x

    return inner


def cube(func):
    def inner():
        x = func()
        x = x * x * x
        print("cube方法执行,x:", x)
        return x

    return inner


def square(func):
    def inner():
        x = func()
        x = x * x
        print("square方法执行,x:", x)
        return x

    return inner


# 多装饰器demo2，先square，再cube，最后add
@add
@cube
@square
def test():
    """
    square方法执行,x: 4
    cube方法执行,x: 64
    add方法执行,x: 65
    65
    """
    return 2


print(test())

# 带参数的装饰器
print("", sep="\n")
print("#" * 5, "带参数的装饰器", "#" * 5, sep=" ")


def logger(msg):
    def time_master(func):
        def inner():
            start = time.time()
            func()
            end = time.time()
            print(f"{msg}函数执行耗时{end - start:.4f}")

        return inner

    return time_master


@logger(msg="my_fun")
def my_fun():
    """
    my_fun正在执行
    my_fun函数执行耗时1.0109
    """
    print("my_fun正在执行")
    time.sleep(1)


my_fun()  # 等于没有装饰器时的调用 logger(msg="my_fun")(my_fun)()