from PyQt6.QtWidgets import QWidget, QVBoxLayout, QPushButton, QLabel
from PyQt6.QtCore import Qt, QTimer

from logger import logger


class Palletizing(QWidget):
    def __init__(self, load_palletizing):
        super().__init__()
        self.load_palletizing = load_palletizing
        self.initUI()

    def initUI(self):
        self.layout = QVBoxLayout()

        self.robotsLabel = QLabel("Паллетирование")
        self.layout.addWidget(self.robotsLabel)

        self.palletButton = QPushButton("Визуализировать")
        self.palletButton.clicked.connect(self.startPalletzing)
        self.layout.addWidget(self.palletButton)

        self.setLayout(self.layout)

    def startPalletzing(self):
        logger.info("start palletizing")
        self.load_palletizing()

    def paintEvent(self, event):
        p = self.palette()
        p.setColor(self.backgroundRole(), Qt.GlobalColor.lightGray)
        self.setPalette(p)
        super().paintEvent(event)