python中的装饰器是增强函数或类的功能的一个函数 / 类,是闭包(closure)的一种应用。
通过使用装饰器,我们可以在不修改原函数代码的前提下拓展函数的功能。
修饰器是增加函数额外功能(这种功能大多是通用的)。
#需要把两个函数包装起来。
import time
def time_calc(func):
def wrapper():
start_time = time.time()
f = func()
exec_time = time.time() - start_time
print('{} :{}'.format(func.__name__, exec_time))
return f
return wrapper
def func1():
print("This is func1")
time.sleep(2)
def func2():
print("This is func2")
time.sleep(5)
decorated1 = time_calc(func1) # 执行函数前必须先对函数进行包装,然后执行包装后的函数.
decorated1()
decorated2 = time_calc(func2)
decorated2()
定义前添加一个装饰器名和@符号,来实现对函数的自动包装。
#自动包装@修饰符,进行自动包装。
import time
def time_calc(func):
def wrapper():
start_time = time.time()
f = func()
exec_time = time.time() - start_time
print('{} :{}'.format(func.__name__, exec_time))
return f
return wrapper
@time_calc
def func1():
print("This is func1")
time.sleep(2)
@time_calc
def func2():
print("This is func2")
time.sleep(5)
func1()
func2()
-----------------------------
>>>This is func1
>>>func1 :2.011098861694336
>>>This is func2
>>>func2 :5.007617473602295