python上下文管理关键字_将 python 生成器改造为上下文管理器

本质上 contextlib.contextmanager 仍然是利用了 yield 生成器的特性，他将函数包装并增加了 __enter__ 与 __exit__ 两个方法

def contextmanager(func):

@wraps(func)

def helper(*args, **kwds):

return _GeneratorContextManager(func, args, kwds)

return helper

class _GeneratorContextManager():

def __init__(self, func, args, kwds):

self.gen = func(*args, **kwds)

self.func, self.args, self.kwds = func, args, kwds

# Issue 19330: ensure context manager instances have good docstrings

doc = getattr(func, "__doc__", None)

if doc is None:

doc = type(self).__doc__

self.__doc__ = doc

def __enter__(self):

try:

return next(self.gen)

except StopIteration:

raise RuntimeError("generator didn't yield") from None

def __exit__(self, type, value, traceback):

if type is None: