from DataModels.Cell import Cell class House(Cell): def __init__(self, x, y, max_rubbish, yellow=0, green=0, blue=0): Cell.__init__(self, x, y, max_rubbish, yellow, green, blue) self.unvisited = True def return_trash(self, collector): self.container.yellow, self.container.green, self.container.blue = collector.container.add( [self.container.yellow, self.container.green, self.container.blue]) self.update_image()