python学习杂记--装饰器

装饰器可以用来在不改变原函数的基础上，为原函数增添功能

示例如下：

import time

def decorator1(function):
    def wrapper(*args,**kwargs):
        function(*args,**kwargs)
        print("------------ running function : %s  --------------"%function.__name__)
    return wrapper

def decorator2(function):
    def wrapper(*args,**kwargs):
        starttime = time.time()
        function(*args,**kwargs)
        endtime = time.time()
        total_time = (endtime - starttime)
        print("------------ the function spend time is %s --------------"%total_time)
    return wrapper

@decorator1
@decorator2
def func_test1(arg1,arg2):
    print("hello")
    time.sleep(2)
    print("world")
    print("arg1 is %s , arg2 is %s"%(arg1,arg2))

@decorator2
def func_test2(arg1,arg2,arg3):
    print("hello")
    time.sleep(2)
    print("world")
    print("arg1 is %s , arg2 is %s , arg3 is %s" % (arg1, arg2,arg3))



if __name__ == '__main__':
    func_test1("A","B")
    func_test2("C","D","E")

对应输出结果为：

hello
world
arg1 is A , arg2 is B
------------ the function spend time is 2.000811815261841 --------------
------------ running function : wrapper  --------------
hello
world
arg1 is C , arg2 is D , arg3 is E
------------ the function spend time is 2.000079393386841 --------------

这个例子要注意的是3点

1.定义装饰器的语法：

如果要装饰的函数是不包含参数的，那么：例如decorator1的function就不需要参数了，加上*args，**kwargs就可以覆盖到function函数中存在任意个参数的情况了

2.定义装饰器时，return返回的值是wrapper，不是wrapper（）

3.多个装饰器装饰同一个函数时，执行的顺序是按照从上到下的顺序依次执行的