AIprojekt-wozek/Types_colors.py

21 lines
628 B
Python

from Package_types import Package_types
class Types_colors(object):
@staticmethod
def get_package_color(package_type):
color = (100, 50, 20)
if package_type == Package_types.fragile:
color = (255, 57, 32)
elif package_type == Package_types.priority:
color = (10, 34, 255)
return color
@staticmethod
def get_shelf_color(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