shuidrop_gui/update_dialog.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
现代化圆形进度条更新对话框
展示完整更新流程：下载 → UAC授权 → 安装 → 重启
"""

from PyQt5.QtWidgets import (QDialog, QVBoxLayout, QLabel, 
                             QPushButton, QHBoxLayout, QWidget)
from PyQt5.QtCore import Qt, QTimer, QPropertyAnimation, QEasingCurve, pyqtProperty
from PyQt5.QtGui import QFont, QPainter, QColor, QPen, QLinearGradient, QConicalGradient
from PyQt5.QtCore import QRectF, QPointF
import math


class CircularProgressBar(QWidget):
    """
    自定义圆形进度条组件
    
    特性：
    - 顺时针旋转动画
    - 渐变色圆环
    - 中心显示百分比
    - 支持平滑过渡
    """
    
    def __init__(self, parent=None):
        super().__init__(parent)
        self._value = 0  # 当前进度值（0-100）
        self._max_value = 100
        self._text = "0%"
        self.setMinimumSize(200, 200)
        
        # 动画效果
        self.animation = QPropertyAnimation(self, b"value")
        self.animation.setDuration(500)  # 500ms过渡动画
        self.animation.setEasingCurve(QEasingCurve.OutCubic)
    
    @pyqtProperty(int)
    def value(self):
        return self._value
    
    @value.setter
    def value(self, val):
        self._value = val
        self.update()  # 触发重绘
    
    def setValue(self, val):
        """设置进度值（带动画）"""
        if val != self._value:
            self.animation.stop()
            self.animation.setStartValue(self._value)
            self.animation.setEndValue(val)
            self.animation.start()
    
    def setText(self, text):
        """设置中心文本"""
        self._text = text
        self.update()
    
    def paintEvent(self, event):
        """绘制圆形进度条"""
        painter = QPainter(self)
        painter.setRenderHint(QPainter.Antialiasing)  # 抗锯齿
        
        # 计算中心和半径
        width = self.width()
        height = self.height()
        size = min(width, height)
        center = QPointF(width / 2, height / 2)
        radius = size / 2 - 15  # 留出边距
        
        # 绘制背景圆环（灰色）
        pen = QPen()
        pen.setWidth(12)
        pen.setColor(QColor(230, 230, 230))
        pen.setCapStyle(Qt.RoundCap)
        painter.setPen(pen)
        
        rect = QRectF(center.x() - radius, center.y() - radius, 
                      radius * 2, radius * 2)
        painter.drawArc(rect, 90 * 16, -360 * 16)  # 绘制完整圆环
        
        # 绘制进度圆环（渐变色）
        if self._value > 0:
            # 创建圆锥渐变（顺时针旋转效果）
            gradient = QConicalGradient(center, 90)  # 从顶部开始
            gradient.setColorAt(0.0, QColor(66, 133, 244))    # 蓝色
            gradient.setColorAt(0.5, QColor(25, 118, 210))    # 深蓝
            gradient.setColorAt(1.0, QColor(21, 101, 192))    # 更深蓝
            
            pen.setBrush(gradient)
            pen.setColor(QColor(66, 133, 244))
            painter.setPen(pen)
            
            # 计算进度角度（顺时针，从顶部开始）
            span_angle = int(-(self._value / self._max_value) * 360 * 16)
            painter.drawArc(rect, 90 * 16, span_angle)
        
        # 绘制中心文本（百分比）
        painter.setPen(QColor(44, 62, 80))
        font = QFont('Microsoft YaHei', 28, QFont.Bold)
        painter.setFont(font)
        painter.drawText(rect, Qt.AlignCenter, self._text)


class UpdateProgressDialog(QDialog):
    """
    现代化更新进度对话框
    
    更新流程阶段：
    1. 下载安装包 (0-50%)
    2. 等待UAC授权 (50-60%)
    3. 执行安装 (60-90%)
    4. 等待程序重启 (90-100%)
    """
    
    # 流程阶段定义
    STAGE_DOWNLOAD = 0      # 下载阶段 (0-50%)
    STAGE_UAC = 1           # UAC授权阶段 (50-60%)
    STAGE_INSTALL = 2       # 安装阶段 (60-90%)
    STAGE_RESTART = 3       # 重启阶段 (90-100%)
    
    def __init__(self, version, parent=None):
        super().__init__(parent)
        self.version = version
        self.downloader = None
        self.current_stage = self.STAGE_DOWNLOAD
        self.download_progress = 0  # 下载进度（0-100）
        self.restart_progress = 0  # 重启阶段进度（0-100），用于外部检查
        
        self.init_ui()
    
    def init_ui(self):
        """初始化UI"""
        self.setWindowTitle("正在更新")
        self.setFixedSize(500, 600)
        # 禁止关闭按钮（只能通过取消按钮关闭）
        self.setWindowFlags(Qt.Dialog | Qt.WindowTitleHint | Qt.CustomizeWindowHint)
        
        # 主布局
        layout = QVBoxLayout()
        layout.setSpacing(20)
        layout.setContentsMargins(30, 30, 30, 30)
        
        # ============ 标题区域 ============
        self.title_label = QLabel(f"正在更新到 v{self.version}")
        self.title_label.setFont(QFont('Microsoft YaHei', 16, QFont.Bold))
        self.title_label.setAlignment(Qt.AlignCenter)
        self.title_label.setStyleSheet("color: #2c3e50; margin-bottom: 10px;")
        layout.addWidget(self.title_label)
        
        # ============ 圆形进度条 ============
        self.circular_progress = CircularProgressBar()
        layout.addWidget(self.circular_progress, alignment=Qt.AlignCenter)
        
        # ============ 阶段提示 ============
        self.stage_label = QLabel("📥 正在下载安装包...")
        self.stage_label.setFont(QFont('Microsoft YaHei', 12))
        self.stage_label.setAlignment(Qt.AlignCenter)
        self.stage_label.setStyleSheet("color: #34495e; margin-top: 10px;")
        layout.addWidget(self.stage_label)
        
        # ============ 详细状态 ============
        self.status_label = QLabel("准备开始下载...")
        self.status_label.setAlignment(Qt.AlignCenter)
        self.status_label.setStyleSheet("color: #7f8c8d; font-size: 11px;")
        self.status_label.setMinimumHeight(25)
        layout.addWidget(self.status_label)
        
        # ============ 重试信息（默认隐藏）============
        self.retry_label = QLabel("")
        self.retry_label.setAlignment(Qt.AlignCenter)
        self.retry_label.setStyleSheet("color: #ff9800; font-size: 11px; font-weight: bold;")
        self.retry_label.setMinimumHeight(25)
        self.retry_label.hide()
        layout.addWidget(self.retry_label)
        
        # ============ 按钮区域 ============
        button_layout = QHBoxLayout()
        button_layout.addStretch()
        
        self.cancel_button = QPushButton("取消更新")
        self.cancel_button.setFixedWidth(120)
        self.cancel_button.setFixedHeight(36)
        self.cancel_button.clicked.connect(self.cancel_download)
        button_layout.addWidget(self.cancel_button)
        
        button_layout.addStretch()
        layout.addLayout(button_layout)
        
        self.setLayout(layout)
        
        # 应用样式
        self.apply_styles()
    
    def apply_styles(self):
        """应用现代化样式"""
        self.setStyleSheet("""
            QDialog {
                background: qlineargradient(
                    x1:0, y1:0, x2:0, y2:1, 
                    stop:0 #f8fafb, 
                    stop:0.5 #f1f4f6,
                    stop:1 #e8ecef
                );
            }
            QLabel {
                background: transparent;
            }
            QPushButton {
                background: qlineargradient(
                    x1:0, y1:0, x2:0, y2:1, 
                    stop:0 #f5f5f5, 
                    stop:1 #e0e0e0
                );
                border: 2px solid #ccc;
                border-radius: 8px;
                padding: 8px 20px;
                font-size: 12px;
                font-weight: bold;
                font-family: 'Microsoft YaHei';
                color: #555;
            }
            QPushButton:hover {
                background: qlineargradient(
                    x1:0, y1:0, x2:0, y2:1, 
                    stop:0 #e8e8e8, 
                    stop:1 #d0d0d0
                );
                border: 2px solid #999;
            }
            QPushButton:pressed {
                background: #d0d0d0;
            }
            QPushButton:disabled {
                background: #f0f0f0;
                color: #aaa;
                border: 2px solid #ddd;
            }
        """)
    
    def update_progress(self, downloaded, total):
        """
        更新下载进度（阶段1：0-50%）
        
        Args:
            downloaded: 已下载字节数
            total: 总字节数
        """
        if self.current_stage != self.STAGE_DOWNLOAD:
            return  # 不在下载阶段，忽略
        
        if total > 0:
            # 计算下载百分比（0-100）
            download_percent = int((downloaded / total) * 100)
            self.download_progress = download_percent
            
            # 映射到总进度的0-50%
            overall_progress = int(download_percent * 0.5)
            
            # 更新圆形进度条
            self.circular_progress.setValue(overall_progress)
            self.circular_progress.setText(f"{overall_progress}%")
            
            # 转换为MB显示
            downloaded_mb = downloaded / (1024 * 1024)
            total_mb = total / (1024 * 1024)
            
            # 更新状态文本
            self.status_label.setText(f"已下载 {downloaded_mb:.1f} MB / {total_mb:.1f} MB")
        else:
            # 无法获取总大小时显示不确定状态
            self.status_label.setText("正在下载，请稍候...")
    
    def show_retry_info(self, current_retry, max_retries):
        """
        显示重试信息
        
        Args:
            current_retry: 当前重试次数
            max_retries: 最大重试次数
        """
        self.retry_label.setText(f"⚠️ 下载失败，正在重试... ({current_retry}/{max_retries})")
        self.retry_label.show()
        
        # 重置下载进度
        self.download_progress = 0
        self.circular_progress.setValue(0)
        self.circular_progress.setText("0%")
        self.status_label.setText("正在重试连接...")
        
        # 更新标题
        self.title_label.setText(f"正在重试下载 v{self.version}")
    
    def download_finished(self, file_path):
        """
        下载完成，进入UAC授权阶段（阶段2：50-60%）
        
        Args:
            file_path: 下载的文件路径
        """
        self.current_stage = self.STAGE_UAC
        
        # 更新进度到50%
        self.circular_progress.setValue(50)
        self.circular_progress.setText("50%")
        
        # 更新阶段提示
        self.stage_label.setText("🔐 等待管理员授权...")
        self.title_label.setText(f"✅ 下载完成")
        self.status_label.setText("请在UAC窗口点击「允许」以继续...")
        self.status_label.setStyleSheet("color: #28a745; font-size: 11px; font-weight: bold;")
        
        # 禁用取消按钮
        self.cancel_button.setEnabled(False)
        self.retry_label.hide()
        
        # 启动UAC等待动画（50% → 60%）
        self.start_uac_waiting_animation()
    
    def start_uac_waiting_animation(self):
        """启动UAC等待动画（进度条在50-60%之间缓慢移动）"""
        self.uac_timer = QTimer(self)
        self.uac_progress = 50
        
        def update_uac_progress():
            self.uac_progress += 0.5  # 每次增加0.5%
            if self.uac_progress <= 60:
                self.circular_progress.setValue(int(self.uac_progress))
                self.circular_progress.setText(f"{int(self.uac_progress)}%")
            else:
                # 超过60%停止（等待用户授权）
                self.uac_timer.stop()
        
        self.uac_timer.timeout.connect(update_uac_progress)
        self.uac_timer.start(500)  # 每500ms更新一次
    
    def uac_authorized(self):
        """
        用户已授权UAC，进入安装阶段（阶段3：60-90%）
        """
        # 停止UAC等待动画
        if hasattr(self, 'uac_timer'):
            self.uac_timer.stop()
        
        self.current_stage = self.STAGE_INSTALL
        
        # 更新进度到60%
        self.circular_progress.setValue(60)
        self.circular_progress.setText("60%")
        
        # 更新阶段提示
        self.stage_label.setText("⚙️ 正在安装新版本...")
        self.title_label.setText("正在安装")
        self.status_label.setText("正在替换程序文件，请稍候...")
        self.status_label.setStyleSheet("color: #007bff; font-size: 11px; font-weight: bold;")
        
        # 启动安装进度动画（60% → 90%）
        self.start_install_animation()
    
    def start_install_animation(self):
        """启动安装进度动画（60% → 90%）"""
        self.install_timer = QTimer(self)
        self.install_progress = 60
        
        def update_install_progress():
            self.install_progress += 1  # 每次增加1%
            if self.install_progress <= 90:
                self.circular_progress.setValue(self.install_progress)
                self.circular_progress.setText(f"{self.install_progress}%")
            else:
                # 安装完成，进入重启阶段
                self.install_timer.stop()
                self.prepare_restart()
        
        self.install_timer.timeout.connect(update_install_progress)
        self.install_timer.start(800)  # 每800ms增加1%（共24秒）
    
    def prepare_restart(self):
        """
        准备重启（阶段4：90-100%）
        """
        self.current_stage = self.STAGE_RESTART
        
        # 更新进度到90%
        self.circular_progress.setValue(90)
        self.circular_progress.setText("90%")
        
        # 更新阶段提示
        self.stage_label.setText("🚀 准备启动新版本...")
        self.title_label.setText("即将完成")
        self.status_label.setText("程序即将重启，请稍候...")
        self.status_label.setStyleSheet("color: #28a745; font-size: 11px; font-weight: bold;")
        
        # 启动重启倒计时动画（90% → 100%）
        self.start_restart_animation()
    
    def start_restart_animation(self):
        """启动重启倒计时动画（90% → 100%）"""
        self.restart_timer = QTimer(self)
        self.restart_progress = 90
        
        def update_restart_progress():
            self.restart_progress += 2  # 每次增加2%
            if self.restart_progress <= 100:
                self.circular_progress.setValue(self.restart_progress)
                self.circular_progress.setText(f"{self.restart_progress}%")
                
                if self.restart_progress == 100:
                    # 🔥 到达100%，更新提示但不自动关闭
                    self.status_label.setText("✅ 安装完成！等待启动确认...")
                    self.stage_label.setText("🎉 更新成功！")
                    self.restart_timer.stop()  # 停止定时器
            else:
                self.restart_timer.stop()
        
        self.restart_timer.timeout.connect(update_restart_progress)
        self.restart_timer.start(500)  # 每500ms增加2%（共2.5秒）
    
    def download_error(self, error_msg):
        """
        下载失败
        
        Args:
            error_msg: 错误信息
        """
        self.title_label.setText(f"❌ 下载失败")
        
        # 错误信息截断（避免太长）
        if len(error_msg) > 80:
            display_msg = error_msg[:80] + "..."
        else:
            display_msg = error_msg
        
        self.stage_label.setText("⚠️ 更新失败")
        self.status_label.setText(display_msg)
        self.status_label.setStyleSheet("color: #dc3545; font-size: 11px; font-weight: bold;")
        
        self.cancel_button.setText("关闭")
        self.cancel_button.setEnabled(True)
        self.retry_label.hide()
    
    def cancel_download(self):
        """取消下载"""
        if self.current_stage == self.STAGE_DOWNLOAD and self.downloader and self.downloader.isRunning():
            # 下载阶段可以取消
            from PyQt5.QtWidgets import QMessageBox
            reply = QMessageBox.question(
                self,
                "确认取消",
                "确定要取消更新吗？",
                QMessageBox.Yes | QMessageBox.No,
                QMessageBox.No
            )
            
            if reply == QMessageBox.Yes:
                self.downloader.cancel()
                self.downloader.wait()
                self.status_label.setText("更新已取消")
                self.reject()  # 会自动清理定时器
        else:
            # 其他阶段或下载已完成，直接关闭
            self.reject()  # 会自动清理定时器
    
    def cleanup_timers(self):
        """清理所有定时器"""
        if hasattr(self, 'uac_timer'):
            self.uac_timer.stop()
        if hasattr(self, 'install_timer'):
            self.install_timer.stop()
        if hasattr(self, 'restart_timer'):
            self.restart_timer.stop()
    
    def accept(self):
        """对话框接受时清理资源"""
        self.cleanup_timers()
        super().accept()
    
    def reject(self):
        """对话框拒绝时清理资源"""
        self.cleanup_timers()
        super().reject()


if __name__ == "__main__":
    # 测试代码
    import sys
    from PyQt5.QtWidgets import QApplication
    
    app = QApplication(sys.argv)
    
    # 创建测试对话框
    dialog = UpdateProgressDialog("1.5.55")
    dialog.show()
    
    # 模拟完整更新流程
    def simulate_update():
        # 阶段1：模拟下载 (0-50%)
        def simulate_download():
            total_size = 100 * 1024 * 1024  # 100MB
            for i in range(0, 101, 5):
                downloaded = int(i * total_size / 100)
                dialog.update_progress(downloaded, total_size)
                QApplication.processEvents()
                QTimer.singleShot(100, lambda: None)  # 延迟
            
            # 下载完成
            QTimer.singleShot(2000, lambda: dialog.download_finished("/tmp/test.exe"))
            
            # 阶段2：模拟UAC授权 (50-60%)
            QTimer.singleShot(5000, lambda: dialog.uac_authorized())
        
        simulate_download()
    
    QTimer.singleShot(500, simulate_update)
    
    sys.exit(app.exec_())