SZI2019SmieciarzWmi/DataModels/Dump.py

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

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