SZI2019SmieciarzWmi/sprites/house.py
2019-03-19 18:10:04 +01:00

36 lines
1.2 KiB
Python

import pygame
import sys, 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 = [0,0,0] #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_blue.png")
print("PLASTIC alert")
if( self.rubbish[GLASS] > self.max_glass ):
self.image = pygame.image.load("images/house_green.png")
print("GLASS alert")
if( self.rubbish[METAL] > self.max_metal ):
self.image = pygame.image.load("images/house_yellow.png")
print("METAL alert")
def check_rubbish_status(self):
print( "plastic: " + str(self.rubbish[PLASTIC]) + " glass: " + str(self.rubbish[GLASS]) + " metal: " + str(self.rubbish[METAL]) )