python - 没有 Yield 的上下文管理器

Question

我能否有一个偶尔不产生的上下文管理器，在这种情况下，with 语句中的代码根本不会执行？

import contextlib

@contextlib.contextmanager
def MayNotYield(to_yield):
  if to_yield:
    yield

with MayNotYield(True):  
  print 'This works.'

with MayNotYield(False):  
  print 'This errors.'

我可以要求用户用 try-catch 包装 with 语句，但这不是首选。我也可以执行以下操作，但它也很丑陋。

import contextlib

@contextlib.contextmanager
def AlwaysYields(to_yield):
  if to_yield:
    yield 1
  else:
    yield 2

with AlwaysYields(True) as result:
  if result == 1:
    print 'This works.'

Answer 1

另一种选择是只使用常规生成器而不是上下文管理器；一个普通的生成器不会有这个限制。但是你必须将它与“for”结构一起使用，而不是使用“with”:

def MayNotYield(to_yield):
   if to_yield:
      yield

for _ in MayNotYield(True):
   print('This prints.')

for _ in MayNotYield(False):
   print('This does not.')