python装饰器函数详解
当我们在调用别人的已经写好的函数时,尤其是在多人项目章,如果不是非常了解其中的原理,一般不会修改其中的代码,只会增加一些功能,这个时候就需要用到装饰器,先上源码。
def wrapper(f):
def inner(*arg,**kwargs):
print('before')
ret = f(*arg,**kwargs)
print('after')
return ret
return inner
@wrapper # 或者 f = wrapper(f)功能相同
def f(*arg,**kwargs):
print('im the middle')
return 'f返回值'
print(f())
#结果:
#before
#im the middle
#after
#f返回值
这个wrapper 函数是一个装饰器函数,下面def的这个f是我们需要装饰的函数,在实际应用中就是我们需要在这个函数中增加功能的一个函数,我思考了一下,为什么增加功能不直接定义另外一个函数f2()
,然后把f1()放在f2中呢,这个点我也很疑惑,因为除了我不能定义一个和f()函数同名的函数这个缺点外,直接在f2中插入函数f()其实也是可以的。如果人可以指出这个装饰器的其他优点可以在下面留言,好了直接进入正题。
f其实是一个函数名指向的内存地址,而f() 则是函数功能,我们把f当做参数传进了wrapper函数名,其实是传入了一个内存地址。我们在wrapper函数嵌套了一个inner函数,arg,**kwargs表示可以是任意参数,这个inner函数的目的是为了执行“装饰”,而我们在wrapper函数的最后return了inner,其实是返回了inner 的内存地址,那么我们在执行f = wrapper(f)或者说@wrapper的时候,wrapper这个函数返回了inner 的内存地址,而左边的f 又等于inner 的内存地址,那左边的f不就是等于inner函数吗?,这样问题就变得简单了。
新的f就是等于inner函数,那么我们为了参数的顺利传递,我们在f中定义的arg,**kwargs,也要传入inner函数中,同时用ret获取原本f函数的返回值,在inner函数中返回ret,这样我们在装饰器中传入了before 和after也没有改变原本f的返回值,这个就是装饰器的一个标准格式。
ps:上面的装饰器有一点不好的,func函数的名称变成了inner不便于函数参数说明的查询。
因此于19-04-01进行一下改良
from functools import wraps
def dec(f):
@wraps(f)
def inner(*args,**kwargs):
# 装饰内容在这里写
print("这里写装饰内容前")
res = f ()
print("这里写装饰内容前")
return res
return inner
@dec
def f():
print("这里是f功能")
return "这里是装饰器返回值"
if __name__ == '__main__':
a = f()
这样做的好处是,我们在打印函数名变量时,函数的地址不会改变