上下文管理器允許你在有需要的時候,精確地分配和釋放資源。
使用上下文管理器最廣泛的案例就是with
語句了。
想象下你有兩個需要結(jié)對執(zhí)行的相關(guān)操作,然后還要在它們中間放置一段代碼。
上下文管理器就是專門讓你做這種事情的。舉個例子:
with open('some_file', 'w') as opened_file:
opened_file.write('Hola!')
上面這段代碼打開了一個文件,往里面寫入了一些數(shù)據(jù),然后關(guān)閉該文件。如果在往文件寫數(shù)據(jù)時發(fā)生異常,它也會嘗試去關(guān)閉文件。上面那段代碼與這一段是等價的:
file = open('some_file', 'w')
try:
file.write('Hola!')
finally:
file.close()
當(dāng)與第一個例子對比時,我們可以看到,通過使用with
,許多樣板代碼(boilerplate code)被消掉了。 這就是with
語句的主要優(yōu)勢,它確保我們的文件會被關(guān)閉,而不用關(guān)注嵌套代碼如何退出。
上下文管理器的一個常見用例,是資源的加鎖和解鎖,以及關(guān)閉已打開的文件(就像我已經(jīng)展示給你看的)。
讓我們看看如何來實現(xiàn)我們自己的上下文管理器。這會讓我們更完全地理解在這些場景背后都發(fā)生著什么。