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代表异常被正确地处理,否则会将异常继续抛出。

相关推荐