快捷導(dǎo)航

python with使用詳解

更新時(shí)間：2025年11月03日 10:07:06 作者：令狐掌門

with語(yǔ)句是Python中一種用于資源管理的語(yǔ)法結(jié)構(gòu),主要用于確保資源（如文件、鎖、數(shù)據(jù)庫(kù)連接等）在使用后被正確釋放,即使發(fā)生異常也不會(huì)遺漏清理操作,本文給大家介紹python with使用,感興趣的朋友跟隨小編一起看看吧

1. Pythonwith語(yǔ)句的原理

with語(yǔ)句是Python中一種用于資源管理的語(yǔ)法結(jié)構(gòu)，主要用于確保資源（如文件、鎖、數(shù)據(jù)庫(kù)連接等）在使用后被正確釋放，即使發(fā)生異常也不會(huì)遺漏清理操作。它本質(zhì)上是基于上下文管理器（Context Manager）協(xié)議實(shí)現(xiàn)的。

核心原理：

with語(yǔ)句會(huì)調(diào)用上下文管理器的兩個(gè)特殊方法：
- __enter__()：進(jìn)入with塊時(shí)調(diào)用，返回一個(gè)值（通常是資源對(duì)象本身），這個(gè)值可以被as關(guān)鍵字賦值給變量。
- __exit__(exc_type, exc_value, traceback)：退出with塊時(shí)調(diào)用，無(wú)論是否發(fā)生異常都會(huì)執(zhí)行。它接收三個(gè)參數(shù)（異常類型、異常值、回溯信息），如果返回True，則抑制異常（不拋出）；否則，異常會(huì)繼續(xù)傳播。
with語(yǔ)句相當(dāng)于簡(jiǎn)化版的try...finally結(jié)構(gòu)：
- 進(jìn)入時(shí)執(zhí)行__enter__()（類似于try前的準(zhǔn)備）。
- 執(zhí)行with塊內(nèi)的代碼。
- 退出時(shí)執(zhí)行__exit__()（類似于finally塊中的清理）。
好處：自動(dòng)處理資源釋放，避免手動(dòng)編寫try...except...finally塊，減少錯(cuò)誤（如忘記關(guān)閉文件）。

如果沒(méi)有異常，__exit__的參數(shù)均為None。這使得with特別適合“獲取-使用-釋放”模式的操作。

2.with語(yǔ)句的用法

with語(yǔ)句的基本語(yǔ)法：

with 上下文管理器 [as 變量]:
    # 在這里使用資源
    # 塊結(jié)束時(shí)自動(dòng)清理

示例1：文件操作（最常見(jiàn)用法）

with open('example.txt', 'r') as file:
    content = file.read()
    print(content)
# 文件在with塊結(jié)束后自動(dòng)關(guān)閉，無(wú)需手動(dòng)調(diào)用file.close()

原理：open()返回的文件對(duì)象實(shí)現(xiàn)了上下文管理器協(xié)議，__enter__返回文件對(duì)象本身，__exit__關(guān)閉文件。

示例2：多上下文管理器

Python支持嵌套或并列使用多個(gè)with：

with open('input.txt', 'r') as infile, open('output.txt', 'w') as outfile:
    content = infile.read()
    outfile.write(content.upper())

示例3：處理異常

try:
    with open('nonexistent.txt', 'r') as file:
        print(file.read())
except FileNotFoundError:
    print("文件不存在")

如果with塊內(nèi)拋出異常，__exit__仍會(huì)執(zhí)行清理，然后異常傳播到外層。

with常用于：

文件I/O（open()）。
線程鎖（threading.Lock()）。
數(shù)據(jù)庫(kù)連接（許多庫(kù)如SQLAlchemy支持）。
臨時(shí)更改（如os.chdir()的上下文版本）。

3. 如何定義一個(gè)對(duì)象使其支持with語(yǔ)句

要讓自定義對(duì)象支持with，需要實(shí)現(xiàn)上下文管理器協(xié)議。有兩種主要方式：類實(shí)現(xiàn)和生成器實(shí)現(xiàn)（使用contextlib模塊）。

方式1：通過(guò)類實(shí)現(xiàn)enter和exit

定義一個(gè)類，添加__enter__和__exit__方法。

class MyResource:
    def __init__(self, name):
        self.name = name
        print(f"初始化資源: {name}")
    def __enter__(self):
        print(f"進(jìn)入with塊: 獲取資源 {self.name}")
        return self  # 返回對(duì)象本身，供as使用
    def __exit__(self, exc_type, exc_value, traceback):
        print(f"退出with塊: 釋放資源 {self.name}")
        if exc_type is not None:
            print(f"發(fā)生異常: {exc_type}")
            return False  # 不抑制異常，讓它傳播
        return True  # 正常退出
# 用法
with MyResource("數(shù)據(jù)庫(kù)連接") as res:
    print(f"使用資源: {res.name}")
    # raise ValueError("模擬異常")  # 可以測(cè)試異常處理

輸出示例（無(wú)異常）：

初始化資源: 數(shù)據(jù)庫(kù)連接
進(jìn)入with塊: 獲取資源 數(shù)據(jù)庫(kù)連接
使用資源: 數(shù)據(jù)庫(kù)連接
退出with塊: 釋放資源 數(shù)據(jù)庫(kù)連接

方式2：使用contextlib模塊的@contextmanager裝飾器（基于生成器）

這是一種更簡(jiǎn)潔的方式，適合臨時(shí)上下文，不需要定義整個(gè)類。

from contextlib import contextmanager
@contextmanager
def my_resource(name):
    print(f"進(jìn)入with塊: 獲取資源 {name}")
    yield name.upper()  # yield前的代碼相當(dāng)于__enter__，yield的值供as使用
    print(f"退出with塊: 釋放資源 {name}")  # yield后的代碼相當(dāng)于__exit__
# 用法
with my_resource("文件鎖") as res:
    print(f"使用資源: {res}")

輸出：