propusk/PropuskWidgets/PCamChecked.py

from PySide6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QLineEdit, QComboBox, QPushButton
from PropuskWidgets.PStackedWidget import PStackedWidget
from module.cam import IPCam, USBCam, load_image, get_list_name_cam
import os

class PCamChecked(QWidget):
    def __init__(self, parent=None, mode: str = 'video') -> None:
        super().__init__(parent)
        
        self.__mode = mode
        self.vLayout = QVBoxLayout(self)
        self.vLayout.setObjectName(u'vLayout')
        
        match self.__mode:
            case 'video': 
                self.line_cam = QComboBox(self)
                #load list cam
                [self.line_cam.addItem(x) for x in get_list_name_cam()]
            case 'snapshot': self.line_cam = QLineEdit(self)

        self.line_cam.setObjectName(u"line_cam")
        self.vLayout.addWidget(self.line_cam)

        self.pStackedWidget = PStackedWidget(self, mode=mode)
        self.pStackedWidget.setObjectName(u'pStackedWidget')
        self.vLayout.addWidget(self.pStackedWidget)

        self.hLayout = QHBoxLayout()
        self.hLayout.setObjectName(u'hLayout')

        self.btn_check = QPushButton(self)
        self.btn_check.setObjectName('btn_check')
        self.btn_check.setText('Проверить')
        self.btn_check.clicked.connect(self._start_cam)
        self.hLayout.addWidget(self.btn_check)

        self.btn_stop = QPushButton(self)
        self.btn_stop.setObjectName('btn_stop')
        self.btn_stop.setText('Остановить')
        self.btn_stop.clicked.connect(self._stop_cam)
        self.hLayout.addWidget(self.btn_stop)
        
        self.vLayout.addLayout(self.hLayout)


    def get_value(self) -> str:
        match self.__mode:
            case 'video':
                return self.line_cam.currentText()
            case 'snapshot':
                return self.line_cam.text()
            case _: return None

    def _start_cam(self) -> None:
        self.pStackedWidget.to_video()
        match self.__mode:
            case 'video':
                self.__wwc = USBCam(self.pStackedWidget.video, self.line_cam.currentText())
                self.__wwc.start_cam()
            case 'snapshot':
                self.__wwc = IPCam()
                self.__wwc.qLabel = self.pStackedWidget.video
                self.__wwc.lnk_connect = self.line_cam.text()
                self.__wwc.start()
                print("Started...")
            case _: return None


    def _stop_cam(self):
        self.__wwc.stop_cam()  
        del self.__wwc
        self.pStackedWidget.to_image()