SZI2019SmieciarzWmi/DataModels/House.py

13 lines
466 B
Python

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