From 41729163f1ce6d7f7986e91d3f8856b17677be7e Mon Sep 17 00:00:00 2001
From: Jeremi Lisek <jerlis2@st.amu.edu.pl>
Date: Fri, 25 Mar 2022 09:48:42 +0100
Subject: [PATCH] :)))

---
 data/CATEGORY.py |  3 ++-
 data/Game.py     | 18 +++++++++++++++++-
 data/Item.py     |  5 +++--
 3 files changed, 22 insertions(+), 4 deletions(-)

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