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);