python函数式编程之装饰器(一) (3)

来分析一下上面例子中，index函数被timmer装饰器和auth装饰器装饰的代码装饰流程

@auth # index=auth(timmer(index)) @timmer # index=timmer(index) def index(): time.sleep(2) print("welcome to index page")

在上面得出结论，一个函数同时被两个装饰器时，加在下面的装饰器先装饰

1.timmer装饰器装饰原始的index，可以写成:index=timmer(index) 2.在timmer装饰器中，timmer装饰器实际上是返回inner的内存地址，所以在这里，index=inner 3.timmer装饰器装饰完成后，由auth装饰器来装饰，此时可以写成index=auth(index)， 4.这里auth括号里的index已经不再是原始index函数，而是已经被timmer装饰过后的index了，所以index=auth(timmer(index)) 5.又因为timmer装饰的结果等于inner函数的内存地址，所以：index=auth(inner)

至此，两个装饰器的装饰过程已经知道了，来看程序的执行过程

6.程序先执行auth装饰器，进入用户认证，请用户输入用户名和密码 7.用户输入正确的用户名和密码后，开始执行func函数，也已经上面分析的inner函数 8.timmer装饰器先定义程序的开始运行时间，然后运行func函数，也就是原生的index函数 9.index函数先睡眠2秒，然后执行print语句，再定义程序的结束时间 10.最后统计并打印程序的运行时间，至此程序运行完毕。

所以这里用户输入用户名和密码的时间不会被timmer装饰器统计在内

7.被装饰函数参数的设置与定义

先来看一段代码

import time def timmer(func): def inner(): start_time=time.time() func() end_time=time.time() print("run time: %s " %(end_time-start_time)) return inner @timmer def index(): time.sleep(2) print("welcome to index page") @timmer def home(name): time.sleep(3) print("welcome to %s home page" % name)

如上所示，home函数添加了一个参数，而index函数并没有参数

按照正常的函数的定义与调用方式，调用index函数和home函数的方式应该是下面这种形式

index() home("python")

然后我们运行程序就会发现，程序抛出了异常

File "E:\python_learn\py_code\test.py", line 28, in <module> home("python") TypeError: inner() takes 0 positional arguments but 1 was given

说个异常说明inner函数不需要位置参数，但是我们给了一个位置参数

回到timmer装饰器定义的部分，可以看到，timmer装饰器的内部函数确实没有定义参数

这样一来，timmer装饰器只能用于装饰没有参数的函数了，

我们可以在timmer装饰器定义的时候为inner函数添加一个参数

import time def timmer(func): def inner(name): start_time=time.time() func(name) end_time=time.time() print("run time: %s " %(end_time-start_time)) return inner @timmer def index(): time.sleep(2) print("welcome to index page") @timmer def home(name): time.sleep(3) print("welcome to %s home page" % name) index() home("python")

但是这样一来，timmer装饰器装饰index函数的时候又会抛出异常，因为index函数没有参数

File "E:\python_learn\py_code\test.py", line 27, in <module> index() TypeError: inner() missing 1 required positional argument: 'name'

在不知道被装饰函数的参数个数的情况下，即被装饰函数的参数可变长，且形式不固定的时候，

可以使用*args和**kwargs，把上面的代码修改

import time def timmer(func): def inner(*args,**kwargs): start_time=time.time() func(*args,**kwargs) end_time=time.time() print("run time: %s " %(end_time-start_time)) return inner @timmer def index(): time.sleep(2) print("welcome to index page") @timmer def home(name): time.sleep(3) print("welcome to %s home page" % name) index() home("python")

再次运行程序，查看运行结果

welcome to index page run time: 2.0 welcome to python home page run time: 3.0

由上可知，在不知道被装饰函数的参数个数时，可以使用*args和**kwargs来表示任意长度任意形式的参数

8.被装饰函数的返回值

修改上面的代码，为home函数定义一个返回值，分别打印index函数和home函数的返回值

import time def timmer(func): def inner(*args,**kwargs): start_time=time.time() func(*args,**kwargs) end_time=time.time() print("run time: %s " %(end_time-start_time)) return inner @timmer def index(): time.sleep(2) print("welcome to index page") @timmer def home(name): time.sleep(3) print("welcome to %s home page" % name) return("home func") index_res=index() print(index_res) home_res=home("python") print(home_res)

运行程序，可以看到

welcome to index page run time: 2.0 None welcome to python home page run time: 3.0 None

可以看到，home函数中定义的返回值并没有被打印出来，显示的值为None

因为这里执行的home函数不是原始定义的home函数，而是wrapper函数的执行结果

因为wrapper函数并没有定义返回值，所以执行被装饰后的home函数并没有打印出返回值