Lesson 90 Python中装饰器的应用示例
1. 示例1:比较for循环和jion方法哪个更快
import random
import time
import functools
import string
li = [random.choice(string.ascii_letters) for i in range(100)]
# print(li)
def timecal(fun):
"""这是一个装饰器"""
@functools.wraps(fun)
def wrapper(*args,**kwargs): #函数的功能是计算函数运行时间
"""这是wrapper函数"""
start_time = time.time() #起始时间
res = fun(*args,**kwargs)
end_time = time.time() #结束时间
print('运行时间为: %.8f' %(end_time - start_time))
return res
return wrapper
@timecal
#定义两个函数分别输出li列表中的元素,定义装饰器用来比较两个函数运行的时间
def con_add():
"""这是con_add函数"""
s = ''
for i in li:
s += (i + ',')
return s
@timecal
def join_add():
return ','.join(li)
# a = con_add()
# print(a)
con_add()
join_add()
#print(con_add.__name__)
#print(con_add.__doc__)
可以看到,在打印列表时,jion方法比使用for循环累加要快
2. 示例2:创建add_log装饰器, 被装饰的函数打印日志信息
创建装饰器, 要求如下:
# 1. 创建add_log装饰器, 被装饰的函数打印日志信息;
# 2. 日志格式为: [字符串时间] 函数名: xxx, 运行时间:xxx,
运行返回值结果:xxx
functools 模块中主要包含了一些函数装饰器和便捷的功能函数
自定义python装饰器时一定要记住使用@functools.wraps(fun)修饰wrapper
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
add(1,2)
可以看到,能够实现装饰器的功能,按照日志格式要求打印日志信息