3b7c00377f
Apply Black formatter to the entire codebase for consistent code style. Co-Authored-By: Claude (glm-5) <noreply@anthropic.com>
115 lines
3.1 KiB
Python
115 lines
3.1 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
进度对话框组件
|
|
|
|
用于显示长时间运行操作的进度。
|
|
"""
|
|
|
|
import tkinter as tk
|
|
from tkinter import ttk
|
|
from typing import Optional, Callable
|
|
|
|
|
|
class ProgressDialog:
|
|
"""进度对话框"""
|
|
|
|
def __init__(
|
|
self,
|
|
parent,
|
|
title: str = "处理中...",
|
|
message: str = "请稍候",
|
|
can_cancel: bool = True,
|
|
on_cancel: Optional[Callable] = None,
|
|
):
|
|
"""
|
|
初始化进度对话框
|
|
|
|
Args:
|
|
parent: 父窗口
|
|
title: 对话框标题
|
|
message: 显示消息
|
|
can_cancel: 是否可以取消
|
|
on_cancel: 取消回调函数
|
|
"""
|
|
self.parent = parent
|
|
self.can_cancel = can_cancel
|
|
self.on_cancel = on_cancel
|
|
self.cancelled = False
|
|
|
|
# 创建对话框
|
|
self.dialog = tk.Toplevel(parent)
|
|
self.dialog.title(title)
|
|
self.dialog.resizable(False, False)
|
|
self.dialog.transient(parent)
|
|
self.dialog.grab_set()
|
|
|
|
# 居中显示
|
|
self._center()
|
|
|
|
# 创建内容
|
|
self._create_widgets(message)
|
|
|
|
def _center(self):
|
|
"""将对话框居中显示"""
|
|
self.dialog.update_idletasks()
|
|
width = 400
|
|
height = 150
|
|
x = (self.dialog.winfo_screenwidth() // 2) - (width // 2)
|
|
y = (self.dialog.winfo_screenheight() // 2) - (height // 2)
|
|
self.dialog.geometry(f"{width}x{height}+{x}+{y}")
|
|
|
|
def _create_widgets(self, message: str):
|
|
"""创建对话框组件"""
|
|
# 消息标签
|
|
self.message_label = ttk.Label(self.dialog, text=message, wraplength=380)
|
|
self.message_label.pack(pady=(20, 10), padx=20)
|
|
|
|
# 进度条
|
|
self.progress = ttk.Progressbar(self.dialog, mode="indeterminate", length=360)
|
|
self.progress.pack(pady=10, padx=20)
|
|
self.progress.start(10)
|
|
|
|
# 取消按钮
|
|
if self.can_cancel:
|
|
button_frame = ttk.Frame(self.dialog)
|
|
button_frame.pack(pady=10)
|
|
|
|
self.cancel_button = ttk.Button(
|
|
button_frame, text="取消", command=self._on_cancel
|
|
)
|
|
self.cancel_button.pack()
|
|
|
|
def _on_cancel(self):
|
|
"""处理取消操作"""
|
|
self.cancelled = True
|
|
if self.on_cancel:
|
|
self.on_cancel()
|
|
self.close()
|
|
|
|
def update_message(self, message: str):
|
|
"""更新显示消息"""
|
|
self.message_label.config(text=message)
|
|
self.dialog.update_idletasks()
|
|
|
|
def set_progress(self, value: int, maximum: int = 100):
|
|
"""
|
|
设置进度值
|
|
|
|
Args:
|
|
value: 当前进度值
|
|
maximum: 最大值
|
|
"""
|
|
self.progress.config(mode="determinate", maximum=maximum)
|
|
self.progress["value"] = value
|
|
self.dialog.update_idletasks()
|
|
|
|
def close(self):
|
|
"""关闭对话框"""
|
|
self.progress.stop()
|
|
self.dialog.destroy()
|
|
|
|
def is_cancelled(self) -> bool:
|
|
"""检查是否已取消"""
|
|
return self.cancelled
|