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 = (141, 35, 15)
        elif shelf_type == Package_types.priority:
            color = (33, 46, 100)
        return color