Python上下文管理器

上下文管理器(context manager)是Python2.5开始支持的一种语法，用于处理指 定代码块进入和退出时的操作。一般使用with语法，也可以直接调用相应的方法。

with语句

with语句是用来简化“try/finally”语句的，通常用于处理共享资源的获取和
释放，比如文件、数据库和线程资源。它的用法如下：

with VAR = EXPR:
BLOCK

其相当于进行了如下操作：

VAR = EXPR
VAR.__enter__()
try:
BLOCK
finally:
VAR.__exit__()

例子如下：

import time
class demo:
def __init__(self,label):
self.label = label

def __enter__(self):
self.start = time.time()

def __exit__(self, exc_ty, exc_val, exc_tb):
end = time.time()
print ('{}: {}'.format(self.label, end - self.start))

with demo('counting'):
n = 10000000
while n > 0:
n -= 1;

#  counting: 0.933553934097

也可利用@contextmanager装饰器改写如下:

from contextlib import contextmanager
import time

@contextmanager
def demo(label):
start = time.time()
try:
yield
finally:
end = time.time()
print('{}: {}'.format(label, end - start))

with demo('counting'):
n = 10000000
while n > 0:
n -= 1
# counting: 0.947228908539

# 其中yield之前的所有代码都类似于__enter__方法的内容。而yield之后的所有
# 代码类似__exit__方法的内容。