leonpan-pc/app/view/widgets/policy_messageBox.py

# coding: utf-8
import logging

from PyQt6.QtCore import Qt, QTimer
from PyQt6.QtWidgets import QListWidgetItem
from qfluentwidgets import (
    InfoBar,
    InfoBarPosition,
    ListWidget,
    MessageBoxBase,
    SubtitleLabel,
)

from app.core import ChangePolicyThread, GetPoliciesThread, policyConfig, signalBus


class PolicyChooseMessageBox(MessageBoxBase):
    def __init__(self, parent=None):
        super().__init__(parent=parent)
        self.currentPath = "/"
        self.policyDict = {}
        self.isLoading = True
        self.originalTitle = "选择存储策略"
        self.setClosableOnMaskClicked(True)

        self.setupUI()

        # 开始获取策略列表
        self.getPoliciesThread = GetPoliciesThread()
        self.getPoliciesThread.successGetSignal.connect(self.refreshPolicyList)
        self.getPoliciesThread.errorSignal.connect(self._handleGetPoliciesError)
        self.getPoliciesThread.start()

        # 初始化更改策略线程（但不启动）
        self.changePolicyThread = None

    def setupUI(self):
        """设置UI界面"""
        self.titleLabel = SubtitleLabel(self.originalTitle, self)
        self.policyListWidget = ListWidget(self)

        # 添加加载提示
        self.loadingItem = QListWidgetItem("正在加载策略列表...")
        self.policyListWidget.addItem(self.loadingItem)

        self.viewLayout.addWidget(self.titleLabel)
        self.viewLayout.addWidget(self.policyListWidget)

        # 隐藏确定取消按钮组
        self.buttonGroup.hide()

    def connectSignals(self):
        """连接信号与槽"""
        self.policyListWidget.currentTextChanged.connect(self.onPolicyChanged)
        self.policyListWidget.itemClicked.connect(self.selfClicked)

    def selfClicked(self, listWidget: QListWidgetItem):
        if listWidget.text() == policyConfig.returnPolicy()["name"]:
            QTimer.singleShot(100, self.accept)

    def onPolicyChanged(self, text):
        """处理策略更改"""
        if not text or self.isLoading or text == "正在加载策略列表...":
            return

        policy_id = self.policyDict.get(text)
        if not policy_id:
            return

        # 如果已经有更改线程在运行，先停止它
        if self.changePolicyThread and self.changePolicyThread.isRunning():
            self.changePolicyThread.quit()
            self.changePolicyThread.wait()

        # 创建并启动新的更改线程
        self.changePolicyThread = ChangePolicyThread(self.currentPath, policy_id)
        self.changePolicyThread.successChangedSignal.connect(
            self._handlePolicyChangeSuccess
        )
        self.changePolicyThread.errorSignal.connect(self._handlePolicyChangeError)
        self.changePolicyThread.start()

        # 更新UI状态 - 只更改标题
        self._setLoadingState(True, f"正在切换到策略: {text}")

    def refreshPolicyList(self, policiesList):
        """刷新策略列表"""
        self.isLoading = False
        self.policyListWidget.clear()
        self.policyDict.clear()

        currentPolicy = policyConfig.returnPolicy()
        currentIndex = 0

        for i, policy in enumerate(policiesList):
            self.policyListWidget.addItem(QListWidgetItem(policy["name"]))
            self.policyDict[policy["name"]] = policy["id"]

            if policy["id"] == currentPolicy["id"]:
                currentIndex = i

        # 设置当前选中项
        if self.policyListWidget.count() > 0:
            self.policyListWidget.setCurrentRow(currentIndex)

        self.currentPath = policyConfig.returnCurrentPath()
        self.connectSignals()

        # 恢复原始标题
        self.titleLabel.setText(self.originalTitle)

    def _handleGetPoliciesError(self, error_msg):
        """处理获取策略列表错误"""
        self.policyListWidget.clear()
        errorItem = QListWidgetItem(f"加载失败: {error_msg}")
        self.policyListWidget.addItem(errorItem)
        self.isLoading = False
        # 恢复原始标题
        self.titleLabel.setText(self.originalTitle)

    def _handlePolicyChangeSuccess(self):
        """处理策略更改成功"""
        self._setLoadingState(False)

        # 显示成功提示
        if self.parent():
            InfoBar.success(
                title="操作成功",
                content="存储策略已成功更改",
                orient=Qt.Orientation.Horizontal,
                isClosable=True,
                position=InfoBarPosition.TOP_RIGHT,
                duration=2000,
                parent=self.window(),
            )
        signalBus.refreshFolderListSignal.emit()
        QTimer.singleShot(1000, self.accept)

    def _handlePolicyChangeError(self, error_msg):
        """处理策略更改错误"""
        self._setLoadingState(False)

        # 显示错误提示
        if self.parent():
            InfoBar.error(
                title="操作失败",
                content=f"更改策略时出错: {error_msg}",
                orient=Qt.Orientation.Horizontal,
                isClosable=True,
                position=InfoBarPosition.TOP_RIGHT,
                duration=3000,
                parent=self.window(),
            )
            QTimer.singleShot(1000, self.reject)

    def _setLoadingState(self, loading, message=None):
        """设置加载状态"""
        if loading:
            self.policyListWidget.setEnabled(False)
            if message:
                logging.info(message)
        else:
            self.policyListWidget.setEnabled(True)
            # 恢复原始标题
            self.titleLabel.setText(self.originalTitle)