How to override iter() for custom class?

Question

What I'm trying to do is reuse generator

class Rect(object):
   ...
   def __iter__(self):
       for y in xrange(self.tl.y, self.br.y + 1):
           for x in xrange(self.tl.x, self.br.x + 1):
               yield Point(x, y)

in child class. I've tried write something like

class Block(Rect):
    ...
    def __iter__(self):
        for p in super(Block, self):
            yield p + self.offset

but obviously that didn't work.

My question is can that be done without copying code from parent class and what is most pythonic approach to re-usability of inherited code.

Perhaps it's a better idea to move Rect.__iter__'s implementation to a function (_iter_as_rect?) you can explicitly reuse. — user824425
– user824425, Commented Nov 25, 2016 at 10:46

Gennady Kandaurov · Accepted Answer · 2016-11-25 10:52:00Z

6

You can do it as follows:

class Block(Rect):
    def __iter__(self):
        for p in super(Block, self).__iter__():
            yield p + self.offset

answered Nov 25, 2016 at 10:52

Gennady Kandaurov

1,9741 gold badge15 silver badges21 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to override iter() for custom class?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related