AL-2020/shelf.py
2020-06-10 00:03:58 +02:00

28 lines
532 B
Python

import pygame
class Shelf:
def __init__(self, field, color):
self.field = field
self.color = color
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_field(self):
return self.field
def get_color(self):
return self.color