一 闭包
在学装饰器之前,我们先来学学闭包,因为装饰器需要用到闭包,那么什么是闭包呢?下面通过一个简单的程序来演示,像以下的代码我们称之为闭包。
def func_out(x):
def func_in():
print(x)
return func_in
test = func_out(6) # P1
test() #P2
简单的来看,就是函数里面嵌套一个函数,当程序执行到P1位置的时候,运行func_out函数,然而func_in函数没有运行,而且以返回值的现实返回给test,也就是说test现在指向这个函数所在的空间。然后当运行到P2位置的时候,func_in函数才开始运行。
二. 装饰器原理
下面来让我们来看一下这个程序
def func_out(func):
def func_in():
print("haha")
func()
return func_in
def test():
print(6)
test = func_out(test) #P1
test() #P2
这也是一个闭包,只不过传递给闭包的参数是一个函数,当程序运行到P1的时候,func是指向test函数的区域的,然后将func_in返回给test,此时原来test函数那块区域由现在的func指向,而现在的test指向func_in函数;当运行到P2的时候,也就是运行的是func_in函数,然后运行func函数。
ps:动手画一下地址的指向,相信你很快就会明白其中的原理的。
从上面来看,同样是调用test,运行的结果在原来test函数的基础上增加了func_out这个闭包的功能,这也就是我们常说的装饰器,在python中可以将其等价于一下写法。
@func_out #P3
def test():
print(6)
test()
就是将P1处的代码写成P3处的代码,这样我们就实现了一个简单的装饰器了。
三带参数函数的装饰器
def func_out(func):
def func_in(a,b):
print("a=%d b=%d"%(a,b))
func(a,b)
return func_in
@func_out #P3
def test(a,b): #P4
print("a+b=%d"%(a+b))
test(11,12)
现在运行的test函数,就是func_in函数,func指向P4处的test,此处test函数包含两个参数,所以func也要传递两个参数,那么这个参数从哪里来呢?显然是从func_in函数来,它的参数是由外部传递过来的,所以func_in需要包含两个形参,然后给func传递。
四.带返回值的装饰器
def func_out(func):
def func_in(a,b):
print("a=%d b=%d"%(a,b))
res = func(a,b) #P1
return res
return func_in
@func_out #P3
def test(a,b): #P4
print("a+b=%d"%(a+b))
return a+b
res = test(11,12) #P5
print(res)
P4处的test函数的返回值返回给P1处的res,然后func_in函数的返回值返回给P5处的res。原因前面也说过了,func其实就是P4处的test函数,而P5处的函数其实就是func_in函数
五.带有参数的装饰器
def func_extern(arg):
def func_out(func):
def func_in(a,b):
print("a=%d b=%d"%(a,b))
res = func(a,b)
print("arg=%s"%arg)
return res
return func_in
return func_out
@func_out("abc") #P3
def test(a,b): #P4
print("a+b=%d"%(a+b))
return a+b
res = test(11,12) #P5
print(res)
在这里已经看到了三个函数嵌套了,肯定晕了吧。其实就里面两个函数来看,就是前面的闭包,只不过在外面嵌套的一个函数,那这个函数的功能就是来接收装饰器传递过来的参数的。
在这里装饰器的基本功能已经说完了,这部分内容主要是理解函数的引用。
Thank for your read !!!!!