from contextlib import contextmanager
import time
@contextmanager
def timer():
start = time.time()
yield
diff = time.time() - start
print(diff)
使用with調用剛纔定義的方法
with timer():
time.sleep(1)
with timer():
time.sleep(2)
with timer():
time.sleep(2)