1.装饰器实现一个函数计时器
import time
import random
import string
import functools
#问题1:被装饰的函数有返回值
#问题2:如何保留被装饰函数的函数名和帮助信息文档
li = [random.choice(string.ascii_letters) for i in range(100)]
# print(li)
def timeit(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
"""这是一个wrapper函数"""
start_time = time.time()
res = fun(*args,**kwargs)
end_time = time.time()
print('运行时间为:%.6f' %(end_time - start_time))
return res
return wrapper
@timeit
def con_add():
"""这是for循环"""
s = ''
for i in li:
s += (i + ',')
print(s)
@timeit
def join_add():
"""这是join方法"""
print(','.join(li))
@timeit
def fun_list(n):
"""这是fun_list函数"""
return [i * 2 for i in range(n)]
@timeit
def fun_map(n):
"""这是fun_map函数"""
return list(map(lambda x:x*2,range(n)))
# con_add()
# join_add()
# print(con_add.__doc__)
# print(con_add.__name__)
# print(join_add.__doc__)
# print(join_add.__name__)
print(fun_list(5000))
print(fun_map(5000))
2.创建装饰器, 要求如下:
1. 创建add_log装饰器, 被装饰的函数打印日志信息;
2. 日志格式为: [字符串时间] 函数名: xxx, 运行时间:xxx,
3.运行返回值结果:xxx
import time
import functools
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,运行时间:%.6f,运行返回值结果:%d' %(time.ctime(),
fun.__name__,end_time - start_time,res))
return res
return wrapper
@add_log
def add(x,y):
time.sleep(1)
return x+y
print(add(1,10))
# print(time.ctime())
3.判断是不是root用户,如果是root用户,那么则可以录入学生信息
import functools
import inspect
def is_root(fun):
@functools.wraps(fun)
def wrapper(*args,**kwargs):
#inspect.getcallargs返回值是字典,key值为:形参,value值为:形参对应实参
inspect_res = inspect.getcallargs(fun,*args,**kwargs)
print('inspect_res的返回值为:%s' %inspect_res)
if inspect_res.get('name') == 'root':
res = fun(*args,**kwargs)
return res
else:
print('not root user')
return wrapper
@is_root
def add_student(name):
print('添加学生信息...')
def del_student(name):
print('删除学生信息...')
add_student('westos')
4.编写装饰器required_ints, 条件如下:
1). 确保函数接收到的每一个参数都是整数; # 如何判断变量的类型?
type(s), isinstance(s,str)
2). 如果参数不是整形数, 抛出异常raise TypeError(“参数必须为整形”)
# print(isinstance(1,int))
# print(isinstance(1.0,float))
# print(isinstance('a',str))
# if not isinstance(1,int):
# raise TypeError("参数必须为整形")
# def add(*args,**kwargs):
# return sum(args + kwargs.values())
#
# print(add(1,2,3,65,67,1.0,a=1,b=2)) 上面是不用装饰器如何实现
import functools
def required_ints(f):
@functools.wraps(f)
def wrapper(*args,**kwargs):
for arg in args + tuple(kwargs.values()):
if not isinstance(arg,int):
raise TypeError('参数必须为整形')
else:
return f(*args,**kwargs)
return wrapper
@required_ints
def add(*args,**kwargs):
return sum(args + tuple(kwargs.values()))
print(add(1,2,3,65,67,5,a=1,b=2))
以上是几个关于装饰器的练习 ~