深入理解 Python 中的 with as 语句

BCKQS • 2025-04-23 18:08 • 编程

Python 的 with as 语句是一个高级使用场合，它能够明显地减少代码中的重复逻辑，提高代码质量。本文从多个方面对 with as 语句做详细的阐述。

一、简介

with as 是 Python 2.5 引入的一种新语法，主要用于对资源进行操作，确保不管使用过程中是否发生异常，都会执行必要的“清理”操作。

例如，使用 with as 来读取一个文件，可以确保文件在使用完毕后自动关闭，而不必担心在使用过程中出现的异常导致文件句柄没有被关闭的问题。

    with open("file.txt") as f:
        # 使用文件对象
    # 文件对象在这里已经关闭

二、contextmanager 装饰器

contextmanager 装饰器是 with as 实现的关键。装饰器是 Python 中一种常见的语法糖，用于修改、扩展或包装函数或类。使用 contextmanager 装饰器，可以方便地编写支持 with as 语句的上下文管理器。

一个简单的上下文管理器的实现：

    from contextlib import contextmanager

    @contextmanager
    def my_with():
        # __enter__ 中的语句在 with as 块执行之前执行
        print("Enter my_with")

        try:
            # yield 语句之前的部分相当于 __enter__ 方法的返回值
            yield "[my_with]"
        finally:
            # __exit__ 中的语句在 with as 块执行之后执行
            print("Exit my_with")

    with my_with() as mw:
        print(mw)

三、多个上下文管理器

在 with as 块中可以使用多个上下文管理器，并且管理器的数量没有限制。

一个实际场景下的例子：

    import contextlib

    @contextlib.contextmanager
    def tag(name):
        print(f"")
        yield
        print(f"")

    with tag("div"), tag("p"), tag("span"):
        print("Hello, world!")

四、扩展 with as 功能

在 Python 的 with as 语句中，上下文管理器可以返回一个值给 with as 语句，使得值可以被 with as 语句块中的代码使用。

例如，一个寻找最大元素的上下文管理器：

    class MaxFinder(object):
        def __init__(self):
            self.max = float("-inf")

        def __enter__(self):
            return self

        def __exit__(self, type, value, traceback):
            pass

        def track(self, value):
            self.max = max(self.max, value)

    with MaxFinder() as mf:
        mf.track(5)
        mf.track(10)
        mf.track(3)

    print(mf.max)  # 输出 10

五、其他用法

除了上述功能，with as 还有其他使用场景，例如：

线程同步：Python 的标准库 threading 中提供了 Lock 和 RLock 类型，可以用于线程间同步，较为常用。
数据库管理：在 Python 连接数据库时，可以使用 with as 来管理数据库连接，避免资源泄露。
文件锁定：在多进程需求情况下，往往需要多进程对同一个文件进行操作，而 with as 能够帮助我们实现对同一个文件的锁定操作。

总结

with as 语句是 Python 中的高级语法糖，用于代码的简化，并且能够避免资源泄露等问题。with as 语句的实现离不开 contextmanager 装饰器的使用，同时可以在上下文管理器中使用多个嵌套的管理器，且上下文管理器可以扩展功能并返回值给 with as 语句。

原创文章，作者：BCKQS，如若转载，请注明出处：https://www.506064.com/n/371796.html