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

python with as的用法

關系 eth return int 簡潔 機制 self clas blog

以下為轉載https://www.cnblogs.com/DswCnblog/p/6126588.html

with。。as。。一個使用場景是文件處理,你需要獲取一個文件句柄,從文件中讀取數據,然後關閉文件句柄。

普通的文件處理如下:

1 file = open("/tmp/foo.txt")
2 data = file.read()
3 file.close()#文件使用完畢後必須關閉,因為文件對象會占用操作系統的資源,並且操作系統同一時間能打開的文件數量也是有限的

由於文件讀寫時都有可能產生IOError,一旦出錯,後面的f.close()就不會調用。所以,為了保證無論是否出錯都能正確地關閉文件,我們可以使用try ... finally來實現:

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()

with所求值的對象必須有一個__enter__()方法,一個__exit__()方法。

 1 class Sample:
 2     def __enter__(self):
 3         print "In __enter__()
" 4 return "Foo" 5 6 def __exit__(self, type, value, trace): 7 print "In __exit__()" 8 9 def get_sample(): 10 return Sample() 11 12 with get_sample() as sample: 13 print "sample:", sample

運行如下:

1 In __enter__()
2 sample: Foo
3 In __exit__()

運行過程:

1. __enter__()方法被執行,返回的對象賦值給變量‘sample‘

3. 執行代碼塊,打印變量"sample"的值為 "Foo"

4. __exit__()方法被調用

with真正強大之處是它可以處理異常。可能你已經註意到Sample類的__exit__方法有三個參數- val, type 和 trace。 這些參數在異常處理中相當有用。我們來改一下代碼,看看具體如何工作的。

 
class Sample:
    def __enter__(self):
        return self
 
    def __exit__(self, type, value, trace):
        print "type:", type
        print "value:", value
        print "trace:", trace
 
    def do_something(self):
        bar = 1/0
        return bar + 10
 
with Sample() as sample:
    sample.do_something()

這個例子中,with後面的get_sample()變成了Sample()。這沒有任何關系,只要緊跟with後面的語句所返回的對象有__enter__()和__exit__()方法即可。此例中,Sample()的__enter__()方法返回新創建的Sample對象,並賦值給變量sample。

代碼執行後:

 1 bash-3.2$ ./with_example02.py
 2 type: <type exceptions.ZeroDivisionError>
 3 value: integer division or modulo by zero
 4 trace: <traceback object at 0x1004a8128>
 5 Traceback (most recent call last):
 6   File "./with_example02.py", line 19, in <module>
 7     sample.do_something()
 8   File "./with_example02.py", line 15, in do_something
 9     bar = 1/0
10 ZeroDivisionError: integer division or modulo by zero

實際上,在with後面的代碼塊拋出任何異常時,__exit__()方法被執行。正如例子所示,異常拋出時,與之關聯的type,value和stack trace傳給__exit__()方法,因此拋出的ZeroDivisionError異常被打印出來了。開發庫時,清理資源,關閉文件等等操作,都可以放在__exit__方法當中。

因此,Python的with語句是提供一個有效的機制,讓代碼更簡練,同時在異常產生時,清理工作更簡單。

python with as的用法