¿Cómo sobrescribir el comportamiento de la lista de Python (iterador)?

Question

La ejecución de este:

class DontList(object): 
    def __getitem__(self, key): 
     print 'Getting item %s' % key 
     if key == 10: raise KeyError("You get the idea.") 
     return None 

    def __getattr__(self, name): 
     print 'Getting attr %s' % name 
     return None 

list(DontList()) 

Produce esto:

Getting attr __length_hint__ 
Getting item 0 
Getting item 1 
Getting item 2 
Getting item 3 
Getting item 4 
Getting item 5 
Getting item 6 
Getting item 7 
Getting item 8 
Getting item 9 
Getting item 10 
Traceback (most recent call last): 
    File "list.py", line 11, in <module> 
    list(DontList()) 
    File "list.py", line 4, in __getitem__ 
    if key == 10: raise KeyError("You get the idea.") 
KeyError: 'You get the idea.' 

cómo puedo cambiar eso para que voy a conseguir [], al tiempo que permite el acceso a dichas claves [1] etc.?

(He intentado poner en def __length_hint__(self): return 0, pero no ayuda.)

Mi caso de uso real: (para la lectura si va a ser útil, no dude en ignorar allá de este punto)

Después de aplicar a certain patch a iniparse, he encontrado un desagradable efecto secundario en mi parche. Tener __getattr__ establecido en mi clase Undefined, que devuelve un nuevo objeto Undefined. Desafortunadamente, esto significa que list(iniconfig.invalid_section) (donde isinstance(iniconfig, iniparse.INIConfig)) está haciendo esto (poner en simples print s en el __getattr__ y __getitem__):

Getting attr __length_hint__ 
Getting item 0 
Getting item 1 
Getting item 2 
Getting item 3 
Getting item 4 

Etcétera hasta el infinito.

Answer 1

Si desea anular la iteración a continuación, sólo se define el método __iter__ en su clase

Answer 2

Sólo levanta IndexError en lugar de KeyError. KeyError es para clases de tipo mapeo (por ejemplo, dict), mientras que IndexError es para secuencias.

Si define el método __getitem__() en su clase, Python generará automáticamente un iterador a partir de él. Y el iterador termina al IndexError - ver PEP234.

Answer 3

Reemplace la iteración de su clase implementando un método __iter__(). Señal de iterador cuando terminan al generar una excepción StopIteration, que es parte del protocolo de iterador normal y no se propaga aún más. He aquí una manera de aplicar eso a su clase de ejemplo:

class DontList(object): 
    def __getitem__(self, key): 
     print 'Getting item %s' % key 
     if key == 10: raise KeyError("You get the idea.") 
     return None 

    def __iter__(self): 
     class iterator(object): 
      def __init__(self, obj): 
       self.obj = obj 
       self.index = -1 
      def __iter__(self): 
       return self 
      def next(self): 
       if self.index < 9: 
        self.index += 1 
        return self.obj[self.index] 
       else: 
        raise StopIteration 

     return iterator(self) 

list(DontList()) 
print 'done' 
# Getting item 0 
# Getting item 1 
# ... 
# Getting item 8 
# Getting item 9 
# done 

Answer 4

creo que el uso de return iter([]) es el camino correcto, pero vamos a empezar a pensar cómo list() obras:

recuperar un elemento de __iter__; Si recibe un error de StopIrteration stops..then conseguir que el elemento ..

Así que sólo tienes que yield un generador de vacío en __iter__, por ejemplo (x for x in xrange(0, 0)), o simplemente iter([]))

Answer 5

como dice @Sven, ese es el error erróneo aumento. Pero ese no es el punto, el punto es que esto está roto porque es , no es algo que deba hacer: evitar que __getattr__ eleve AttributeError significa que ha anulado la metodología predeterminada de Python para probar si un objeto tiene un atributo y lo reemplazó con un nuevo uno (ini_defined(foo.bar)).

¡Pero Python ya tiene hasattr! ¿Por qué no usar eso?

>>> class Foo: 
...  bar = None 
... 
>>> hasattr(Foo, "bar") 
True 
>>> hasattr(Foo, "baz") 
False