Python上下文管理器深入講解

更新時間：2022年12月21日 16:59:13 作者：ZYM66

Python有三大神器，一個是裝飾器，一個是迭代器、生成器，最后一個就是今天文章的主角 -- 「上下文管理器」。上下文管理器在日常開發(fā)中的作用是非常大的，可能有些人用到了也沒有意識到這一點

引子

上下文管理器是一種簡化代碼的有力方式，其內(nèi)部也蘊含了很多Python的編程思想，今天我們就來探究一下Python的上下文管理器。

大家之前都知道，使用Python打開文件的時候最好要使用with語句，因為這樣就算在文件操作中出現(xiàn)了異常，也不會導(dǎo)致內(nèi)存泄露

代碼示例：

最簡單的文件打開方式(容易導(dǎo)致內(nèi)存泄露)

f = open("./xxx.txt")
f.close()

安全的文件打開方式，雖然相對復(fù)雜但是有效

# 打開文件
f = open('file.txt')
try:
    for line in f:
        # 讀取文件內(nèi)容 執(zhí)行其他操作
        # do_something...
finally:
    # 保證關(guān)閉文件
    f.close()

我們最經(jīng)常使用的文件打開方式

with open('file.txt', "r") as r:
	pass

如果一個對象要用with語句，那內(nèi)部要定義__enter__()方法和__exit__()方法。

以下為一個示例

class TestContext:
    def __init__(self, val):
        self.val = val
    def __enter__(self):
        print('進入enter')
        return self.val
    def __exit__(self, exc_type, exc_value, exc_tb):
        print("進入exit")
        print('exc_type: %s' % exc_type)
        print('exc_value: %s' % exc_value)
        print('exc_tb: %s' % exc_tb)
with TestContext(12) as t:
    print("進入with內(nèi)部")
    raise ValueError
    print('t: %s' % t)

運行后結(jié)果如下圖