With 语法#

1
f = open('1.txt')
2
for line in f:
3
  ...
4
f.close()
5
# 上面的代码会导致文件读取期间出现异常的话文件句柄无法释放
6
# 对于这种情况,我们可以使用with来解决
7
with open('1.txt') as f:
8
  for line in f:
9
    ...

上下文管理器#

那么 with 后面的代码是可以随意地吗?

1
with context_expression [as target(s)]:
2
  with-body

答案是否定的,with 后的代码需要实现上下文管理器协议

在 python 中,只要实现了刷新方法就实现了上下文管理器协议

__enter__: 在进入 with 前调用,会赋值给 with 的 target
__exit__: 退出 with 调用,一般作为异常处理

1
```python
2
class TestContext:
3

4
    def __enter__(self):
5
        print('__enter__')
6
        return 1
7

8
    def __exit__(self, exc_type, exc_value, exc_tb):
9
        print('exc_type: %s' % exc_type)
10
        print('exc_value: %s' % exc_value)
11
        print('exc_tb: %s' % exc_tb)
12

13
with TestContext() as t:
14
    print('t: %s' % t)

Contextlib 模块#

对于需要上下文管理的场景,除了自己实现 enter 和 exit,另一种方法就是使用 contextlib 模块

使用了 contextlib,就可以把上下文当做一个装饰器来使用

1
from contextlib import contextmanager
2

3
@contextmanager
4
def test():
5
    print('before')
6
    yield 'hello'
7
    print('after')
8

9
with test() as t:
10
    print(t)

执行 test() 方法，先打印出 before
执行 yield 'hello'，test 方法返回，hello 返回值会赋值给 with 语句块的 t 变量
执行 with 语句块内的逻辑，打印出 t 的值 hello
又回到 test 方法中，执行 yield 后面的逻辑，打印出 after 不过有一点需要我们注意：在使用 contextmanager 装饰器时，如果被装饰的方法内发生了异常，那么我们需要在自己的方法中进行异常处理，否则将不会执行 yield 之后的逻辑。