AIprojekt-wozek/Shelf.py

import pygame
from Field import Field
from Global_variables import Global_variables as G_var
from Package_types import Package_types


class Shelf(Field):
    def __init__(self, window, x, y, type):
        Field.__init__(self, window, x, y)
        self.type = type
        self.color = self.get_shelf_color(self.type)
        self.block = pygame.Rect(self.x * G_var().RECT_SIZE, self.y *
                                 G_var().RECT_SIZE, G_var().RECT_SIZE, G_var().RECT_SIZE)

    def draw(self):
        self.block = pygame.Rect(self.x * G_var().RECT_SIZE, self.y *
                                 G_var().RECT_SIZE, G_var().RECT_SIZE, G_var().RECT_SIZE)
        pygame.draw.rect(self.window, self.color, self.block)

    def get_shelf_color(self, shelf_type):
        color = (143, 68, 33)
        if shelf_type == Package_types.fragile:
            color = (191, 35, 15)
        elif shelf_type == Package_types.priority:
            color = (33, 46, 140)
        return color