with語句常用來處理一些事務的事先處理與事后清理工作。像文件處理時要先取得文件句柄,處理完后要關閉文件句柄
file = open('tes.txt')
data = file.read()
file.close()
這里沒有對讀取數據發生的異常作任何處理,而且容易忘記把文件close掉。下面代碼添加異常處理
try:
file = open('tes.txt')
except:
print('fail to open')
try:
data = file.read()
# do something
except:
print('read err')
# do something
finally:
file.close()
這段代碼雖然運行良好,但是太冗長了。下面使用with優雅地實現文件操作
with open('tes.txt') as file:
data = file.read()
# do something
這里with會自動處理文件關閉操作。這里沒有對打開文件異常作處理,數據讀取異的處理是文件對象中定義的,因為是C寫的模塊,不知道實際處理情況。實際使用時候用try except外層處理一次應該就夠了
自定義with處理對象
定義一個對象并實現enter()與exit()方法,分別進行預處理與后處理。enter()返回一個任意對象給as使用
class Tes:
def __enter__(self):
print('in enter')
# raise ValueError('enter error') # 這里的異常會直接退出,不會執行with block與__exit__()
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print('in exit')
print(exc_type)
print(exc_val)
print(exc_tb)
if exc_type is ValueError:
return True #返回ture時 with block的ValueError異常會被攔截, 程序會正常執行
# 這個不是必需的
def dosomething(self):
print('do something')
with Tes() as t:
t.dosomething()
raise ValueError('block error')
## 運行結果
# in enter
# do something
# in exit
# <class 'ValueError'>
# block error
# <traceback object at 0x102a255c8>
with多個對象
with open('1.txt') as f1, open('2.txt') as f2:
# do something with f1, f2