st-ten-1/src/ui/barcode_recipe_selection/barcode_recipe_selection.py

import sys

import peewee
from PyQt5 import Qt

from lib.helpers import timing
from PyQt5.QtCore import Qt, QTimer, QThread
from PyQt5.QtGui import QKeySequence, QPixmap, QPalette, QColor
from PyQt5.QtWidgets import QShortcut, QApplication

from ui.test_test import Test_Test
from ui.widget import Widget
from lib import db
from lib.db.models import Recipes
test_scan="xxx\nyyy\nzzz"

class Barcode_Recipe_Selection(Test_Test):
    def __init__(self, parent):
        super().__init__()
        self.parent=parent
        self.components=parent.components
        self.recipe_db_model=Recipes
        self.debounce_time = 200
        self.barcode_input_l.textChanged.connect(self.debounce)
        self.status_palettes = {
            True: QPalette(),
            "": QPalette(),
            "warning": QPalette(),
            False: QPalette(),
            None: QPalette(),
        }
        self.status_palettes[True].setColor(QPalette.Base, Qt.green)
        self.status_palettes[False].setColor(QPalette.Base, Qt.red)
        self.status_palettes["warning"].setColor(QPalette.Base, QColor(255, 165, 0))
        self.status_palettes[""].setColor(QPalette.Base, QColor(255, 255, 0))

        self.recipe_selection_b.clicked.connect(self.parent.set_recipe_mode_table)
        self.barcode_input_l.setFocus()

        self.focus_timer = QTimer()
        self.focus_timer.setSingleShot(True)
        self.focus_timer.timeout.connect(self.set_focus)

        self.wait_timer = QTimer()
        self.wait_timer.setSingleShot(True)
        self.wait_timer.timeout.connect(self.get)

        self.ok_timer = QTimer()
        self.ok_timer.setSingleShot(True)
        self.ok_timer.timeout.connect(self.set_recipe)


    def start(self, recipe=None, step=None, pieces=None):

        self.barcode_input_l.setPalette(self.status_palettes[None])
        self.barcode_input_l.setPlainText("")
        self.barcode_input_l.setFocus()
        self.focus_timer.start(500)

    def debounce(self):
        if self.wait_timer.remainingTime()>0:
            self.wait_timer.setInterval(self.debounce_time)
        else:
            self.wait_timer.start(self.debounce_time)

    def get(self, data=None, override=False):
        if data is None:
            data = self.barcode_input_l.toPlainText()
        if not len(data):
            data = None
        if data is None:
            return
        else:
            lines = data.splitlines()
            candidates = [i for i in lines if len(i)==10]
            if len(candidates)>0:
                # RECIPE CODE FOUND
                self.recipe=candidates[-1]
                self.barcode_input_l.setPalette(self.status_palettes[True])

                if "--write-tags" not in sys.argv:
                    self.ok_timer.start(2000)
                else:
                    self.parent.components["rfid"].write_tag(data)

            else:
                # RECIPE CODE NOT FOUND
                self.barcode_input_l.setPalette(self.status_palettes[False])
                self.focus_timer.start(3000)


    def set_recipe(self):
        try:
            # LOOKUP RECIPE
            recipe = self.recipe_db_model.get_by_id(self.recipe)
            self.parent.set_recipe(recipe)
        except peewee.DoesNotExist:
            self.barcode_input_l.setPalette(self.status_palettes[False])
            self.focus_timer.start(3000)
            self.parent_assembly_widget().set_text("RICETTA NON TROVATA",bg_color="red")


    def set_focus(self):
        self.barcode_input_l.setFocus()
        self.barcode_input_l.setPlainText("")
        self.barcode_input_l.setPalette(self.status_palettes[""])
        if getattr(self.parent.centralWidget,"set_text"):
            if "fixture_id" in self.components.keys():
                self.parent_assembly_widget().set_text(
                    "INSERIRE LA DIMA PER SELEZIONARE AUTOMATICAMENTE LA RICETTA DI COLLAUDO")
            else:
                self.parent_assembly_widget().set_text("SCANSIONARE BARCODE SELEZIONE RICETTA")