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