Python中的with...as的用法
阿新 • • 發佈:2019-02-04
這個語法是用來代替傳統的try...finally語法的。
with EXPRESSION [ as VARIABLE] WITH-BLOCK
基本思想是with所求值的物件必須有一個__enter__()方法,一個__exit__()方法。
緊跟with後面的語句被求值後,返回物件的__enter__()方法被呼叫,這個方法的返回值將被賦值給as後面的變數。當with後面的程式碼塊全部被執行完之後,將呼叫前面返回物件的__exit__()方法。
- file = open("/tmp/foo.txt")
- try:
- data = file.read()
-
finally:
- file.close()
使用with...as...的方式替換,修改後的程式碼是:
- with open("/tmp/foo.txt") as file:
- data = file.read()
- #!/usr/bin/env python
- # with_example01.py
- class Sample:
- def __enter__(self):
- print "In __enter__()"
- return "Foo"
-
def __exit__(self, type, value, trace):
- print "In __exit__()"
- def get_sample():
- return Sample()
- with get_sample() as sample:
- print "sample:", sample
執行結果為
- In __enter__()
- sample: Foo
- In __exit__()
1. __enter__()方法被執行
2. __enter__()方法返回的值 - 這個例子中是"Foo",賦值給變數'sample'
3. 執行程式碼塊,列印變數"sample"的值為 "Foo"
4. __exit__()方法被呼叫with真正強大之處是它可以處理異常。可能你已經注意到Sample類的__exit__方法有三個引數- val, type 和 trace。這些引數在異常處理中相當有用。