Add download progress dialog and some mod loader parse function.

core_lib/qt/progress_dialog.py

@@ -0,0 +1,168 @@
+from PySide6.QtCore import Qt, Slot, QTimer
+from PySide6.QtWidgets import (
+    QDialog,
+    QLabel,
+    QProgressBar,
+    QPushButton,
+    QScrollArea,
+    QSizePolicy,
+    QVBoxLayout,
+    QWidget,
+)
+
+
+class ProgressRow(QWidget):
+    def __init__(self, name: str, parent=None):
+        super().__init__(parent)
+
+        self.setSizePolicy(
+            QSizePolicy.Policy.Expanding,
+            QSizePolicy.Policy.Fixed,
+        )
+
+        self.name_label = QLabel(name, self)
+        self.name_label.setWordWrap(True)
+        self.name_label.setTextInteractionFlags(
+            Qt.TextInteractionFlag.TextSelectableByMouse
+        )
+
+        self.status_label = QLabel("Preparing...", self)
+        self.status_label.setWordWrap(True)
+        self.status_label.setTextInteractionFlags(
+            Qt.TextInteractionFlag.TextSelectableByMouse
+        )
+
+        self.progress_bar = QProgressBar(self)
+        self.progress_bar.setRange(0, 100)
+        self.progress_bar.setValue(0)
+
+        # Flags
+        self.finished = False
+
+        layout = QVBoxLayout(self)
+        layout.setContentsMargins(8, 8, 8, 8)
+        layout.setSpacing(4)
+        layout.addWidget(self.name_label)
+        layout.addWidget(self.status_label)
+        layout.addWidget(self.progress_bar)
+
+
+class ProgressDialog(QDialog):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+
+        self.setWindowTitle("Downloads")
+        self.resize(500, 300)
+        self.setMinimumSize(360, 220)
+
+        self.rows: dict[str, ProgressRow] = {}
+
+        self.scroll_area = QScrollArea(self)
+        self.download_widget = QWidget(self.scroll_area)
+        self.download_layout = QVBoxLayout(self.download_widget)
+        self.download_layout.setContentsMargins(4, 4, 4, 4)
+        self.download_layout.setSpacing(8)
+        self.download_layout.setAlignment(Qt.AlignmentFlag.AlignTop)
+
+        self.scroll_area.setWidget(self.download_widget)
+        self.scroll_area.setWidgetResizable(True)
+        self.scroll_area.setHorizontalScrollBarPolicy(
+            Qt.ScrollBarPolicy.ScrollBarAlwaysOff
+        )
+
+        self.close_button = QPushButton("Close", self)
+        self.close_button.clicked.connect(self.hide)
+
+        layout = QVBoxLayout(self)
+        layout.setContentsMargins(8, 8, 8, 8)
+        layout.setSpacing(8)
+        layout.addWidget(self.scroll_area, 1)
+        layout.addWidget(
+            self.close_button,
+            0,
+            Qt.AlignmentFlag.AlignRight,
+        )
+
+        self.close_timer = QTimer(self)
+        self.close_timer.setSingleShot(True)
+        self.close_timer.timeout.connect(self.cleanup)
+
+    @Slot(str, str)
+    def add_task(self, task_id: str, name: str):
+        if task_id in self.rows:
+            return
+
+        row = ProgressRow(name, self.download_widget)
+        self.rows[task_id] = row
+
+        self.download_layout.addWidget(row)
+
+        if not self.isVisible():
+            self.show()
+
+    @Slot(str, str, float)
+    def update_task(
+        self,
+        task_id: str,
+        status: str,
+        percent: float,
+    ):
+        row = self.rows.get(task_id)
+        if row is None:
+            # Prevent started、progress missing due to timing issues
+            self.add_task(task_id, task_id)
+            row = self.rows[task_id]
+
+        value = max(0, min(100, round(percent)))
+        row.status_label.setText(status)
+        row.progress_bar.setValue(value)
+
+    @Slot(str)
+    def finish_task(self, task_id: str):
+        row = self.rows.get(task_id)
+        if row is None:
+            return
+
+        row.status_label.setText("Completed")
+        row.progress_bar.setValue(100)
+        row.finished = True
+
+        if self.all_tasks_finished():
+            self.close_timer.start(2000)
+
+    def all_tasks_finished(self):
+        return all(row.finished == True for task_id, row in self.rows.items())
+
+    def cleanup(self):
+        """
+        If you want to use this function with a timer. use self.close_timer.start(TIMEOUT)
+        :return:
+        """
+        QTimer.singleShot(2000, self.cleanup)
+        for task_id in list(self.rows):
+            self.remove_task(task_id)
+
+        self.hide()
+
+    @Slot(str, str)
+    def fail_task(self, task_id: str, error: str):
+        row = self.rows.get(task_id)
+        if row is None:
+            self.add_task(task_id, task_id)
+            row = self.rows[task_id]
+
+        row.status_label.setText(f"Failed: {error}")
+        row.progress_bar.setStyleSheet(
+            "QProgressBar::chunk { background-color: #c62828; }"
+        )
+        # Move to top
+        self.download_layout.removeWidget(row)
+        self.download_layout.insertWidget(0, row)
+
+    @Slot(str)
+    def remove_task(self, task_id: str):
+        row = self.rows.pop(task_id, None)
+        if row is not None:
+            self.download_layout.removeWidget(row)
+            row.setParent(None)
+            row.deleteLater()