Python中with管理上下文

上下文管理器

上下文管理器本质就是能够支持with操作。

任何实现了 __enter__() 和 __exit__() 方法的对象都可称之为上下文管理器，上下文管理器对象可以使用 with 关键字。显然，文件（file）对象也实现了上下文管理器协议。

实现上下文管理器的另外方式

Python 还提供了一个 contextmanager 的装饰器，更进一步简化了上下文管理器的实现方式。通过 yield 将函数分割成两部分，yield 之前的语句在 __enter__ 方法中执行，yield 之后的语句在 __exit__ 方法中执行。紧跟在 yield 后面的值是函数的返回值。

contextmanager装饰器的底层实现就是将yield之前的方法传入到__enter__方法中，将yield之后的方法传入到__exit__方法中

使用contextmanager 装饰器需要导入模块

from contextlib import contextmanager

使用enter和exit实现一个上下文管理器

读取同级目录下的hello文件的内容

class my_open(object):def __init__(self, file_name, file_open_tyope):self.file_name = file_nameself.file_open_type = file_open_tyope# 创建上文方法def __enter__(self):print("进入上文方法")self.file = open(self.file_name, self.file_open_type)return self.file# 创建下文方法def __exit__(self, exc_type, exc_val, exc_tb):print("进入下文方法")self.file.close()def main():with my_open("hello", 'r') as file:result = file.read()print(result)if __name__ == '__main__':main()"""
进入上文方法
hello python
进入下文方法
"""

使用装饰器contextmanager实现一个上下文管理器

读取同级目录下的hello文件的内容

from contextlib import contextmanager@contextmanager
def my_open(file_name, file_open_type):# 进入上文操作print("进入上文")file = open(file_name, file_open_type)# 返回file，进行操作yield file# 进入下文操作print("进入下文")file.close()if __name__ == '__main__':with my_open("hello", "r") as file:result = file.read()print(result)
"""
进入上文
hello python
进入下文
"""