需求:想要计算每个模块执行的时间。
> 传统方法,在每一个模块里面书写一个time时间模块即可。
import time
def count():
t1 = time.time()
sum = 0
for i in range(0,1000):
sum = sum + i
t2 = time.time()
print(t2-t1)
return sum
def count1():
t1 = time.time()
sum = 0
for i in range(0,1000):
sum = sum + i
t2 = time.time()
print(t2-t1)
return sum
> 装饰器:实现一个模块计算时间,一个模块书写应用,两个功能拆分开
import time
# 书写装饰器实现计算func()方法执行的时间,wrapper翻译为包装纸,包装材料
"""第一步:定义装饰器函数"""
def time1(func):
def wrapper():
t1 = time.time()
func()
t2 = time.time()
print(t2 - t1)
return wrapper
"""第二步:将转装饰器函数放在需要进行计算的模块上方"""
@time1
def count():
sum = 0
for i in range(0,1000):
sum = sum + i
return sum
# 拓展一下需求:需要给count传递参数,如何实现
"""
核心在于参数的携带方式
*args表示任意多个参数
"""
def time2(func):
def wrapper(*args):
t1 = time.time()
res = func(*args)
t2 = time.time()
print(t2 - t1)
"""因为func即就是count具有返回值,因此需要显性的赋值给res变量"""
print(res)
return wrapper
@time1
def count(x,y):
sum = 0
for i in range(x,y):
sum = sum + i
return sum
if __name__ == '__main__':
#将参数携带进来
count(2,10000)