上下文管理器是Python中用於自動(dòng)管理資源獲取與釋放的機(jī)制，通過(guò)實(shí)現(xiàn)__enter__()和__exit__()方法或使用contextmanager裝飾器實(shí)現(xiàn)；其核心作用是簡(jiǎn)化代碼結(jié)構(gòu)、避免資源洩漏，並確保資源在使用後正確釋放；常見(jiàn)應(yīng)用場(chǎng)景包括文件操作、數(shù)據(jù)庫(kù)連接、線程鎖及臨時(shí)狀態(tài)變更等；用戶(hù)可通過(guò)定義類(lèi)或結(jié)合生成器函數(shù)與裝飾器兩種方式自定義上下文管理器；實(shí)際使用時(shí)支持嵌套或多對(duì)象並行處理，且可在退出時(shí)執(zhí)行異常處理或狀態(tài)恢復(fù)邏輯。

在寫(xiě)Python 代碼時(shí)，如果你經(jīng)常處理文件、數(shù)據(jù)庫(kù)連接或者鎖機(jī)制，那麼你應(yīng)該已經(jīng)聽(tīng)說(shuō)過(guò)context manager（上下文管理器）了。它能幫你自動(dòng)管理資源的獲取和釋放，而不需要你手動(dòng)調(diào)用close 或者try/finally 塊。用好它，不僅能讓代碼更簡(jiǎn)潔，還能避免很多常見(jiàn)的資源洩漏問(wèn)題。

什麼是上下文管理器？

簡(jiǎn)單來(lái)說(shuō)，上下文管理器就是一個(gè)實(shí)現(xiàn)了__enter__()和__exit__()方法的對(duì)象。當(dāng)你使用with語(yǔ)句時(shí)，Python 就會(huì)自動(dòng)調(diào)用這兩個(gè)方法。

最常見(jiàn)的例子是打開(kāi)文件：

 with open('example.txt', 'r') as f:
    content = f.read()

在這個(gè)例子裡， open()返回的就是一個(gè)上下文管理器。進(jìn)入時(shí)打開(kāi)文件，退出時(shí)自動(dòng)關(guān)閉，不用你再操心忘記關(guān)文件了。

除了文件操作，還有像線程鎖、數(shù)據(jù)庫(kù)連接池、臨時(shí)目錄切換等場(chǎng)景，也都可以用上下文管理器來(lái)簡(jiǎn)化邏輯。

如何自己寫(xiě)一個(gè)上下文管理器？

你可以通過(guò)兩種方式自定義上下文管理器：一種是定義類(lèi)並實(shí)現(xiàn)__enter__()和__exit__() ；另一種是用contextlib.contextmanager裝飾器配合生成器函數(shù)。

方法一：用類(lèi)實(shí)現(xiàn)

class MyContext:
    def __enter__(self):
        print("進(jìn)入上下文")
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("退出上下文")
        # 返回True 可以抑制異常return False

使用方式：

 with MyContext():
    print("做點(diǎn)事")

方法二：用裝飾器生成器

from contextlib import contextmanager

@contextmanager
def my_context():
    print("進(jìn)入")
    try:
        yield
    finally:
        print("退出")

這個(gè)寫(xiě)法更輕量，適合快速封裝一些小功能。

上下文管理器的實(shí)際應(yīng)用場(chǎng)景

1. 資源清理：比如打開(kāi)文件、網(wǎng)絡(luò)連接、數(shù)據(jù)庫(kù)游標(biāo)等，用完就釋放。
2. 臨時(shí)狀態(tài)變更：比如切換工作目錄、修改全局變量後恢復(fù)原值。
3. 異常處理：在__exit__()中可以捕獲異常並決定是否繼續(xù)拋出。
4. 並發(fā)控制：結(jié)合threading.Lock 或asyncio.Lock 使用，避免競(jìng)爭(zhēng)條件。

舉個(gè)例子，你想臨時(shí)修改某個(gè)配置項(xiàng)，並在完成後恢復(fù)默認(rèn)值：

 @contextmanager
def temp_config(config, key, value):
    original = config.get(key)
    config[key] = value
    try:
        yield
    finally:
        config[key] = original

這樣在with 塊結(jié)束後，配置就會(huì)自動(dòng)還原。

上下文管理器的一些細(xì)節(jié)注意

• 如果你在__exit__()返回True，就可以阻止異常繼續(xù)傳播。

• 多個(gè)上下文可以用嵌套的方式寫(xiě)，也可以用逗號(hào)寫(xiě)在同一行：

   with open('a.txt'), open('b.txt'):
      ...

• 如果你用了contextmanager ，記得把可能出錯(cuò)的部分放在try...finally裡，確保退出邏輯一定能執(zhí)行。

基本上就這些。用起來(lái)不復(fù)雜，但真的能讓你的代碼更清晰、安全。

以上是理解和使用Python中的上下文經(jīng)理的詳細(xì)內(nèi)容。更多資訊請(qǐng)關(guān)注PHP中文網(wǎng)其他相關(guān)文章！