Можно ли определить метод __iter__ для класса? [Дубликат]

Как и в случае с Python 3.4+, создание итерации класса немного проще с помощью enum.Enum .

from enum import Enum

class Foo(Enum):
    bar = "qux"
    baz = 123

>>> print(*Foo)
Foo.bar Foo.baz

names = [m.name for m in Foo]
>>> print(*names)
bar baz

values = [m.value for m in Foo]
print(*values)
>>> qux 123

Вы можете перебирать невидимые атрибуты класса с помощью for attr in (elem for elem in dir(Foo) if elem[:2] != '__').

Менее ужасный способ написания:

def class_iter(Class):
    return (elem for elem in dir(Class) if elem[:2] != '__')

, затем

for attr in class_iter(Foo):
    pass

class MetaItetaror(type):
    def __iter__(cls):
        return iter(
            filter(
                lambda k: not k[0].startswith('__'),
                cls.__dict__.iteritems()
            )
        )


class Klass:
    __metaclass__ = MetaItetaror

    iterable_attr_names = {'x', 'y', 'z'}
    x = 5
    y = 6
    z = 7


for v in Klass:
    print v

это то, как мы делаем объект класса итерабельным. предоставить класс с помощью метода iter и next (), тогда вы можете перебирать атрибуты класса или их значения. Вы можете оставить метод next (), если хотите, или вы можете определить next () и повысить StopIteration при некотором условии .

например:

class Book(object):
      def __init__(self,title,author):
          self.title = title
          self.author = author

      def __iter__(self):
          for each in self.__dict__.keys():
              yield self.__getattribute__(each)

>>> book  = Book('The Mill on the Floss','George Eliot')
>>> for each in book: each
...
'George Eliot'
'The Mill on the Floss'

этот класс выполняет итерацию над значением атрибута класса Book. Объект класса можно сделать итерабельным, предоставив ему метод getitem. например:

class BenTen(object):
    def __init__(self, bentenlist):
        self.bentenlist = bentenlist

    def __getitem__(self,index):
        if index <5:
            return self.bentenlist[index]
        else:
            raise IndexError('this is high enough')

>>> bt_obj = BenTen([x for x in range(15)])
>>>for each in bt_obj:each
...
0
1
2
3
4

теперь, когда объект класса BenTen используется во внутреннем цикле, getitem вызывается с успешным более высоким значением индекса, пока он не вызовет IndexError.