37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
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]))
|