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")