Utilizando geradores (generators)

O Python 2.3 introduziu um novo conceito para a linguagem: o de geradores, como o abaixo:

   1 def deznumeros():
   2     print "Parte A"
   3     for i in range(10):
   4         print "Parte B (num %d)" % i
   5         yield i
   6         print "Parte C (num %d)" % i 
   7     print "Parte D"

Aparentemente, apenas mais uma função, porém com uma palavra-chave nova. Porém, como especificado na própria PEP 255, esta não é uma simples função, e sim uma função geradora. Para comprovar, apenas execute-a:

   1 >>> deznumeros
   2 <function deznumeros at 0x40207a74>
   3 >>> variavel = deznumeros()
   4 >>> variavel
   5 <generator object at 0x40209cac>

Ops, à primera vista, uma simples função, mas o resultado dela é muito diferente do que o de uma função qualquer.

Por dentro de um objeto gerador

Sim, mas para quê serve um objeto como o que obtivemos no exemplo anterior? Para isso, a função dir vem para ajudar:

   1 >>> dir(variavel)
   2 ['__class__', '__delattr__', '__doc__', '__getattribute__', '__hash__', '__init__', '__iter__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__setattr__', '__str__', 'gi_frame', 'gi_running', 'next']
   3 >>> # Como a única coisa significativa é o objeto next, o exploraremos mais
   4 >>> print variavel.next
   5 <method-wrapper object at 0x40209b8c>
   6 >>> print '"%s"' % variavel.next()
   7 Parte A
   8 Parte B (num 0)
   9 "0"
  10 >>> print '"%s"' % variavel.next()
  11 Parte C (num 0)
  12 Parte B (num 1)
  13 "1"

Algo familiar? Sim, na mosca, estamos recebendo cada um dos objetos que demos na função. Em outras palavras, em cada next() recebemos, em ordem, um dos números do range(10). Mais interessante ainda! É como se a função desse uma pausa e continuasse quando nós quisessemos! Perceba que a Parte A foi executada apenas uma vez e que a Parte C é executada no início da próxima, e não no fim da mesma em que está contida. Vamos então continuar até o final para ver no que dá:

   1 >>> # ...
   2 >>> print '"%s"' % variavel.next()
   3 Parte C (num 8)
   4 Parte B (num 9)
   5 "9"
   6 >>> # Ops, esse foi o último... E a Parte D ? E a Parte C do 9 ?
   7 >>> print '"%s"' % variavel.next()
   8 Parte C (num 9)
   9 Parte D
  10 Traceback (most recent call last):
  11   File "<stdin>", line 1, in ?
  12 StopIteration
  13 >>> print '"%s"' % variavel.next()
  14 Traceback (most recent call last):
  15   File "<stdin>", line 1, in ?
  16 StopIteration

Nada foi perdido e, para avisar que a função terminou, a exceção StopIteration foi levantada. Essa é aquele tipo de exceção que não quer dizer que o seu programa está com algo de errado, apenas quer dizer que a iteração sobre o gerador terminou, ou seja, acabaram as palavras-chave yield.

Perguntas básicas que podem surgir na sua mente

Por favor, leitor, ajude a complementar essa seção.

O que pode ser passado para um gerador?

Tudo. Desde números até módulos e outros geradores, ou seja, qualquer objeto ou coisa que possa ser transformada em objeto. Você pode usar geradores para passar classes para um programa, passar funções, passar listas de vários números em uma determinada ordem, passar pontos em um gráfico ou até passar as letras do alfabeto! O que der na telha!

Onde na função eu posso usar yield ?

Digamos que em qualquer lugar que você possar colocar um print =) . Se é isso que você estava pensando, é possível sim:

   1 def teste (N1, N2, N3):
   2     # É claro que esse é um exemplo infeliz que eu inventei...
   3     yield N1
   4     for i in range(N2, N3):
   5         for j in range(N2, N3, N1):
   6             yield i * j
   7     if N1 > N2 > N3:
   8         yield N1 * N2 * N3
   9     if N1 + N2 + N3 < 10:
  10         yield N1 + N2 + N3
  11     print "Obrigado pela preferência"

E agora?

Bom, em duas palavras: use geradores =). Bom, na verdade foram duas palavras e um smile, mas isso faz diferença?

FelipeAlmeidaLessa