如何判断发电机是否刚刚启动？

Question

我想要一个函数，is_just_started其行为类似于以下内容：

>>> def gen(): yield 0; yield 1
>>> a = gen()
>>> is_just_started(a) 
True
>>> next(a)
0
>>> is_just_started(a) 
False
>>> next(a)
1
>>> is_just_started(a) 
False
>>> next(a)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>> is_just_started(a)
False

如何实现此功能？

我看了看.gi_running属性，但它似乎用于其他用途。

如果我知道需要发送到生成器的第一个值，则可以执行以下操作：

def safe_send(gen, a):
    try:
        return gen.send(a)
    except TypeError as e:
        if "just-started" in e.args[0]:
            gen.send(None)
            return gen.send(a)
        else:
            raise

但是，这似乎很可恶。

Answer 1

这仅适用于Python 3.2+：

>>> def gen(): yield 0; yield 1
... 
>>> a = gen()
>>> import inspect
>>> inspect.getgeneratorstate(a)
'GEN_CREATED'
>>> next(a)
0
>>> inspect.getgeneratorstate(a)
'GEN_SUSPENDED'
>>> next(a)
1
>>> inspect.getgeneratorstate(a)
'GEN_SUSPENDED'
>>> next(a)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>> inspect.getgeneratorstate(a)
'GEN_CLOSED'

因此，请求的功能是：

import inspect

def is_just_started(gen):
    return inspect.getgeneratorstate(gen) == inspect.GEN_CREATED:

出于好奇，我研究了CPython以弄清它是如何确定的……显然，它看起来generator.gi_frame.f_lasti是“最后尝试的指令在字节码中的索引”。如果是，-1那么它还没有开始。

这是py2版本：

def is_just_started(gen):
    return gen.gi_frame is not None and gen.gi_frame.f_lasti == -1