make class iterable

SLOtoSF
107 ●7 ●13 ●19

How do I make a class iterable?

I want to create a class, which is a set of objects that I created, that has attributes of its own.

The user experience I am thinking of is the following:

(let's say A, B, and C are Objects in memory)

sage: D = Box((A,B,C))

I want to be able to say for i in D: ....

But I can't.

Any tips?

2 Answers

Sort by » oldest newest most voted

Volker Braun
3301 ●11 ●30 ●75

Your class must implement an __iter__ method:

class Foo(SageObject):

    def __iter__(self):
        for i in range(10):
            yield(i)

Then you can just say:

sage: bar = Foo()
sage: list(bar)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

SLOtoSF
107 ●7 ●13 ●19

The answer to this question is the derive the class from a set, but I think Volker's answer is to the point.

Like so:

class MyClass(set):

Please start posting anonymously - your entry will be published after you log in or create a new account.

Asked: 2012-10-03 03:14:12 +0100

Seen: 830 times

Last updated: Oct 03 '12