¿Un autoincrementing invocable?

Question

Tengo un método dentro del cual necesito pasar un entero cada vez mayor a otra función.

puedo hacer esto de esta manera:

def foo(i): 
    print i 

def bar(): 

    class Incrementer(object): 
     def __init__(self, start=0): 
      self.i = start 

     def __get__(self): 
      j = self.i 
      self.i += 1 
      return j 

    number = Incrementer() 
    foo(number) 
    foo(number) 
    foo(number) 

que emite correctamente 0 1 2 ... pero siento que estoy pasando por alto una manera mucho más fácil (o integrado) de hacer esto?

Answer 1

Trate itertools.count() - que hace exactamente lo que necesita:

>>> c = itertools.count() 
>>> next(c) 
0 
>>> next(c) 
1 
>>> next(c) 
2 

Answer 2

En general, si usted necesita para conservar el estado entre una llamada a una función y el siguiente, lo que quiere es o bien un objeto (su solución) o un generador. En algunos casos, uno será más simple que el otro, pero no hay nada incorrecto con la forma en que lo ha hecho, en principio (aunque parece que tiene algunos problemas con la implementación).

La sugerencia de Sven, itertools.count(), es un generador. Su puesta en práctica es algo como esto:

def count(): 
    i = 0 
    while True: 
     yield i 
     i += 1 

Ahora, si usted quiere que sea exigible como una función, en lugar de tener que hacer next(c), se podría definir un envoltorio que lo hizo tan:

def count(c=itertools.count()): 
    return next(c) 

O lo inevitable lambda de una línea:

count = lambda c=itertools.count(): next(c) 

Entonces count() devuelve el siguiente número entero cada vez que la llame.

Por supuesto, si usted quiere ser capaz de crear cualquier número de funciones que se puede llamar, cada uno con su propio contador, se puede escribir una fábrica para que:

def counter(): 
    return lambda c=itertools.count(): next(c) 

Entonces es:

c = counter() 
print c() # 0 
print c() # 1 
# etc 

Esto todavía me parece más simple que un objeto, pero no por mucho. Si su estado o lógica fuera más complejo, la encapsulación del objeto podría ganar.