27 lines
985 B
Python
27 lines
985 B
Python
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
|