在Python中让自己创建的函数、类、对象支持with语句，就实现了上线文管理协议。我们经常使用with open(file, "a+") as f:这样的语句，无需手动调用f.close()关闭文件。这种用法不仅优雅，而且避免遗忘释放资源，十分方便。所以，当操作某些资源时，需要对资源的获取与释放进行自动操作，就可以用上线文管理器。比如：数据库的连接，查询，关闭处理；socket的连接和断开。本篇主要介绍，如何让自己创建的类、对象、函数等支持with语句，详细请看下文。
1 让对象支持上下文管理协议
在类中，实现 __enter__()和__exit__()方法，类创建的对象就支持with语句。如下：
class A:
def __enter__(self):
print("in __enter__")
return [1, 2, 3]
def __exit__(self, exc_type, exc_val, exc_tb):
print("in __exit__")
obj = A()
with obj as o:
print("看看o是什么：", o)
print("do something")
print("end")
注意本例的输出结果的顺序：
in __enter__ 看看o是什么： [1, 2, 3] do something in __exit__ end
（1）当执行 with 语句的时候，对象的 __enter__() 方法被触发， 它返回的值(如果有的话)会被赋值给 as 声明的变量。对应输出应该是【in __enter__】和将[1,2,3]赋值给o【字母o】。
（2）然后，with 语句块里面的代码开始执行。对应的输出是【看看o是什么： [1, 2, 3]】和【do someting】。
（3）最后，__exit__() 方法被触发进行清理工作。对应的输出是【in __exit__】。
补充说明：
__exit__()方法的第三个参数包含了异常类型、异常值和追溯信息(如果有的话)。 __exit__()方法能自己决定怎样利用这个异常信息，或者忽略它并返回一个None值。
如果 __exit__() 返回 True ，那么异常会被清空，就好像什么都没发生一样， with 语句后面的程序继续在正常执行。
上面的例子还不支持多个with嵌套使用，下面是一个可以嵌套使用with语句的例子：
from socket import socket, AF_INET, SOCK_STREAM
class Connection:
def __init__(self, address, family=AF_INET, type=SOCK_STREAM):
self.address = address
self.family = family
self.type = type
self.connections = []
def __enter__(self):
sock = socket(self.family, self.type)
sock.connect(self.address)
self.connections.append(sock)
return sock
def __exit__(self, exc_ty, exc_val, tb):
self.connections.pop().close()
conn = Connection(('www.python.org', 80))
with conn as s1:
print(s1)
with conn as s2:
print(s2)
2 装饰器版上下文管理器
上面介绍了在类和对象中实现上下文管理协议，其实Python标准库中contextlib包的@contextmanager装饰器能够轻松实现一个上下文管理器，下例是用其实现统计代码块耗时的上下文管理器：
import time
from contextlib import contextmanager
# 来看一个装饰器版本的上下文管理器
# 检查代码消耗时间块
@contextmanager
def timecount(name):
start = time.time()
try:
yield
finally:
end = time.time()
print('{}: {}'.format(name, end - start))
with timecount('cost time:'):
time.sleep(2)
# cost time:: 2.000200033187866
timecount（）中，yield之前的代码相当于__enter__()方法；yield 之后的代码相当于 __exit__()方法。如果有异常会自动在yield一行抛出。
上下文管理器可以应用在事务中：
# 更高级的事务管理
@contextmanager
def list_transaction(orig_list):
working = list(orig_list)
yield working
orig_list[:] = working
lis = [1, 2, 3]
with list_transaction(lis) as work:
work.append(5)
work.append(6)
print(lis)
# [1, 2, 3, 5, 6]
一旦with语句内有异常产生，之前的操作不会生效：
lis = [1, 2, 3]
with list_transaction(lis) as work:
work.append(5)
work.append(6)
print(lis)
raise RuntimeError("回滚")
如果在交互式命令行中打印lis,依然会发现lis的内容没有改变，也就是说，with语句中的代码出现异常，with语句中的操作都不会生效，只有with无异常退出，才会生效。对于事务管理来说是比较有用的。
3 小结
（1）当操作一些需要打开、连接、断开或释放的资源时，让对象或函数支持with语句十分方便、省事、优雅。如数据库的链接断开、套接字的连接断开、事务、锁资源等。
（2）类中实现__enter__()和__exit__()方法，即可实现上下文管理协议，对象可使用with语句。
（3）通过contextlib包中的@contextmanager装饰器装饰一个函数，该函数即可使用with语句。