with - what are python generators used for

¿Cómo puedo saber si un generador recién se inició?

Question

Me gustaría una función, is_just_started , que se comporte como la siguiente:

>>> 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

¿Cómo puedo implementar esta función?

Miré el atributo .gi_running pero parece ser usado para otra cosa.

Si conozco el primer valor que debe enviarse al generador, puedo hacer algo como esto:

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

Sin embargo, esto parece aborrecible.

Answer 1

Esto solo funciona en 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''

Entonces, la función solicitada es:

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

Por curiosidad, busqué en CPython para descubrir cómo estaba determinando esto ... Aparentemente se ve en generator.gi_frame.f_lasti que es el "índice del último intento de instrucción en bytecode". Si es -1 entonces aún no ha comenzado.

Aquí hay una versión de py2:

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

Answer 2

Haga un nuevo generador que simplemente rinda de su generador de interés. Establece un indicador una vez que se ha consumido el primer valor . Después, simplemente puede usar el yield from resto de los artículos.

Utilice el generador sustituto como sustituto del generador que le interesa para supervisar el estado "is_just_started".

Esta técnica no es intrusiva y se puede usar incluso en generadores para los cuales no tiene control sobre el código fuente.

Answer 3

Puede crear un iterador y establecer el indicador como la propiedad de la instancia en la clase de iterador como:

class gen(object):
def __init__(self, n):
self.n = n
self.num, self.nums = 0, []
self.is_just_started = True # Your flag
def __iter__(self):
return self
# Python 3 compatibility
def __next__(self):
return self.next()
def next(self):
self.is_just_started = False # Reset flag with next
if self.num < self.n:
cur, self.num = self.num, self.num+1
return cur
else:
raise StopIteration()

Y su función de verificación de valor sería como:

def is_just_started(my_generator):
return my_generator.is_just_started

Muestra de ejecución:

>>> a = gen(2)
>>> 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>
File "<stdin>", line 19, in next
StopIteration

Para saber la diferencia entre el iterador y el generador , verifique la diferencia entre los generadores de Python y los iteradores