python中with的作用_Python中with的用法

在Python中，with语句的目的是简化try/finally的模式，使得代码可以保证运行完毕后执行某一项操作，哪怕是发生了异常、return等调用而终止，也一定会执行指定的操作。为了使代码可以使用with语句，with语句中的类必须实现__enter__和__exit__两个方法，分别在with语句开始运行前和执行后运行。

比如我们平常用得最多的用法：

with open('mirror.py') as fp:

src = fp.read(60)

读取文件用的就是with语句，open返回一个文件类型TextIOWrapper，传值给fp。可以对fp执行读取文件操作，当出现异常时会执行TextIOWrapper的__exit__方法。

下面看一个更复杂的例子，摘抄自《流畅的Python》，虽然这段代码我试过在Python2.7中无法运行(由于无法更改stdout.write)函数，但可以一窥其中的思想：

class LookingGlass:

def __enter__(self):

import sys

self.original_write = sys.stdout.write

sys.stdout.write = self.reverse_write

return "Jacy"

def reverse_write(self, text):

self.original_write(text[::-1])

def __exit__(self, exc_type, exc_value, traceback):

import sys

sys.stdout.write = self.original_write

if exc_type is ZeroDivisionError:

print "zero division error"

return True

with LookingGlass() as what:

print "El psy congroo"

print what

print what

按照书中的解释，在LookingGlass对象__enter__时，将Python输出的函数予以替换，这样可以将输入的文字全部倒置；然后with语句中输出的所有文本就都是倒置的。在__exit__时再将输出流设定回初始值。其中__enter__的返回值就是what，因此两次输出what分别是“ycaJ”和“Jacy”。

最后在__exit__函数中可以处理with块的异常情况，比如上面的示例代码就处理了ZeroDivisionError的情况。该函数返回True代表异常被正确地处理，否则会将异常继续抛出。