diff --git a/data/CATEGORY.py b/data/CATEGORY.py index f7afe7b..cf6357d 100644 --- a/data/CATEGORY.py +++ b/data/CATEGORY.py @@ -4,4 +4,5 @@ from enum import Enum class CATEGORY(Enum): PIZZA = 1 PASTA = 2 - EGG = 3 \ No newline at end of file + EGG = 3 + UNKNOWN = 4 \ No newline at end of file diff --git a/data/Game.py b/data/Game.py index 1434394..5171059 100644 --- a/data/Game.py +++ b/data/Game.py @@ -1,3 +1,4 @@ +from data.CATEGORY import CATEGORY from data.Item import Item from data.Order import Order @@ -13,4 +14,19 @@ class Game: self.deliveryItem = deliveryItem self.agentHandId = agentHandId self.orderStock = orderStock - self.orderList = orderList \ No newline at end of file + self.orderList = orderList + + def move(self, x: int, y:int): + self.agentPos = (x, y) + + def pickUp(self, item: Item): + self.deliveryItem = item + + def drop(self, item: Item): + self.deliveryItem = -1 + + def identify(item: Item, category: CATEGORY): + item.category = category + + def finishOrder(order: Order): + order.id = -1 \ No newline at end of file diff --git a/data/Item.py b/data/Item.py index 9ba86eb..958f577 100644 --- a/data/Item.py +++ b/data/Item.py @@ -2,6 +2,7 @@ from data.CATEGORY import CATEGORY class Item: - def __init__(self, id: int, category: CATEGORY): + def __init__(self, id: int, category: CATEGORY, price: int): self.id = id - self.category = category \ No newline at end of file + self.category = category + self.price = price \ No newline at end of file