AL-2020/shelf.py

import pygame


class Shelf:

    def __init__(self, x, y):
        self.x = x
        self.y = y
        self.level1 = []
        self.level2 = []
        self.level3 = []


    def add_product(self, product):
        if product[7] == 1:
            self.level1.append(product)
        elif product[7] == 2:
            self.level2.append(product)
        else:
            self.level3.append(product)

    def get_product(self):
        return self.products[0]

    def get_coordinates(self):
        c = [self.x, self.y]
        return c

    def is_empty(self):
        if len(self.products) == 0:
            return True
        else:
            return False