From 0ed316f044f9b8812d9694480abb10219513a5b2 Mon Sep 17 00:00:00 2001
From: shaaqu <pawel.lukaszewicz99@gmail.com>
Date: Tue, 9 Jun 2020 18:22:14 +0200
Subject: [PATCH] agent is getting path to the shelf

---
 main.py    | 5 +++++
 product.py | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/main.py b/main.py
index 840a782..79cf339 100644
--- a/main.py
+++ b/main.py
@@ -70,8 +70,13 @@ def run():
             new_product = FinalProduct(supply_depot.item[0], supply_depot.item[1], supply_depot.item[2],
                                        supply_depot.item[3], prediction[0])
             print(new_product)
+            # list [x, y]
+            dest_shelf = new_product.shelf()
+            dest_field = board[dest_shelf[0], dest_shelf[1]]
+            path = functions.a_star(board[agent.y][agent.x], dest_field, board)
             agent.item = new_product
 
+
         if next_step is not None:
             time.sleep(0.5)
             if functions.check_turn(agent, next_step):
diff --git a/product.py b/product.py
index 0f28fa0..e16ea9c 100644
--- a/product.py
+++ b/product.py
@@ -1,4 +1,5 @@
 from coder import create_image
+from assigning import assigning
 
 
 class Product:
@@ -29,3 +30,6 @@ class FinalProduct:
             self.price = 4.99
 
         self.img = create_image(self)
+
+    def shelf(self):
+        return assigning(self.img);