装饰器
- 装饰器和闭包有很大的联系。有时你需要在不改变源代码的情况下修改已经存在的函数。装饰器的运用可以提高效率,减少重复的代码。
- 装饰器的实质是一个函数。它把一个函数作为输入并且返回另一个函数。
在装饰器中,通常使用下面这些python技巧:
1.*args 和 **kwargs
2.闭包
3.作为参数的函数
例
函数 document_it() 定义了一个装饰器,会实现如下的功能:
- 打印输出函数的名字和参数的值
- 执行含有参数的函数
- 打印输出的结果
- 返回修改后的函数
def document_it(func):
def new_fuction(*args, **kwargs):
print('Running fuction:',func,_name_)
print('positional arguments:',args)
print('keyword arguments',kwargs)
result = func(*args, **kwargs)
print('result:',result)
return result
return new_fuction
无论传入 document_it() 的函数 func 是什么,装饰器都会返回一个新的函数,其中包含函数 document_it() 增加的额外语句。实际上,装饰器并不需要执行函数 func 中的代码,只是在结束前函数 document_it() 调用函数 func 以便得到 func 的返结果和附加代码的结果。
那么如何使用装饰器呢?
- 人工赋值
def document_it(func): #装饰器
def new_fuction(*args, **kwargs):
print('Running fuction:',func)
print('positional arguments:',args)
print('keyword arguments',kwargs)
result = func(*args, **kwargs)
print('result:',result)
return result
return new_fuction
def add_ints(a,b): #被装饰函数
return a+b
cooler_add_ints = document_it(add_ints) #人工对装饰器赋值
cooler_add_ints(3,5)
####################################
结果
Running fuction: <function add_ints at 0x000001EAECBF9950>
positional arguments: (3, 5)
keyword arguments {}
result: 8
- 直接赋值(在要装饰的函数前添加装饰器的名字 例 @decorator_name)
def document_it(func): #装饰器
def new_fuction(*args, **kwargs):
print('Running fuction:',func)
print('positional arguments:',args)
print('keyword arguments',kwargs)
result = func(*args, **kwargs)
print('result:',result)
return result
return new_fuction
@document_it
def add_ints(a,b): #被装饰函数
return a+b
add_ints(3,5)
###################################
结果
Running fuction: <function add_ints at 0x000001F47DB89AE8>
positional arguments: (3, 5)
keyword arguments {}
result: 8
同样一个函数可以有多个装饰器。
下面我们写一个对结果求平法的装饰器 square_it ():
def document_it(func): #装饰器
def new_fuction(*args, **kwargs):
print('Running fuction:',func)
print('positional arguments:',args)
print('keyword arguments',kwargs)
result = func(*args, **kwargs)
print('result:',result)
return result
return new_fuction
def square_it(func):
def new_function(*args,**kwargs):
result = func(*args,**kwargs)
return result * result
return new_function
@document_it
@square_it
def add_ints(a,b): #被装饰函数
return a + b
add_ints(3,5)
############################结果
Running fuction: <function square_it.<locals>.new_function at 0x000001E63B199EA0>
positional arguments: (3, 5)
keyword arguments {}
result: 64
靠近函数定义 (def上面) 的装饰器最先执行,然后依次执行上面的,任何顺序都会得到相同的结果。
def document_it(func): #装饰器
def new_fuction(*args, **kwargs):
print('Running fuction:',func)
print('positional arguments:',args)
print('keyword arguments',kwargs)
result = func(*args, **kwargs)
print('result:',result)
return result
return new_fuction
def square_it(func):
def new_function(*args,**kwargs):
result = func(*args,**kwargs)
return result * result
return new_function
@square_it
@document_it ###交换后
def add_ints(a,b): #被装饰函数
return a + b
add_ints(3,5)
######################结果
Running fuction: <function add_ints at 0x000001FD5D859D08>
positional arguments: (3, 5)
keyword arguments {}
result: 8