第八章 Python文件操作 04-with语句

with语句

1. 为什么要使用with

无论是在读取文件，还是在写文件的时候，我们需要先使用open语句，建立一个程序与文件的连接，获取到数据流。然后对文件进行读、写的操作。操作之后需要将这个连接断掉，释放资源。

# 1. 建立程序与文件的连接
read_data = open("./test.py", "r", encoding="utf8")
# 2. 读取数据
read_data.readline()
# 3. 释放资源
read_data.close()

在实际的操作过程中，有可能会出现资源没有释放，大概两种情况会导致：

• 自己忘了写close了
• 读取数据过程中出现异常，导致后面的close没有执行

第一个我们可以通过更加细心来解决，而第二个我们可以通过try语句来解决:

# 1. 建立程序与文件的连接
read_data = open("./test.py", "r", encoding="utf8")
try:
    # 2. 读取数据
    read_data.readline()
finally:
    # 3. 释放资源
    read_data.close()

问题的确解决了，但是这样的代码看着就非常麻烦了。那么能不能简化这个过程呢？with就出现了，我们可以将上述的语句修改如下：

with open("./test.py", "r", encoding="utf8") as read_data:
    read_data.readline()

2. with语句是什么

with的基本语法是：

with 对象获取方式 as 对象:
    对象的操作

可以在with后面获取到一个对象，然后在with的代码段中使用这个对象。但是这个对象并不是随便都可以的，需要实现两个魔术方法：

• _enter_: 通过with获取对象的时候调用，返回一个对象
• _exit_: 代码段执行结束，或者异常结束的时候会执行

class Demo:
    def __enter__(self):
        print("Demo Enter")
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("Demo Exit")
        print(f"{exc_type = }")
        print(f"{exc_val = }")
        print(f"{exc_tb = }")

    def test_exception(self):
        print(10 / 0)


with Demo() as d:
    d.test_exception()
    print("aaa")

3. 文件操作中的with

在进行文件读写的时候，open返回的是_io.TextIOWrapper对象，而在这个对象中已经实现了__enter__和__exit__，并且在__exit__中完成了对数据流的关闭操作。因此，我们在进行文件操作的时候，就可以直接简化如下:

# 使用with操作文件，完成文件拷贝
def copy_file(src, dst):
    with open(src, "rb") as read_data, open(dst, "wb") as write_data:
        while len(data := read_data.read(1024)) != 0:
            write_data.write(data)
            write_data.flush()


copy_file("./相思.txt", "./相思2.txt")