练习
练习一
import time
import functools
def log_kind(kind):
def add_log(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
start_time = time.time()
res = fun(*args,**kwargs)
end_time = time.time()
print('<%s>[%s] 函数名: %s,运行时间: %.6f,运行返回值结果: %d' %(kind,time.ctime(),fun.__name__,end_time-start_time,res))
return res
return wrapper
return add_log
@log_kind('debug')
def add(x,y):
time.sleep(1)
return x + y
print(add(1,2))
运行结果:
练习二
编写装饰器required_types, 条件如下:
1). 当装饰器为@required_types(int,float)确保函数接收到的每一个参数都是int或者float类型;
2). 当装饰器为@required_types(list)确保函数接收到的每一个参数都是list类型;
3). 当装饰器为@required_types(str,int)确保函数接收到的每一个参数都是str或者int类型;
4). 如果参数不满足条件, 打印 TypeError:参数必须为xxxx类型
注意:
isinstance方法 :判断参数类型
格式:字符串,类型
raise关键字 系统报错,抛出异常
使用示例:
代码如下:
import functools
def required_type(*kind): #设置装饰器接受任意多个参数
def required(fun): #定义函数,接收参数
@functools.wraps(fun)
def wrapper(*args,**kwargs):
for i in args: #遍历所有参数
if not isinstance(i,kind): #判断每个参数的类型是否为函数中给定的并且传到装饰器的参数
print('TypeError:参数必须为',kind) #不是,报错
else:
res = fun(*args,**kwargs)
return res
return wrapper
return required #给装饰器返回指定的多个参数
@required_type(int,float)
def add(a,b):
return a + b
print(add(1,2.2))
执行结果:
或者将报错部分写为以下:
import functools
def required_type(*kind): #设置装饰器接受任意多个参数
def required(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
for i in args: #遍历所有参数
if not isinstance(i,kind): #判断每个参数的类型是否为函数中给定的并且传到装饰器的参数
raise TypeError('参数必须为%s,%s' %kind)
else:
res = fun(*args,**kwargs)
return res
return wrapper
return required #给装饰器返回指定的多个参数
@required_type(int,float)
def add(a,b):
return a + b
print(add([1,2,3],2.2))