“”“”
什么是装饰器?
"""
还是通过一个例子来慢慢说明
先看下面的例子
def func_1(x):
return x*2
def fun_2(x):
return x*3
def fun_3(x, y i, j):
return x(i)+y(j)
r = func_3(func_1,func_2,2,3) #把函数引用作为参数传递给func_3
print(r)
#通过上面的例子明白:函数的参数可以是函数,也可以是一个类,其实可以是任何类型
def runtime_noargs():
pass
@runtime_noargs #这里就是一个装饰器,装饰器就是一个闭包函数,回忆下上篇文章的闭包感念
def function_demo1()
print("demo1函数被运行")
那么是如何把某种功能附加给上面的装饰器,让它帮助被装饰的函数实现某种功能的呢?
我们先对装饰器函数进行修改
import time
def runtime_noargs(function_name): #将来这里的function_name接受被装饰的函数名
def wrapper():
start_time = time.time()
function_name() #在这里执行被装饰器装饰的函数
end_time = time.time()
print(end_time - start_time)
return wrapper
@runtime_noargs #当装饰器用来装饰某函数时,被装饰的函数名被自动传参给装饰器函数
def function_demo1()
time.sleep(1)
print("demo1函数被运行")
function_demo1() #可以使用debug单步运行模式来看看函数的具体执行步骤
上面是一个没有参数的函数,通过装饰器来装饰的,那么有参数的函数如何通过装饰器来装饰呢?
def args_is_string(function_name):
def wrapper(a):
t=type(a)
if not isinstance(t(), str):
print("参数错误")
else:
funtion_name(a)
return wrapper
@args_is_string #通过一个装饰器函数来限制被装饰函数输入的参数只能是字符串
def function_demo2(args): #函数里的参数和装饰器函数里的内部函数wrapper参数对应,这样 print(args) #args参数就自动传给了装饰器中内部函数的参数a上,这是python的 #规则
function_demo2("aaaa")
上面的例子中,被装饰的函数只能传一个参数,那么要传多个参数呢?
通过一个例子来说明如何实现多个参数的函数的是如何被装饰的
def many_args(function_name):
def wrapper(*args):
print(*args) #不定长参数这里一个*,打印时需要
function_name(*args)
return wrapper
@many_args
def function_demo3(*args) #*args可以接受不定长参数,可以接受若干个参数
print(*args)
function_demo3(1,2,3,4,5,6)
如果关键字参数这种怎么办呢?
def dict_args(funtion_name):
def wrapper(**kwargs):
print(kwargs) #不定长字典类型的参数,两个*,在打印时不需要加*,注意和上面的区别
function_name(**kwargs)
return wrapper
@dict_args
def function_demo4(**kwargs): #**kwargs用来传不定长的字典形式参数
print(kwargs)
funtion_demo4(name = "fang", age = 10, address = "北京")
如果是混合型的参数呢?
def combo_args(function_name):
def wrapper(*args, **kwargs):
print(*args,kwargs)
return warpper
@combo_args
def function_demo5(*args, **kwargs):
pass
function_demo5(1,2,name="fang",age = "10",address="北京")
上面是装饰器的基本使用方法。
再介绍一种进阶版的装饰器,监控函数运行是否超时
def max_runtime(function_time);
def out_wrapper(funtion_name):
def wrapper(*args, **kwargs):
start_time = time.time()
i = function_name(*args,**kwargs) #如果函数需要返回值,这里必须由接住函数 #返回值
end_time = time.time()
use_time = end_time - start_time
if use_time > timeout:
raise RuntimeError("函数运行超时")
return i #这里必须把函数返回值返回去,即便在被装饰的函数本体里已经写了 #return
return wrapper
return out_wrapper
@max_runtime(timeout=1)
def function_demo6(*args, **kwargs):
time.sleep(2)
print("demo6运行")
return 1
result = function_demo6()
print(result)
上面的这个例子告诉我们,装饰器后面带参数时,装饰器的闭包函数体就要多写一层函数,见上图之间的对应关系。