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

Python with语句

news 来源：原创 2024/7/1 20:08:17

with 语句介绍：

python的with语句是一种用于简化资源管理的语法结构。

它主要用于确保在代码块执行完毕后，正确地释放资源，例如文件操作、锁、网络连接等。

with语句可以确保即使在发生异常的情况下，资源也能被正确地关闭和清理。

只要一个类实现了以下两个方法就可以使用with语句。

def __enter__(self):pass
def __exit__(self, exc_type, exc_value, traceback):pass

一般是上下文协议的对象或者资源管理器中使用with。

实现了__enter__和__exit__都是上下文或者资源管理器。

常见的使用

with open('a.txt', 'r') as f:content = f.read()print(content)

使用open函数以只读模式打开文件a.txt。with语句确保文件在操作完成后会被正确关闭，即使发生异常也是如此。文件的内容被读取到content变量中。打印出文件的内容。

显然这是没有营养的~

不知道我们没有思考过这么一个问题？上述代码中的f变量是一个什么是东西?

我在刚接触python的这个语法的时候，会认为f就是open()对象的返回值。后来发现是我肤浅了。

看下这段代码，自定义一个资源管理器类实现__enter__和__exit__。

class MyResourceManager:def __enter__(self):print("Resource is being init...")a = 1print(a) # 1def __exit__(self, exc_type, exc_value, traceback):b = 2print(b) # 2print("Resource is being cleaned up.")with MyResourceManager() as resource:  # resource 不是上下文管理器，是__enter__方法的返回值print("Using the resource")print(type(resource))  # <class 'NoneType'>print(resource)        # Noneprint("Test End")

你会发现这里的resource是一个None，按道理应该是MyResourceManager类型的。

直接说一下结果，with MyResourceManager() as resource这里的resource是MyResourceManager里__enter__函数的返回值。大部分使用with好像都是前面对象本身的类型，那是因为__enter__函数返回了self对象本身。

with XX as resource 不是上下文或资源管理器对象，是enter方法的返回值 。

在看份代码，也许还会有发现~

class MyResourceManager:test_lst = ["a", "b", 12]def __enter__(self):print("Resource is being init...")print(self.test_lst) # ['a', 'b', 12]return selfdef __exit__(self, exc_type, exc_value, traceback):self.test_lst[2] = "c"print("Resource is being cleaned up")with MyResourceManager() as resource:print("Using the resource")print(type(resource))    # <class '__main__.MyResourceManager'>print(resource.test_lst) # ['a', 'b', 12]lst = resource.test_lst  # 取资源管理器里面的值 print(lst)               # ['a', 'b', 12]print(lst)                   # ['a', 'b', 'c']
print("Test End")