python with as的用法
以下為轉載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的用法