对空生成器做出反应的 Pythonic 方式？ [复制]

Question

我想为生成器编写一个包装器，它检查生成器是否产生任何东西，并且（例如）如果没有则引发异常。

我会写：

def my_wrapper(input):
    if input is None:
        return

    found = False
    for elem in my_yielding_function(input):
        found = True
        yield elem

    if not found:
        raise MyException("Empty Generator")

有没有更 Pythonic 的方式来做到这一点？

有一个非常similar question，但它已经有 10 多年的历史了——也许事情已经改变了？

上下文：

很难解释 - 我正在使用一个给定的 API 函数，它可能不会产生任何结果，但在这种情况下 my 函数与空输入有区别。

Answer 1

除了避免无用的 for 循环之外，这消除了对标志的需要。您还可以将其调整为装饰器，以使其可以转发调用参数并且仍然可以重用

def check_if_empty_first(gen):
  it = gen() # This is optional, depends if you want to make it reusable, and it you want to call with check_if_empty_first(gen) or check_if_empty_first(gen())
  try:
    yield next(it)
  except StopIteration as e:
    raise MyException("Empty Generator") from e
  yield from it

装饰器版本：

from functools import wraps
def check_if_empty_first(gen):
  @wraps(gen)
  def inner(*args, **kwargs
    it = gen(*args, **kwargs)
    try:
      yield next(it)
    except StopIteration as e:
      raise MyException("Empty Generator") from e
    yield from it
  return inner