from PySide6.QtCore import Qt, Signal, Slot
from PySide6.QtWidgets import QApplication, QWidget

from project import Project

from .blockLabelDialog import BlockLabelDialog
from .projectEntry_Manage_ui import Ui_ProjectEntry_Manage
from .yieldProgress import YieldProgress


class ProjectEntry_Manage(Ui_ProjectEntry_Manage, QWidget):
    reloadProject = Signal()

    def __init__(self, parent=None):
        super().__init__(parent)
        self.setupUi(self)
        self.project = None
        self.abort = False

    def setProject(self, project: Project):
        self.project = project
        self.updateLabels()

    def updateLabels(self):
        if not self.project:
            self.projectNameLabel.setText("-")
            self.projectDescriptionLabel.setText("-")
            return

        with BlockLabelDialog(self) as block:
            block.setText(f"{self.project.name}<br>Updating status")
            block.show()

            QApplication.processEvents()
            self.projectNameLabel.setText(self.project.name)
            self.projectDescriptionLabel.setText(
                "<br>".join(
                    [
                        str(self.project.path.resolve()),
                        f"{len(self.project.sources)} sources",
                        f"{len(self.project.samples)} samples",
                        f"- {len(self.project.samplesClassified)} classified",
                        f"- {len(self.project.samplesIgnored)} ignored",
                        f"- {len(self.project.samplesUnclassified)} unclassified",
                    ]
                )
            )

    @Slot()
    def on_updateButton_clicked(self):
        self.updateLabels()

    def setAbort(self, b: bool):
        self.abort = b

    @Slot()
    def on_extractButton_clicked(self):
        if not self.project:
            return

        progressDialog = YieldProgress(self)
        progressDialog.setWindowModality(Qt.WindowModality.ApplicationModal)

        self.abort = False
        progressDialog.abortButton.clicked.connect(lambda: self.setAbort(True))

        progressDialog.show()

        iterator = iter(self.project.extractSamplesYield())
        progressDialog.progressBar.setMinimum(0)
        while not self.abort:
            try:
                path, i, total = next(iterator)
                progressDialog.label.setText(path.name)
                progressDialog.progressBar.setValue(i)
                if i <= 2:
                    progressDialog.progressBar.setMaximum(total)

                if total > 10 and i % 10 == 0 or total <= 10:
                    progressDialog.update()
                    QApplication.processEvents()
            except StopIteration:
                break

        self.abort = False
        self.reloadProject.emit()
        progressDialog.close()
        progressDialog.deleteLater()

    @Slot()
    def on_redactSourcesButton_clicked(self):
        if not self.project:
            return

        progressDialog = YieldProgress(self)
        progressDialog.setWindowModality(Qt.WindowModality.ApplicationModal)

        self.abort = False
        progressDialog.abortButton.clicked.connect(lambda: self.setAbort(True))

        progressDialog.show()

        iterator = iter(self.project.redactSourcesYield())
        progressDialog.progressBar.setMinimum(0)
        while not self.abort:
            try:
                path, i, total = next(iterator)
                progressDialog.label.setText(path.name)
                progressDialog.progressBar.setValue(i)
                if i <= 2:
                    progressDialog.progressBar.setMaximum(total)

                if total > 10 and i % 10 == 0 or total <= 10:
                    progressDialog.update()
                    QApplication.processEvents()
            except StopIteration:
                break

        self.abort = False
        progressDialog.close()
        progressDialog.deleteLater()

    @Slot()
    def on_trainButton_clicked(self):
        if not self.project:
            return

        with BlockLabelDialog(self) as block:
            block.setText(f"{self.project.name}<br>Training")
            block.show()

            self.project.train()