@contextlib.contextmanager
- 作用:使用该装饰器,让一个"生成器"变成一个"上下文管理器"
import contextlib
@contextlib.contextmanager
def test():
print(1) # __enter__()
yield
print(2) # __exit__()
with test() as x:
print(3)
1
3
2
import contextlib
@contextlib.contextmanager
def test():
print(1) # __enter__()
yield "abc"
print(2) # __exit__()
with test() as x:
print(3, x)
1
3 abc
2
添加了@contextlib.contextmanager
装饰器的生成器方法,会把yield
前面的所有语句当作__enter__()
,把yield
后面的所有语句当作__ex