与传统的高级语言相比,Python添加了两个重要的关键字,即yield和with。yield关键字的用法在前面的几篇博文中已经介绍的较为详细,从本博文开始,介绍with的用法。
与with语句密切相关的一个数据结构,叫做context manager。一个context manager类,至少需要定义__enter__和__exit__两个方法。下面是一个最简单的context manager类的定义。
class MyContextManager:
def __init__(self):
print "MyContextManager initialized"
def __enter__(self):
print "Entered MyContextManager"
def __exit__(self, *exc):
print "Exited MyContextManager"
__exit__方法需要有三个参数,这里为了简化,将它笼统定义为*exc。后面的博文会对这些参数做更详细的说明。
下面的with语句利用了刚刚定义的context manager类。
with MyContextManager():
print "In main block"
上面的代码输出如下
MyContextManager initialized
Entered MyContextManager
In main block
Exited MyContextManager
由此,我们可以大概了解with语句的执行过程:
1. 评估with语句中的context表达式,并创建context manager对象。(事实上,一个with语句中可以有多个context manager对象,后文再介绍。)
2. 加载context manager的__exit__方法,以备后用。
3. 执行context manager的__enter__方法。
4. 执行with语句块。
5. 执行context manager的__exit__方法。
参见https://docs.python.org/3/reference/compound_stmts.html#with