装饰器
为了动态拓展函数,而不去破坏原函数的调用,产生了装饰器。通常使用装饰器给函数加log。
单层装饰器
def trim(f):
def inner(*args,**kwargs):
print("1"*10)
ret = f(*args,**kwargs)
print("1"*10)
return ret
return inner
"""
@trim : func = trim(func)
"""
@trim
def func(*args,**kwargs):
for i,j in enumerate(*args):
print(i,j)
print(kwargs)
return args
if __name__ == '__main__':
func([1,3],name="hhh")
以上是一个单层装饰器,trim是装饰函数,func是被装饰函数。@trim是Python中的一颗语法糖,它等价于func = trim(func)。
使用func.__name__查看func的名字可以发现它其实是inner。
tips:
- 如果被装饰函数没有参数 ,inner也可以不用带参数,但是如果被装饰函数有了参数,inner函数必须带有参数,因为在inner函数中调用func时,必须要有参数,这时的参数只能来自于inner。一般来说,无论被装饰函数有没有参数,我们都会给inner两个参数,一个可变参数,一个关键字参数。
- 同样如果被装饰函数有返回值,需要在inner中调用func进行保存,然后在inner中返回这个值
多层装饰器
def trim(f):
def inner(*args,**kwargs):
print("1"*10)
ret = f(*args,**kwargs)
print("1"*10)
return ret
return inner
def wrap(f):
def wahia(*args,**kwargs):
print("2"*10)
ret = f(*args,**kwargs)
print("2"*10)
return ret
return wahia
"""
@wrap : func = wrap(func)
@trim : func = trim(func)
"""
@wrap
@trim
def func(*args,**kwargs):
for i,j in enumerate(*args):
print(i,j)
print(kwargs)
return args
if __name__ == '__main__':
func([1,3],name="hhh")
# print(func.__name__)
多层装饰器的时候,要注意位置,位置不同所产生的结果也会不同。以上是func先被trim装饰,返回inner,此时func = inner,在此之上再被wrap装饰,返回wahia。所以func.__name__是wahia。