1. 程式人生 > >Python中的with...as的用法

Python中的with...as的用法

這個語法是用來代替傳統的try...finally語法的。 

with EXPRESSION [ as VARIABLE] WITH-BLOCK 

基本思想是with所求值的物件必須有一個__enter__()方法,一個__exit__()方法。

緊跟with後面的語句被求值後,返回物件的__enter__()方法被呼叫,這個方法的返回值將被賦值給as後面的變數。當with後面的程式碼塊全部被執行完之後,將呼叫前面返回物件的__exit__()方法。

  1. file = open("/tmp/foo.txt")  
  2. try:  
  3.     data = file.read()  
  4. finally:  
  5.     file.close()  

使用with...as...的方式替換,修改後的程式碼是:

  1. with open("/tmp/foo.txt") as file:  
  2.     data = file.read()  
  1. #!/usr/bin/env python  
  2. # with_example01.py  
  3. class Sample:  
  4.     def __enter__(self):  
  5.         print "In __enter__()"  
  6.         return "Foo"  
  7.     def __exit__(self, type, value, trace):  
  8.         print "In __exit__()"  
  9. def get_sample():  
  10.     return Sample()  
  11. with get_sample() as sample:  
  12.     print "sample:", sample  

執行結果為
  1. In __enter__()  
  2. sample: Foo  
  3. In __exit__()  

1. __enter__()方法被執行

2. __enter__()方法返回的值 - 這個例子中是"Foo",賦值給變數'sample'

3. 執行程式碼塊,列印變數"sample"的值為 "Foo"

4. __exit__()方法被呼叫with真正強大之處是它可以處理異常。可能你已經注意到Sample類的__exit__方法有三個引數- val, type 和 trace。這些引數在異常處理中相當有用。