fix:优化启动方式
This commit is contained in:
161
ui/app_window.py
Normal file
161
ui/app_window.py
Normal file
@@ -0,0 +1,161 @@
|
||||
import tkinter as tk
|
||||
from tkinter import ttk, scrolledtext, messagebox, filedialog
|
||||
import threading
|
||||
import queue
|
||||
import sys
|
||||
|
||||
from loguru import logger
|
||||
from config.config import load_config
|
||||
from utils.log_handler import log_queue
|
||||
|
||||
# 导入业务逻辑
|
||||
from utils.generate_utils import (
|
||||
generate_template,
|
||||
generate_comment_all,
|
||||
batch_convert_folder,
|
||||
generate_report,
|
||||
generate_zodiac,
|
||||
)
|
||||
from utils.file_utils import export_templates_folder, initialize_project, export_data
|
||||
|
||||
config = load_config("config.toml")
|
||||
|
||||
class ReportApp:
|
||||
def __init__(self, root):
|
||||
self.root = root
|
||||
self.root.title("🌱 尚城幼儿园成长报告助手")
|
||||
self.root.geometry("720x760")
|
||||
|
||||
# 线程控制
|
||||
self.stop_event = threading.Event()
|
||||
self.is_running = False
|
||||
|
||||
self._setup_ui()
|
||||
self._start_log_polling()
|
||||
|
||||
def _setup_ui(self):
|
||||
# 样式配置
|
||||
self.style = ttk.Style()
|
||||
self.style.theme_use("clam")
|
||||
self.style.configure("TButton", font=("微软雅黑", 10), padding=5)
|
||||
self.style.configure("Title.TLabel", font=("微软雅黑", 16, "bold"), foreground="#2E8B57")
|
||||
self.style.configure("Stop.TButton", foreground="red", font=("微软雅黑", 10, "bold"))
|
||||
|
||||
# 1. 标题
|
||||
header = ttk.Frame(self.root, padding="10 15 10 5")
|
||||
header.pack(fill=tk.X)
|
||||
ttk.Label(header, text="🌱 尚城幼儿园成长报告助手", style="Title.TLabel").pack()
|
||||
ttk.Label(header, text="By 寒寒 | 这里的每一份评语都充满爱意", font=("微软雅黑", 9), foreground="gray").pack()
|
||||
|
||||
# 2. 功能区容器
|
||||
main_content = ttk.Frame(self.root, padding=10)
|
||||
main_content.pack(fill=tk.X)
|
||||
|
||||
# === A组: 核心功能 ===
|
||||
self._create_btn_group(main_content, "🛠️ 核心功能", [
|
||||
("📁 生成图片路径", lambda: self.run_task(generate_template)),
|
||||
("🤖 生成评语 (AI)", lambda: self.run_task(generate_comment_all)),
|
||||
("📊 生成报告 (PPT)", lambda: self.run_task(generate_report)),
|
||||
("📑 格式转换 (PDF)", lambda: self.run_task(batch_convert_folder, config.get("output_folder"))),
|
||||
("🐂 生肖转化 (生日)", lambda: self.run_task(generate_zodiac)),
|
||||
], columns=3)
|
||||
|
||||
# === B组: 数据管理 ===
|
||||
self._create_btn_group(main_content, "📦 数据管理", [
|
||||
("📦 导出模板 (Zip)", self.run_export_template),
|
||||
("📤 导出备份 (Zip)", self.run_export_data),
|
||||
], columns=2)
|
||||
|
||||
# === C组: 系统操作 (含停止按钮) ===
|
||||
self._create_btn_group(main_content, "⚙️ 系统操作", [
|
||||
("⛔ 停止当前任务", self.stop_current_task),
|
||||
("⚠️ 初始化系统", self.run_init),
|
||||
("🚪 退出系统", self.quit_app),
|
||||
], columns=3, special_styles={"⛔ 停止当前任务": "Stop.TButton"})
|
||||
|
||||
# 3. 日志区
|
||||
log_frame = ttk.LabelFrame(self.root, text="📝 系统实时日志", padding=10)
|
||||
log_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=(0, 10))
|
||||
|
||||
self.log_text = scrolledtext.ScrolledText(log_frame, height=10, state="disabled", font=("Consolas", 9))
|
||||
self.log_text.pack(fill=tk.BOTH, expand=True)
|
||||
|
||||
def _create_btn_group(self, parent, title, buttons, columns=2, special_styles=None):
|
||||
frame = ttk.LabelFrame(parent, text=title, padding=10)
|
||||
frame.pack(fill=tk.X, pady=5)
|
||||
special_styles = special_styles or {}
|
||||
|
||||
for i, (text, func) in enumerate(buttons):
|
||||
style = special_styles.get(text, "TButton")
|
||||
btn = ttk.Button(frame, text=text, command=func, style=style)
|
||||
r, c = divmod(i, columns)
|
||||
btn.grid(row=r, column=c, padx=5, pady=5, sticky="ew")
|
||||
|
||||
for i in range(columns):
|
||||
frame.columnconfigure(i, weight=1)
|
||||
|
||||
def _start_log_polling(self):
|
||||
while not log_queue.empty():
|
||||
try:
|
||||
msg = log_queue.get_nowait()
|
||||
self.log_text.config(state="normal")
|
||||
self.log_text.insert(tk.END, msg)
|
||||
self.log_text.see(tk.END)
|
||||
self.log_text.config(state="disabled")
|
||||
except queue.Empty:
|
||||
break
|
||||
self.root.after(100, self._start_log_polling)
|
||||
|
||||
# --- 任务运行核心逻辑 ---
|
||||
def run_task(self, target_func, *args, **kwargs):
|
||||
if self.is_running:
|
||||
messagebox.showwarning("忙碌中", "请先等待当前任务完成或点击【停止当前任务】")
|
||||
return
|
||||
|
||||
self.stop_event.clear()
|
||||
self.is_running = True
|
||||
|
||||
def thread_worker():
|
||||
try:
|
||||
# 尝试传入 stop_event
|
||||
try:
|
||||
target_func(*args, stop_event=self.stop_event, **kwargs)
|
||||
except TypeError:
|
||||
# 如果旧函数不支持 stop_event,则普通运行
|
||||
target_func(*args, **kwargs)
|
||||
except Exception as e:
|
||||
logger.error(f"任务出错: {e}")
|
||||
import traceback
|
||||
logger.error(traceback.format_exc())
|
||||
finally:
|
||||
self.is_running = False
|
||||
logger.info("--- 就绪 ---")
|
||||
|
||||
threading.Thread(target=thread_worker, daemon=True).start()
|
||||
|
||||
def stop_current_task(self):
|
||||
if not self.is_running:
|
||||
return
|
||||
if messagebox.askyesno("确认", "确定要中断当前任务吗?"):
|
||||
self.stop_event.set()
|
||||
logger.warning("正在发送停止信号...")
|
||||
|
||||
# --- 具体按钮事件 ---
|
||||
def run_export_template(self):
|
||||
path = filedialog.askdirectory()
|
||||
if path: self.run_task(export_templates_folder, path)
|
||||
|
||||
def run_export_data(self):
|
||||
path = filedialog.askdirectory()
|
||||
if path: self.run_task(export_data, path)
|
||||
|
||||
def run_init(self):
|
||||
if messagebox.askokcancel("警告", "确定重置系统吗?数据将丢失!"):
|
||||
self.run_task(initialize_project)
|
||||
|
||||
def quit_app(self):
|
||||
if self.is_running:
|
||||
messagebox.showwarning("提示", "请先停止任务")
|
||||
return
|
||||
self.root.destroy()
|
||||
sys.exit()
|
||||
Reference in New Issue
Block a user