@contextlib.contextmanager 装饰器

  @contextlib.contextmanager 是一个装饰器，由它修饰的方法会有两部分构成，中间由 yield 关键字分开。由此方法创建的上下文管理器，在代码块执行前会先执行yield上面的语句，在代码块执行后会再执行yield下面的语句，案例：

import contextlib
import time


@contextlib.contextmanager
def timeit(title):
    print('1...')
    start = time.time()
    yield
    print('2...')
    end = time.time()
    usedTime = (end - start) * 1000
    print('Use time %d ms' % usedTime)


with timeit(1):
    print('3...')
    time.sleep(1)

with timeit(2):
    print('4...')
    time.sleep(2)

输出结果：

1...
3...
2...
Use time 1001 ms
1...
4...
2...
Use time 2002 ms

引用与感谢

• http://www.bjhee.com/python-context.html