Как переопределить поведение в списке Python (итератор)?

Question

Запуск этого:

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()) 

Производит это:

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.' 

Как я могу изменить это так, что я буду получать [], в то же время предоставляя доступ к этим ключам [1] и т.д.?

(я попытался положить в def __length_hint__(self): return 0, но это не помогает.)

Мой реальный случай использования: (для прочтения, если это будет полезно, не стесняйтесь игнорировать прошлое этой точки)

После применения a certain patch к неглазу, я нашел неприятный побочный эффект для моего патча. Имея __getattr__, установите на мой класс Undefined, который возвращает новый объект Undefined. К сожалению, это означает, что list(iniconfig.invalid_section) (где isinstance(iniconfig, iniparse.INIConfig)) делает это (положить в простых print с в __getattr__ и __getitem__):

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

И так далее до бесконечности.

Answer 1

Если вы хотите переопределить итерации, то просто определить метод __iter__ в классе

Answer 2

Просто поднимите IndexError вместо KeyError. KeyError предназначен для картографических классов (например, dict), а IndexError предназначен для последовательностей.

Если вы определяете метод __getitem__() в своем классе, Python автоматически генерирует из него итератор. Итератор заканчивается на IndexError - см. PEP234.

Answer 3

Переопределите способ повторения вашего класса с помощью метода __iter__(). Итератор, когда они закончены, вызывая исключение StopIteration, которое является частью обычного протокола итератора и не распространяется дальше. Вот один из способов применения, что ваш пример класса:

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

Я думаю, что с помощью return iter([]) это правильный путь, но давайте начнем думать, как list() работы:

Получить элемент из __iter__; если появляется сообщение об ошибке StopIrteration stops..then получить этот элемент ..

Итак, вы просто yield пустой генератор в __iter__, например (x for x in xrange(0, 0)), или просто iter([]))

Answer 5

Как @Sven говорит, что это неправильно ошибка поднять. Но дело не в этом, дело в том, что это не так, потому что это не то, что вы должны делать: предотвращение __getattr__ от повышения AttributeError означает, что у вас есть переопределенная методология по умолчанию для Python для проверки того, имеет ли объект атрибут и заменил его новым один (ini_defined(foo.bar)).

Но Python уже имеет hasattr! Почему бы не использовать это?

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