Python 的context manager 用於確保資源在使用後被正確清理，即使發(fā)生異常也能安全釋放。 1. 使用類實現(xiàn)時，通過定義__enter__ 和__exit__ 方法控制資源的獲取與釋放，如文件操作中自動關(guān)閉文件；2. 使用contextlib.contextmanager 裝飾器可將生成器函數(shù)轉(zhuǎn)化為上下文管理器，適用於計時、臨時配置等輕量場景，yield 前代碼相當(dāng)於__enter__，finally 塊相當(dāng)於__exit__；3. 實際應(yīng)用如數(shù)據(jù)庫連接管理，使用@contextmanager 封裝連接的創(chuàng)建與關(guān)閉，確保連接最終被釋放。綜上，當(dāng)需要成對操作資源、減少try...finally 模板代碼並提升代碼可讀性與健壯性時，應(yīng)使用context manager，其核心機(jī)制是with 語句背後的__enter__ 和__exit__ 協(xié)議，能有效簡化資源管理流程並增強(qiáng)異常處理能力。

Python 的context manager（上下文管理器）是一種用於資源管理的機(jī)制，最常用在文件操作、數(shù)據(jù)庫連接、鎖的獲取釋放等場景中。它通過with語句來使用，能確保資源在使用後被正確清理，即使發(fā)生異常也能安全釋放。

下面是一個清晰、實用的Python context manager 示例：

1. 使用類實現(xiàn)context manager

 class FileManager:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.file = None

    def __enter__(self):
        print(f"Opening file: {self.filename}")
        self.file = open(self.filename, self.mode)
        return self.file

    def __exit__(self, exc_type, exc_value, traceback):
        print("Closing file...")
        if self.file:
            self.file.close()
        # 如果有異常，打印提示if exc_type is not None:
            print(f"An exception occurred: {exc_value}")
        # 返回False 表示異常不會被抑制；返回True 則會忽略異常return False

# 使用示例with FileManager("test.txt", "w") as f:
    f.write("Hello, Context Manager!")
    # 即使這裡拋出異常，文件也會被關(guān)閉

? 優(yōu)點：清晰控制__enter__和__exit__行為
? __exit__方法的三個參數(shù)用於處理異常（類型、值、追蹤）

2. 使用contextlib 簡化實現(xiàn)（裝飾器方式）

如果你不想寫一個完整的類，可以用contextlib.contextmanager裝飾器，把一個生成器變成context manager。

 from contextlib import contextmanager

@contextmanager
def timer():
    import time
    start = time.time()
    print("Timer started")
    try:
        yield # 這裡是with 語句塊執(zhí)行的地方finally:
        end = time.time()
        print(f"Time taken: {end - start:.2f} seconds")

# 使用示例with timer():
    sum(i**2 for i in range(100_000))

輸出類似：

 Timer started
Time taken: 0.02 seconds

? 適合輕量級資源管理，比如計時、臨時修改配置、加鎖等?? yield前的代碼相當(dāng)於__enter__ ， finally塊相當(dāng)於__exit__

3. 實際應(yīng)用場景舉例：數(shù)據(jù)庫連接

import sqlite3
from contextlib import contextmanager

@contextmanager
def db_connection(db_name):
    conn = sqlite3.connect(db_name)
    try:
        yield conn
    finally:
        conn.close()

# 使用with db_connection("example.db") as conn:
    cursor = conn.cursor()
    cursor.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)")
    cursor.execute("INSERT INTO users (name) VALUES (?)", ("Alice",))
    conn.commit()
# 連接自動關(guān)閉

小結(jié)：什麼時候用context manager？

• 需要成對操作資源（打開/關(guān)閉、加鎖/釋放、連接/斷開）
• 想避免try...finally的樣板代碼
• 提高代碼可讀性和健壯性

基本上就這些。只要記?。?with語句背後是__enter__和__exit__協(xié)議，用好了能少寫很多錯誤處理代碼。

以上是Python上下文管理器示例的詳細(xì)內(nèi)容。更多資訊請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！