python装饰器小结

import time
def my_sum():
	res = 1+2
#运行函数
my_sum()

#此时想要统计耗时
def my_sum():
	t1 = time.time()
	res = 1+2
	t2 = time.time()
	print("耗时：{}s".format(t2-t1))

#上面的代码可读性很差，核心代码和其他代码混合在一起，此时用装饰器
def display_time(fun):
	def inner():
		t1 = time.time()
		fun()
		t2 = time.time()
		print("耗时：{}s".format(t2-t1))
	return inner
	
@display_time
def my_sum():
	res = 1+2
	print(res)

#函数中有返回值
def display_time(fun):
	def inner():
		t1 = time.time()
		**res** = fun()
		t2 = time.time()
		print("耗时：{}s".format(t2-t1))
		return **res**
	return inner
	
@display_time
def my_sum():
	res = 1+2
	return res

#函数中有返回值+入参
def display_time(fun):
	def inner(***args**):
		t1 = time.time()
		res = fun(***args**)
		t2 = time.time()
		print("耗时：{}s".format(t2-t1))
		return res
	return inner
	
@display_time
def my_sum(a,b):
	res = a+b
	return res