Utiliser Return et Yield ensemble dans les générateurs Python

Dans Python 2, une instruction return dans une fonction génératrice qui utilisait également le rendement entraînerait une erreur. Cependant, dans Python 3.3, un changement subtil s'est produit.

Démonstration de code

Considérez le code Python 3.3 suivant :

def f():
  return 3
  yield 2
  
x = f()
print(x.__next__())

Explication

Dans ce code, la fonction f comprend à la fois une instruction return et une instruction rendement. Lorsque la fonction est appelée, l'instruction return est exécutée en premier et la valeur 3 est renvoyée. Par conséquent, l'instruction rendement n'est pas exécutée.

Lorsque le générateur x est itéré en appelant sa méthode next, une exception StopIteration est levée avec une valeur de 3. Cela signifie que l'instruction return n'est pas exécutée. l'itérateur du générateur est épuisé et la valeur renvoyée par l'instruction return est disponible comme attribut de valeur de l'exception.

Nouveau mécanisme en Python 3.3

Selon PEP 380 , ce comportement est une nouvelle fonctionnalité introduite dans Python 3.3. Cela équivaut à écrire :

def f():
  yield 3
  raise StopIteration

Exemple avec Yield from

L'exemple suivant montre comment ce comportement affecte les générateurs délégués à l'aide de la syntaxe Yield from :

def f():
    return 1
    yield 2

def g():
    x = yield from f()
    print(x)

# g is still a generator so we need to iterate to run it:
for _ in g():
    pass

Dans ce cas, l'instruction rendement de délègue au générateur f. Puisque f renvoie une valeur mais n'a plus d'instructions rendement, la valeur 1 est imprimée. Cependant, l'instruction rendement 2 dans f n'est pas exécutée.