SZI2019SmieciarzWmi/DataModels/Dump.py

15 lines
610 B
Python

import pygame
from DataModels.Cell import Cell
class Dump( Cell ):
def __init__( self, x, y, max_rubbish, dump_type, yellow = 0, green = 0, blue = 0 ):
Cell.__init__( self, x, y, max_rubbish, yellow, green, blue, dump_type )
self.dump_type = dump_type
self.unvisited = True
def return_trash(self, collector):
dump_type = self.dump_type.lower()[5:]
self.container.yellow, self.container.green, self.container.blue = collector.container.empty(dump_type,
[self.container.yellow, self.container.green, self.container.blue])
self.update_image()