AL-2020/shelf.py
2020-06-09 21:57:37 +02:00

27 lines
531 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