from PyQt6.QtCore import QThread, pyqtSignal
from typing import Callable, Any

class TaskRunner(QThread):
    # 定義一個(gè)信號(hào)，用于在任務(wù)完成后通知主線程
    finished_signal = pyqtSignal()

    def __init__(self, parent: Any | None, task: Callable):
        super().__init__()
        self.parent = parent
        self.task = task

    def run(self):
        """
        在新的線程中執(zhí)行耗時(shí)任務(wù)。
        """
        try:
            self.task(self.parent)
        finally:
            # 任務(wù)完成后發(fā)出信號(hào)
            self.finished_signal.emit()

from src.gui.loading import Ui_Form # 假設(shè)Ui_Form是你的UI定義
from PyQt6.QtWidgets import QWidget, QApplication
from PyQt6.QtCore import QTimer, QThread, pyqtSignal
from typing import Callable, Any

class Loading(Ui_Form, QWidget):
    def __init__(self,
                 parent: QWidget | None,
                 next_widget: QWidget | None,
                 action: str,
                 time: int,
                 task: Callable,
                 task_len: int,
                 initial_task: str):
        super().__init__()
        self.setupUi(self)
        self.setParent(parent)
        self.parent = parent
        self.next_widget = next_widget
        self.time = time

        # 直接實(shí)例化 TaskRunner (QThread)
        self.task_thread = TaskRunner(self, task)
        # 連接任務(wù)完成信號(hào)
        self.task_thread.finished_signal.connect(self.on_task_finished)

        self.current_time = 0
        self.tasks_done = 0
        self.all_tasks = task_len

        self.Task.setText(action)
        self.Estimation.setText(f"estimated time: {self.int_to_time(time)}")
        self.progressBar.setValue(0)
        self.TimeLeft.setText("")
        self.Current.setText("")
        self.Task.setText("")

        # 啟動(dòng)任務(wù)線程
        self.run_tasks()
        self.task_done(initial_task)

        self.timer = QTimer(self)
        self.timer.timeout.connect(self.update_time)
        self.timer.start(1000)

    @staticmethod
    def int_to_time(time: int) -> str:
        if time >= 3600:
            return f"{time / 3600} hours"
        elif time >= 60:
            return f"{time / 60} minutes"
        else:
            return f"{time} seconds"

    def update_time(self):
        self.current_time += 1
        self.TimeLeft.setText(self.int_to_time(self.current_time))

    def task_done(self, next_task: str = None):
        # 這里的tasks_done邏輯可能需要根據(jù)實(shí)際任務(wù)數(shù)量調(diào)整
        # 如果只有一個(gè)大任務(wù)，那么只在on_task_finished中更新一次即可
        self.tasks_done += 1
        if not next_task:
            self.Current.setText("finished all tasks, closing window")
            self.Tasks.setText(f"{self.tasks_done} out of {self.all_tasks}")
            # 任務(wù)完成后，可以更新進(jìn)度條到100%
            self.progressBar.setValue(100) 
        elif self.tasks_done != self.all_tasks:
            self.Current.setText(f"currently: {next_task}")
            self.Tasks.setText(f"{self.tasks_done} out of {self.all_tasks}")
            # 如果有多個(gè)子任務(wù)，這里可以根據(jù)tasks_done更新進(jìn)度條
            self.progressBar.setValue(int(self.tasks_done / self.all_tasks * 100))

    def on_task_finished(self):
        """
        當(dāng)后臺(tái)任務(wù)完成后，此槽函數(shù)會(huì)被調(diào)用。
        """
        self.task_done(None) # 標(biāo)記所有任務(wù)完成
        # 停止計(jì)時(shí)器
        self.timer.stop()
        # 確保線程正確退出
        self.task_thread.quit()
        self.task_thread.wait() # 阻塞直到線程完全退出
        self.close() # 關(guān)閉窗口

    def closeEvent(self, event):
        """
        重寫closeEvent以確保在窗口關(guān)閉時(shí)線程和計(jì)時(shí)器都被正確停止。
        """
        self.timer.stop()
        if self.task_thread.isRunning():
            self.task_thread.quit()
            self.task_thread.wait()
        super().closeEvent(event) # 調(diào)用父類的closeEvent

    def run_tasks(self):
        # 直接啟動(dòng) QThread
        self.task_thread.start()

import time
from unittest import TestCase
from PyQt6.QtWidgets import QApplication

# 假設(shè) Loading 類和 Ui_Form 都在可導(dǎo)入的路徑中
# from your_module import Loading, Ui_Form 

class TestLoading(TestCase):
    def test_task(self):
        def foo(loading_page: Loading):
            # 模擬耗時(shí)操作
            time.sleep(5)
            # 可以在這里更新進(jìn)度或狀態(tài)，通過信號(hào)發(fā)送回主線程
            # loading_page.update_progress_signal.emit(50) 
            time.sleep(5) # 模擬更多工作

        app = QApplication([])
        # 注意：這里的 task_len 應(yīng)該與 foo 函數(shù)中模擬的實(shí)際任務(wù)階段數(shù)匹配，
        # 如果 foo 只是一個(gè)整體任務(wù)，那么 task_len 可以設(shè)為1，或者根據(jù)你的需求調(diào)整
        self.loader = Loading(None, None, "doing something", 10, foo, 1, "testing") 
        self.loader.show()
        app.exec()