如果有一个类包含 __enter__ 方法和 __exit__ 方法,像这样:
class controlled_execution:
def__enter__(self):
set things up
return thing
def__exit__(self, type, value, traceback):
tear things down
那么它就可以和with一起使用,像这样:
with controlled_execution() as thing: some code 当with语句被执行的时候,python对表达式进行求值,对求值的结果(叫做“内容守护者”)调用__enter__方法,并把__enter__ 方法的返回值赋给as后面的变量。然后python会执行接下来的代码段,并且无论这段代码干了什么,都会执行“内容守护者”的__exit__ 方法。
作为额外的红利,__exit__方法还能够在有exception的时候看到exception,并且压制它或者对它做出必要的反应。要压制 exception,只需要返回一个true。比如,下面的__exit__方法吞掉了任何的TypeError,但是让所有其他的exceptions 通过:
def __exit__(self, type, value, traceback): return isinstance(value, TypeError) 在Python2.5中,file object拥有__enter__和__exit__方法,前者仅仅是返回object自己,而后者则关闭这个文件: >>> f = open("x.txt") >>> f <open file 'x.txt', mode 'r' at 0x00AE82F0> >>> f.__enter__() <open file 'x.txt', mode 'r' at 0x00AE82F0> >>> f.read(1) 'X' >>> f.__exit__(None, None, None) >>> f.read(1) Traceback (most recent call last): File "<stdin>", line 1, in <module> ValueError: I/O operation on closed file 这样要打开一个文件,处理它的内容,并且保证关闭它,你就可以简简单单地这样做: with open("x.txt") as f: data = f.read() do something with data 我的补充: 数据库的连接好像也可以和with一起使用,我在一本书上看到以下内容: conn = sqlite.connect("somedb") with conn: conn.execute("insert into sometable values (?,?)",("foo","bar")) 在这个例子中,commit()是在所有with数据块中的语句执行完毕并且没有错误之后自动执行的,如果出现任何的异常,将执行rollback() 操作,再次提示异常。