import pygame import sys import random from sprites.cell import Cell PLASTIC = 0 # blue GLASS = 1 # green METAL = 2 # yellow class House(Cell): def __init__(self, x, y, max_plastic, max_glass, max_metal): Cell.__init__(self, x, y) self.image = pygame.image.load("images/house.png") self.rubbish = [random.randint(0, max_plastic), random.randint( 0, max_glass), random.randint(0, max_metal)] # plastic, glass, metal self.max_plastic = max_plastic self.max_glass = max_glass self.max_metal = max_metal def generate_rubbish(self): if(random.randint(0, 5) == 1): # 1/5 szansa na wyrzucenie śmiecia w klatce thrash_type = random.randint(0, 2) self.rubbish[thrash_type] = self.rubbish[thrash_type] + 1 if(self.rubbish[PLASTIC] > self.max_plastic): self.image = pygame.image.load("images/house_plastic.png") if(self.rubbish[GLASS] > self.max_glass): self.image = pygame.image.load("images/house_glass.png") if(self.rubbish[METAL] > self.max_metal): self.image = pygame.image.load("images/house_metal.png") def check_rubbish_status(self): print("plastic: " + str(self.rubbish[PLASTIC]) + " glass: " + str( self.rubbish[GLASS]) + " metal: " + str(self.rubbish[METAL]))