python迭代器、生成器、装饰器之装饰器
装饰器。。。。。。
定义:本质是函数,为其他函数添加附加功能
原则: 1.不能修改被装饰的函数的源代码
2.不能修改被装饰函数的调用方式
仔细观察下面代码,看看有什么发现。
内嵌函数+高阶函数+闭包=》装饰器
import time
# 内嵌函数
def timmer(func):
def wrapper(*args, **kwargs):
start_time = time.time()
res = func(*args, **kwargs)
stop_time = time.time()
print('run time is %s' % (stop_time - start_time))
return res
return wrapper # 返回函数名
@timmer
def foo():
time.sleep(3)
print('from foo')
foo()
高阶函数。。。。。下面的例子中foo就是属于一个高阶函数。
'''
高阶函数:
1、函数接收的参数是一个函数名
2、函数的返回值是一个函数名
只要满足其中一个就属于高阶函数
'''
def fun():
print("from fun")
def foo(fun1):
fun1()
foo(fun)
返回值是函数名的高阶函数
def fun():
print("from fun")
def foo(fun1):
return fun1
foo(fun)()
函数嵌套。。。。。。通过下面的例子,可以看出,函数内可以套函数,而函数也是一个变量,
通过locals()可以看到。
'''
函数嵌套
'''
def fun1():
print("from fun1")
def fun2():
print("from fun2")
print(locals())
fun1()
# 运行结果
#from fun1
#{'fun2': <function fun1.<locals>.fun2 at 0x000002A2C4D4AD90>}
闭包闭包,一个函数一个包。。。主要还是作用域,请看https://blog.csdn.net/June_King/article/details/87090970
重点来了。装饰器。。。。。。
# 简单的装饰器例子
def timer(func):
def wrapper():
func()
return wrapper
def test():
time.sleep(3)
print("test函数执行完毕")
res = timer(test)
res()
这是不对的,前面说过,装饰器不能修改被装饰函数的源代码,不能修改被装饰函数的调用方式。
这里调用时,修改了调用的方式。。。
那这样呢?
test = timer(test)
test()
看着是没有修改调用方式,但是每次调用时都会重新赋值。。。这样不合理
那应该怎么做?下面这样写,仅仅是在函数调用之前加上@装饰器名字即可
@timer
test()
看下面的例子,看看都打印了什么。
def timer(func):
def wrapper():
func()
return wrapper
@timer
def test():
time.sleep(2)
print("test函数执行完毕")
return "test()的返回结果"
res = test()
print(res)
可以发现,我们并没有得到test()函数的返回值,如果想要得到被装饰函数的返回值,需要这样写
def timer(func):
def wrapper():
res = func()
return res
return wrapper
在装饰器中通过变量来接收被装饰函数的返回值,然后利用return返回。这样一个带有返回值的装饰器就书写完成了。
这些还远远不够,有时候我们被修饰的函数还需要传入一些参数
def timer(func):
def wrapper(name, age):
res = func(name, age)
return res
return wrapper
@timer
def test(name, age):
time.sleep(2)
print("name:%s,age:%d," % (name, age))
print("test函数执行完毕")
return "test()的返回结果"
res = test('june', 18)
print(res)
上面的例子虽然可以传递参数,但依然存在不足,当test()的参数变化时,上面的例子就不能用了,需要对装饰器进行改进
def timer(func):
def wrapper(*args, **kwargs):
res = func(*args, **kwargs)
return res
return wrapper
上面的这个例子可以实现无论被修饰函数的参数如何让改变,都能够接收。关于*args,**kwargs可以参考https://blog.csdn.net/June_King/article/details/87085052
好了,目前为止我们写的装饰器可以传入参数,也可以拥有返回值了,那么如果我们的装饰器还需要参数的话,那。。。。
def auth(filed):
print(filed)
def fun1(fun):
def fun2(*args, **kwargs):
fun(*args, **kwargs)
return fun2
return fun1
@auth("我是装饰器的参数")
def foo():
print("from foo")
foo()
在原来的基础上再加一层函数。。。
小结,不修改原函数代码,不修改原函数的调用方式。