AL-2020/assigning.py

import os

from codes_recognizer.rocognizer import recognizer


def assigning(product_code_path, shelfs):
    c = None
    code = product_code_path
    # code = recognizer(product_code_path)
    product_color = str(code[0]) + str(code[1])
    for shelf in shelfs:
        if shelf.color == product_color:
            shelf.add_product(code)
            c = shelf.get_field()


    print("Product:", product_code_path, "is on", c.x, c.y)
    return c