python - 如何检查是否使用了生成器？

Question

是否可以知道是否使用了生成器？即

def code_reader(code):
   for c in code:
        yield c

code_rdr = code_reader(my_code)

a = code_rdr.next()

foo(code_rdr)

foo 调用后，我想知道 .next() 是否被 foo 在 code_rdr 上调用了或不。当然，我可以用一些带有 next() 调用计数器的类来包装它。有什么简单的方法吗？

Answer 1

Python 3.2+ 有 inspect.getgeneratorstate() .所以你可以简单地使用 inspect.getgeneratorstate(gen) == 'GEN_CREATED':

>>> import inspect
>>> gen = (i for i in range(3))
>>> inspect.getgeneratorstate(gen)
'GEN_CREATED'
>>> next(gen)
0
>>> inspect.getgeneratorstate(gen)
'GEN_SUSPENDED'