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)

    def return_trash(self, collector):
        self.container.y, self.container.g, self.container.b = collector.container.add(
            [self.container.y, self.container.g, self.container.b])
        print(collector.container.status())
        self.update_image()