python3-装饰器

Posted 2020-08-22

# Author: Aaron Fan


"""
装饰器（别名：语法糖）：
定义：
    本质是函数（装饰其它函数），就是为其它函数添加一些附件的功能
原则：
    1、不能修改被装饰的函数的源代码
    2、不能修改被装饰的函数的调用方式
    3、装饰器存在对被装饰的函数是完全透明的（就是被装饰的函数完全感觉不到装饰器的存在）

实现装饰器的只是储备：
1、函数就是变量
2、高阶函数
3、嵌套函数

高阶函数 + 嵌套函数 == 装饰器
"""
import time


def timer(func):
    """
    一个最简单的装饰器：高阶函数 + 嵌套函数 == 装饰器
    计算一个函数的执行时间
    """
    def deco(*args,**kwargs):
        start_time = time.time()
        func(*args,**kwargs)  #这里运行传进来的实参（一个被装饰的函数）的函数体，这个示例中是指test1和test2
        stop_time = time.time()
        print("The func run time is %s" % (stop_time - start_time))
    return deco #返回deco的内存地址


@timer  # 给test1这个函数，添加一个装饰器，等同于 test1=timer(test1)=deco
        # 注意timer的return是deco，所以test1=timer(test1)，就是调用timer函数，并把deco的内存地址赋值给test1
        # 所以以后都会不在直接执行这个用def定义的最原始的test1()这个函数
        # 程序后面的所调用的test1()，其实执行的是deco()，通过deco来执行test1里面的函数体
        # 这里的deco()仅仅是timer()的一个嵌套函数，相当于timer的局部变量是无法直接在全局引用的
        # 但是timer这个高阶函数，返回了deco的内存地址，所以通过test1=timer(test1)之后，test1就等于deco的内存地址了，就可以直接使用test1()了
            #此时这个test()其实就是deco()，而不再是最原始的定的那个def test()
def test1():
    time.sleep(1)
    print("in the test1")


@timer
def test2(name1,name2):
    time.sleep(1)
    print("in the test2")
    print(name1,name2)


test1() #这个现在执行的其实已经不是test1这个函数了，而是执行的deco函数
test2(‘James‘,‘Aaron‘)