一、debug日志装饰器
代码块:
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
add(1,2)
示例及运行结果:
二、编写装饰器required_types
题目描述:
编写装饰器required_types, 条件如下:
1). 当装饰器为@required_types(int,float)确保函数接收到的
每一个参数都是int或者float类型;
2). 当装饰器为@required_types(list)确保函数接收到的
每一个参数都是list类型;
3). 当装饰器为@required_types(str,int)确保函数接收到的每
一个参数都是str或者int类型;
4). 如果参数不满足条件, 打印 TypeError:参数必须为xxxx类
型
代码块:
import functools
def required_types(*kinds):
def required(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
for i in args:
if not isinstance(i,kinds):
# print('TypeError:参数必须为',kinds)
# break
raise TypeError('参数必须为%s,%s' %kinds)
else:
res = fun(*args,**kwargs)
return res
return wrapper
return required
@required_types(int,int)
def add(a,b):
return a + b
print(add(1.2,2.0))
示例及运行结果:
当参数的类型为int型时,会有如下效果:
三、多个装饰器
代码块:
def decorator_a(fun):
def inner_a(*args,**kwargs):
print('Get in inner_a')
return fun(*args,**kwargs)
return inner_a
def decorator_b(fun):
def inner_b(*args,**kwargs):
print('Get in inner_b')
return fun(*args,**kwargs)
return inner_b
@decorator_b
@decorator_a
def f(x):
print('Get in f')
return x*2
f(1)
示例及运行结果:
四、装饰器练习(添加、删除学生信息)
代码块:
import functools
import inspect
login_tables = ['root','admin','westos']
def is_admin(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
#inspect.getcallargs返回一个字典,key值:形参 value值:对应的实参
inspect_res = inspect.getcallargs(fun,*args,**kwargs)
print('inspect的返回值: %s' %inspect_res)
if inspect_res.get('name') == 'root':
res = fun(*args,**kwargs)
return res
else:
print('Permission denied!')
return wrapper
def login(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
if args[0] in login_tables:
res = fun(*args,**kwargs)
return res
else:
print('Error:无此用户')
return wrapper
@login
@is_admin
def add_student(name):
print('添加学生信息')
def del_student(name):
print('删除学生信息')
add_student('root')
示例及运行结果: