SZI2019SmieciarzWmi/DataModels/House.py

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()