Python上下文管理器深入講解

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

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

引子

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

大家之前都知道，使用Python打開文件的時(shí)候最好要使用with語句，因?yàn)檫@樣就算在文件操作中出現(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

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

以下為一個(gè)示例

class TestContext:
    def __init__(self, val):
        self.val = val
    def __enter__(self):
        print('進(jìn)入enter')
        return self.val
    def __exit__(self, exc_type, exc_value, exc_tb):
        print("進(jìn)入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("進(jìn)入with內(nèi)部")
    raise ValueError
    print('t: %s' % t)

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

不難看出，語句的調(diào)用順序

先進(jìn)入__enter__()中，該方法的返回值賦值給上述代碼中with后邊的t。如果with下的語句出現(xiàn)了異常就會先執(zhí)行__exit__()語句中的內(nèi)容。如果沒有發(fā)生異常，就在執(zhí)行完畢with下方語句之后執(zhí)行__exit__()語句。