python 中 with 语句最常用在打开文件，处理文件的输入输出操作，至于为什么这么做，最笼统的解释就是释放资源。
上下文管理器
上下文管理器(context manager)是用来在 python 里面解决资源管理操作时造成的资源泄露问题。
一个很简单的例子:
- for x in range(10000000):
- OSError: [Errno 23] Too many open files in system: 'test.txt'
python 就采用上下文管理器来解决这个问题，最典型的应用是 with 语句。
上面的代码就可以写成：
- for x in range(10000000):
- with open('test.txt', 'w') as f:
用 with 语句读写文件等同于下面的代码：
- f = open('test.txt', 'w')
不过这样的代码略显冗余，也会容易漏写，而且Python的内置file类型是支持上下文管理协议的，所以更倾向于用 with 语句。
上下文管理器的实现
要实现上下文管理器，必须实现两个方法 – 一个负责进入语句块的准备操作，另一个负责离开语句块的善后操作。也就是说Python 类包含两个特殊的方法，分别名为：__enter__ 以及 __exit__ 。
上下文管理器有两种方法实现。方法一：类实现 __enter__ 以及 __exit__ 方法，方法二：contextlib 模块装饰器和生成器实现。
1.通过类实现实现__enter__以及__exit__方法
- def __init__(self, name, method):
- print('calling __init__ method')
- self.file = open(name, method)
- print('calling __enter__ method')
- def __exit__(self, exc_type, exc_val, exc_tb):
- print('calling __exit__ method')
- if __name__ == '__main__':
- with File('text.txt', 'w') as f:
执行 with 语句出现异常时，会把 exception_type、exception_value 和 traceback 传入 __eixt__ 方法，如果要处理抛出的异常可以这么写：
- [/code][code] def __enter__(self):
- print('__enter__ called')
- def __exit__(self, exc_type, exc_value, exc_tb):
- print(f'exc_type: {exc_type}')
- print(f'exc_value: {exc_value}')
- print(f'exc_traceback: {exc_tb}')
- print('exception handled')
- # 正常返回需要添加 return True,否则仍会抛出异常
- raise Exception('exception raised').with_traceback(None)
- exc_value: exception raised
- from contextlib import contextmanager
- [/code][code]@contextmanager
- def read_file(name, mode):
- if __name__ == '__main__':
- with read_file('text.txt', 'r') as f:
- for line in f.readlines():
函数 read_file() 是一个生成器，执行 with 语句时，会打开文件，执行到 yiled 时会把对象返回给 as 后的参数，with 语句执行完后，会继续执行 yiled 之后的语句。
以上就是本篇文章全部。
结合自己的理解以及专栏学习笔记所写。
|
|
转播
