当前位置：首页 > article >正文

python语言基础-5 进阶语法-5.5 上下文管理协议（with语句）

article 2024/11/24 2:10:12

声明：本内容非盈利性质，也不支持任何组织或个人将其用作盈利用途。本内容来源于参考书或网站，会尽量附上原文链接，并鼓励大家看原文。侵删。

5.5 上下文管理协议（with语句）（参考链接：https://blog.csdn.net/Ego_Bai/article/details/80873242）

5.5.1 with语句及其使用

有一些任务，可能事先需要设置，事后做清理工作。对于这种场景，Python的with语句提供了一种非常方便的处理方式。

其中一个很好的例子是文件处理，你需要获取一个文件句柄，从文件中读取数据，然后关闭文件句柄。
如果不用with语句，代码如下：

file = open("/tmp/foo.txt")
data = file.read()
file.close()

这里有两个问题：一是，可能忘记关闭文件句柄；二是，文件读取数据发生异常，没有进行任何处理。下面是处理异常的加强版本：

file = open("/tmp/foo.txt")
try:
    data = file.read()
finally:
    file.close()

这段代码运行良好，但是太冗长。这时候with便体现出了优势：不但有更优雅的语法，还可以很好的处理上下文环境产生的异常。下面是with版本的代码：

with open("/tmp/foo.txt") as file:
    data = file.read()

在操作多线程时，也有很多相关对象是支持上下文管理协议的。如下：

with some_lock:
    # do something...
    
# 上面的代码相当于
some_lock.acquire()
try:
    # do something...
finally:
    some_lock.release()

由此可见，with用来操作支持上下文管理协议的对象。

5.5.2 with是如何工作的（如何支持上下文管理协议）

一个对象要想支持上下文件管理协议，必须满足要求：对象必须有一个enter()方法，一个exit()方法。因为with语句的执行过程是：紧跟with后面的语句被求值后，返回对象的__enter__()方法被调用，这个方法的返回值将被赋值给as后面的变量。当with后面的代码块全部被执行完之后，将调用前面返回对象的__exit()__方法。

我们可以自定义一个对象并使用with。如下：

class Sample:
    def __enter__(self):
        print("in __enter__")
        return "Foo"

    def __exit__(self, exc_type, exc_val, exc_tb):  # exc_type：错误的类型；exc_val：错误类型对应的值；exc_tb：代码中错误发生的位置 
        print("in __exit__")

def get_sample():
    return Sample()

with get_sample() as sample:
    print("Sample: " ,sample)