Programmation PYTHON - Zenk - Security - Repository

148
Éléments du langage
DEUXIÈME PARTIE
Generators
La méthode la plus simple pour rendre un objet compatible avec les itérations est d’y
implémenter directement la méthode next() et de renvoyer self dans __iter__().
Iterator simple
>>> class Iterable:
... index = 0
... def __iter__(self):
... return self
... def next(self):
... if self.index > 5:
... raise StopIteration
... self.index += 1
... return self.index
...
>>> for element in Iterable():
... print(element)
...
1
2
3
4
5
6
À SAVOIR Gestion des iterators avec itertool
Le module itertool, présenté dans le chapitre 8, fournit des utilitaires rapides de création et de manipulation
d’iterators.
Les generators permettent de générer de manière très simple et très puissante des iterators.
La création d’un iterator par le biais d’un generator se résume à l’écriture d’une
fonction qui parcourt les éléments de la séquence. Au lieu de retourner ces éléments
par la directive return, la fonction doit faire appel à la directive yield, qui sert à
définir un point de sauvegarde.
Cette fonction peut ensuite être utilisée dans une boucle for sans avoir à implémenter
toute la garniture nécessaire à un iterator, ou à gérer la levée d’une exception
StopIteration.
Generator simple
>>> def iterable():
... print('début de boucle')