论WebBrowser捕获window.close

本文介绍了一种在WebBrowser组件关闭时同步关闭宿主WinForm窗口的方法。通过重写WndProc方法并监听WM_PARENTNOTIFY与WM_DESTROY消息,实现了浏览器窗口关闭时触发宿主窗口的退出事件。

     最近在做项目的时候用到了WebBrowser,需要在关闭网页的同时关闭WebBroser所在的WinForm界面,在网上及MSDN找到的方法都不是很好,偶然在code project中找到了比较好的方法,特意摘录再次,供大家分享。

   

实现方案:

  1. 重写 WndProc 方法
  2. 判断 WM_PARENTNOTIFY 消息
  3. 判断 WM_DESTROY 消息参数
  4. 执行事件委托方法

实现源码参考:

 

refer:

Jeroen Landheer

import os import subprocess import platform import tkinter as tk from tkinter import ttk, messagebox, simpledialog, filedialog, Menu from openpyxl import Workbook, load_workbook from openpyxl.styles import Alignment, Font, Border, Side import json import webbrowser import datetime import time EXCEL_FILE = '零件登记.xlsx' PAIR_FILE = '零件配对.json' TITLE_FONT = Font(name='微软雅黑', size=18, bold=True) BODY_FONT = Font(name='微软雅黑', size=11) TITLE_ALIGN = Alignment(horizontal='center', vertical='center') BODY_ALIGN = Alignment(horizontal='center', vertical='center') thin = Side(border_style='thin', color='000000') all_border = Border(top=thin, left=thin, right=thin, bottom=thin) class PairManager: """零件配对管理类""" def __init__(self, master): self.master = master self.master.title("零件配对管理") self.master.geometry("600x400") self.master.resizable(False, False) # 设置为模态窗口,禁用主窗口 self.master.grab_set() self.master.transient(master.master) # 加载现有配对 self.pairs = self.load_pairs() # 创建界面 self.create_widgets() self.update_tree() # 设置窗口居中 self.center_window() def center_window(self): """使窗口居中显示""" self.master.update_idletasks() width = self.master.winfo_width() height = self.master.winfo_height() x = (self.master.winfo_screenwidth() // 2) - (width // 2) y = (self.master.winfo_screenheight() // 2) - (height // 2) self.master.geometry(f'+{x}+{y}') def create_widgets(self): # 主框架 main_frame = ttk.Frame(self.master) main_frame.pack(fill='both', expand=True, padx=10, pady=10) # 使用Treeview替代Listbox tree_frame = ttk.Frame(main_frame) tree_frame.pack(fill='both', expand=True) # 创建带滚动条的Treeview scrollbar = ttk.Scrollbar(tree_frame) scrollbar.pack(side='right', fill='y') self.tree = ttk.Treeview( tree_frame, columns=('零件编号', '零件名称'), show='headings', selectmode='extended', yscrollcommand=scrollbar.set ) self.tree.pack(fill='both', expand=True) scrollbar.config(command=self.tree.yview) # 设置列 self.tree.heading('零件编号', text='零件编号', anchor='center') self.tree.heading('零件名称', text='零件名称', anchor='center') self.tree.column('零件编号', width=150, anchor='center') self.tree.column('零件名称', width=400, anchor='center') # 按钮框架 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill='x', pady=10) ttk.Button(btn_frame, text="添加", command=self.add_pair).pack(side='left', padx=5) ttk.Button(btn_frame, text="编辑", command=self.edit_pair).pack(side='left', padx=5) ttk.Button(btn_frame, text="删除", command=self.delete_pair).pack(side='left', padx=5) ttk.Button(btn_frame, text="保存并关闭", command=self.save_and_close).pack(side='right', padx=5) def load_pairs(self): """加载零件配对数据""" try: if os.path.exists(PAIR_FILE): with open(PAIR_FILE, 'r', encoding='utf-8') as f: return json.load(f) except Exception as e: messagebox.showerror("错误", f"加载配对文件失败: {str(e)}") return {} def save_pairs(self): """保存零件配对数据""" try: with open(PAIR_FILE, 'w', encoding='utf-8') as f: json.dump(self.pairs, f, ensure_ascii=False, indent=2) return True except Exception as e: messagebox.showerror("错误", f"保存配对文件失败: {str(e)}") return False def update_tree(self): """更新Treeview显示""" # 清空现有数据 for item in self.tree.get_children(): self.tree.delete(item) # 添加新数据 for code, name in self.pairs.items(): self.tree.insert('', 'end', values=(code, name)) def add_pair(self): """添加新配对""" dialog = PairDialog(self.master, "添加零件配对") if dialog.result: code, name = dialog.result if code in self.pairs: messagebox.showwarning("警告", f"零件编号 {code} 已存在!") else: self.pairs[code] = name self.update_tree() def edit_pair(self): """编辑现有配对""" selection = self.tree.selection() if not selection: messagebox.showwarning("提示", "请先选择一个配对进行编辑") return # 如果只选了一个,执行单编辑 if len(selection) == 1: item = selection[0] code = self.tree.item(item, 'values')[0] name = self.pairs[code] dialog = PairDialog(self.master, "编辑零件配对", code, name) if dialog.result: new_code, new_name = dialog.result if new_code != code and new_code in self.pairs: messagebox.showwarning("警告", f"零件编号 {new_code} 已存在!") else: if new_code != code: del self.pairs[code] self.pairs[new_code] = new_name self.update_tree() else: # 多选编辑 - 批量编辑名称 selected_codes = [self.tree.item(item, 'values')[0] for item in selection] dialog = BatchEditDialog(self.master, "批量编辑零件名称", selected_codes, self.pairs) if dialog.result: for code in selected_codes: self.pairs[code] = dialog.result self.update_tree() def delete_pair(self): """删除配对""" selection = self.tree.selection() if not selection: messagebox.showwarning("提示", "请先选择一个配对") return selected_codes = [self.tree.item(item, 'values')[0] for item in selection] if messagebox.askyesno("确认", f"确定要删除选中的 {len(selected_codes)} 个配对吗?"): for code in selected_codes: del self.pairs[code] self.update_tree() def save_and_close(self): """保存并关闭窗口""" if self.save_pairs(): self.master.destroy() class PairDialog(simpledialog.Dialog): """零件配对对话框""" def __init__(self, parent, title, code="", name=""): self.initial_code = code self.initial_name = name super().__init__(parent, title) def body(self, frame): ttk.Label(frame, text="零件编号:").grid(row=0, column=0, padx=5, pady=5, sticky='e') self.code_var = tk.StringVar(value=self.initial_code) code_entry = ttk.Entry(frame, textvariable=self.code_var, width=25) code_entry.grid(row=0, column=1, padx=5, pady=5, sticky='we') ttk.Label(frame, text="零件名称:").grid(row=1, column=0, padx=5, pady=5, sticky='e') self.name_var = tk.StringVar(value=self.initial_name) name_entry = ttk.Entry(frame, textvariable=self.name_var, width=25) name_entry.grid(row=1, column=1, padx=5, pady=5, sticky='we') return code_entry # 初始焦点 def validate(self): code = self.code_var.get().strip() name = self.name_var.get().strip() if not code or not name: messagebox.showwarning("输入错误", "零件编号和名称都不能为空") return False return True def apply(self): self.result = (self.code_var.get().strip(), self.name_var.get().strip()) class BatchEditDialog(simpledialog.Dialog): """批量编辑对话框""" def __init__(self, parent, title, codes, pair_dict): self.codes = codes self.pair_dict = pair_dict super().__init__(parent, title) def body(self, frame): ttk.Label(frame, text=f"批量编辑 {len(self.codes)} 个零件").grid(row=0, column=0, columnspan=2, pady=10) # 显示前几个零件名称 preview_text = "\n".join([f"{code}: {self.pair_dict[code]}" for code in self.codes[:3]]) if len(self.codes) > 3: preview_text += f"\n...等 {len(self.codes)} 个零件" preview = tk.Text(frame, height=6, width=40, font=('微软雅黑', 9)) preview.grid(row=1, column=0, columnspan=2, padx=10, pady=5) preview.insert(tk.END, preview_text) preview.config(state=tk.DISABLED) ttk.Label(frame, text="新零件名称:").grid(row=2, column=0, padx=5, pady=10, sticky='e') self.name_var = tk.StringVar() name_entry = ttk.Entry(frame, textvariable=self.name_var, width=25) name_entry.grid(row=2, column=1, padx=5, pady=10, sticky='we') return name_entry def validate(self): name = self.name_var.get().strip() if not name: messagebox.showwarning("输入错误", "零件名称不能为空") return False return True def apply(self): self.result = self.name_var.get().strip() class HelpDialog: """帮助对话框""" def __init__(self, parent): self.parent = parent self.window = tk.Toplevel(parent) self.window.title("使用帮助") self.window.geometry("600x450") self.window.resizable(False, False) # 设置为模态窗口,禁用主窗口 self.window.grab_set() self.window.transient(parent) # 创建标签页 self.notebook = ttk.Notebook(self.window) self.notebook.pack(fill='both', expand=True, padx=10, pady=10) # 基本操作标签 basic_frame = ttk.Frame(self.notebook) self.notebook.add(basic_frame, text="基本操作") self.create_basic_tab(basic_frame) # 快捷键标签 shortcut_frame = ttk.Frame(self.notebook) self.notebook.add(shortcut_frame, text="快捷键") self.create_shortcut_tab(shortcut_frame) # 零件管理标签 pair_frame = ttk.Frame(self.notebook) self.notebook.add(pair_frame, text="零件管理") self.create_pair_tab(pair_frame) # 底部按钮 btn_frame = ttk.Frame(self.window) btn_frame.pack(fill='x', padx=10, pady=10) ttk.Button(btn_frame, text="关闭", command=self.window.destroy).pack(side='right') self.center_window() def center_window(self): """使窗口居中显示""" self.window.update_idletasks() width = self.window.winfo_width() height = self.window.winfo_height() x = (self.window.winfo_screenwidth() // 2) - (width // 2) y = (self.window.winfo_screenheight() // 2) - (height // 2) self.window.geometry(f'+{x}+{y}') def create_basic_tab(self, frame): content = """ 1. 添加零件 - 在左侧输入零件编号、名称和数量 - 点击【添加并写入 Excel】按钮或按 Ctrl+Enter - 零件信息将添加到右侧列表并保存到Excel 2. 编辑零件 - 在右侧列表右键点击零件 - 选择【编辑】修改单个零件 - 选择【批量编辑数量】修改多个零件数量 3. 删除零件 - 在右侧列表右键点击零件 - 选择【删除】删除选中零件 - 或按 Delete 键删除 4. 加载Excel - 点击【加载Excel】按钮导入已有数据 - 支持覆盖当前数据或追加数据 5. 打开Excel - 点击【打开 Excel】查看生成的Excel文件 """ text = tk.Text(frame, wrap=tk.WORD, font=('微软雅黑', 10)) text.pack(fill='both', expand=True, padx=10, pady=10) text.insert(tk.END, content.strip()) text.config(state=tk.DISABLED) def create_shortcut_tab(self, frame): content = """ 常用快捷键列表: Ctrl + Enter : 添加当前零件并写入Excel Delete : 删除选中的零件 Ctrl + E : 打开零件配对管理 Ctrl + O : 打开Excel文件 Ctrl + L : 加载Excel数据 Ctrl + H : 打开帮助文档 """ text = tk.Text(frame, wrap=tk.WORD, font=('微软雅黑', 10)) text.pack(fill='both', expand=True, padx=10, pady=10) text.insert(tk.END, content.strip()) text.config(state=tk.DISABLED) def create_pair_tab(self, frame): content = """ 零件配对管理: 1. 添加配对 - 点击【添加】按钮创建新零件配对 - 输入零件编号和名称 2. 编辑配对 - 选择单个配对点击【编辑】修改 - 选择多个配对点击【编辑】批量修改名称 3. 删除配对 - 选择单个或多个配对 - 点击【删除】按钮删除 4. 自动填充 - 在主界面输入零件编号时自动填充名称 - 输入零件名称时自动填充编号 """ text = tk.Text(frame, wrap=tk.WORD, font=('微软雅黑', 10)) text.pack(fill='both', expand=True, padx=10, pady=10) text.insert(tk.END, content.strip()) text.config(state=tk.DISABLED) class LogDialog: """日志对话框(只包含操作日志)""" def __init__(self, parent, logs): self.parent = parent self.logs = logs self.window = tk.Toplevel(parent) self.window.title("操作日志") self.window.geometry("800x500") self.window.resizable(True, True) # 设置为模态窗口,禁用主窗口 self.window.grab_set() self.window.transient(parent) # 创建框架 frame = ttk.Frame(self.window) frame.pack(fill='both', expand=True, padx=10, pady=10) # 创建带滚动条的文本框 scrollbar = ttk.Scrollbar(frame) scrollbar.pack(side='right', fill='y') self.log_text = tk.Text(frame, wrap=tk.WORD, font=('微软雅黑', 10), yscrollcommand=scrollbar.set) self.log_text.pack(fill='both', expand=True, padx=10, pady=10) scrollbar.config(command=self.log_text.yview) # 添加日志内容 for log in self.logs: self.log_text.insert(tk.END, log + "\n") self.log_text.config(state=tk.DISABLED) # 底部按钮 btn_frame = ttk.Frame(self.window) btn_frame.pack(fill='x', padx=10, pady=10) ttk.Button(btn_frame, text="关闭", command=self.window.destroy).pack(side='right') self.center_window() def center_window(self): """使窗口居中显示""" self.window.update_idletasks() width = self.window.winfo_width() height = self.window.winfo_height() x = (self.window.winfo_screenwidth() // 2) - (width // 2) y = (self.window.winfo_screenheight() // 2) - (height // 2) self.window.geometry(f'+{x}+{y}') class PartRegApp: def __init__(self, root): self.root = root root.title("零件登记工具") root.geometry("950x580") root.resizable(False, False) # 创建菜单栏 self.create_menu() # 加载零件配对数据 self.pair_dict = self.load_pair_data() # 初始化日志 self.logs = [] self.log("程序启动") # ---------- 顶部:设备型号 ---------- top_frame = ttk.Frame(root) top_frame.pack(fill='x', padx=5, pady=5) ttk.Label(top_frame, text="设备型号:").pack(side='left') self.device_model = tk.StringVar(value='DS-9C') ttk.Entry(top_frame, textvariable=self.device_model, width=15).pack(side='left', padx=5) # ---------- 主体 ---------- paned = ttk.PanedWindow(root, orient='horizontal') paned.pack(fill='both', expand=True, padx=5, pady=5) # 左侧 left = ttk.Frame(paned, width=300) paned.add(left) left.pack_propagate(False) # 零件信息输入框 frm = ttk.LabelFrame(left, text="零件信息", padding=10) frm.pack(fill='x') ttk.Label(frm, text="零件编号:").pack(anchor='w') self.code_var = tk.StringVar() # 使用Combobox替代Entry self.code_combo = ttk.Combobox(frm, textvariable=self.code_var, width=20) self.code_combo.pack(fill='x', pady=2) self.code_combo.bind('<KeyRelease>', self.on_code_keyrelease) self.code_combo.bind('<<ComboboxSelected>>', self.on_code_selected) # 绑定变量变化事件 self.code_var.trace_add('write', self.on_code_var_changed) self.code_timer = None # 用于延迟检索的计时器 ttk.Label(frm, text="零件名称:").pack(anchor='w', pady=(8, 0)) self.name_var = tk.StringVar() # 使用Combobox替代Entry self.name_combo = ttk.Combobox(frm, textvariable=self.name_var, width=20) self.name_combo.pack(fill='x', pady=2) self.name_combo.bind('<KeyRelease>', self.on_name_keyrelease) self.name_combo.bind('<<ComboboxSelected>>', self.on_name_selected) # 绑定变量变化事件 self.name_var.trace_add('write', self.on_name_var_changed) self.name_timer = None # 用于延迟检索的计时器 ttk.Label(frm, text="数量:").pack(anchor='w', pady=(8, 0)) self.qty_var = tk.StringVar() qty_entry = ttk.Entry(frm, textvariable=self.qty_var) qty_entry.pack(fill='x', pady=2) # 添加快捷键绑定 qty_entry.bind('<Return>', lambda event: self.add_and_write()) qty_entry.bind('<KP_Enter>', lambda event: self.add_and_write()) # 小键盘回车 btn = ttk.Button(frm, text="添加并写入 Excel (Ctrl+Enter)", command=self.add_and_write) btn.pack(pady=(15, 0)) # 操作按钮 - 垂直排列 btn_frame = ttk.Frame(left) btn_frame.pack(fill='x', pady=(10, 5)) # 按钮垂直排列 ttk.Button(btn_frame, text="加载Excel (Ctrl+L)", command=self.load_excel).pack(side='top', fill='x', padx=5, pady=2) ttk.Button(btn_frame, text="打开 Excel (Ctrl+O)", command=self.open_excel).pack(side='top', fill='x', padx=5, pady=2) ttk.Button(btn_frame, text="零件配对管理 (Ctrl+E)", command=self.open_pair_manager).pack(side='top', fill='x', padx=5, pady=2) # 右侧预览 right = ttk.Frame(paned) paned.add(right) self.tree = ttk.Treeview(right, columns=('零件编号', '零件名称', '数量'), show='headings', height=22) self.tree.pack(fill='both', expand=True) for col in ('零件编号', '零件名称', '数量'): self.tree.heading(col, text=col) self.tree.column(col, anchor='center', stretch=True) # 添加滚动条 scrollbar = ttk.Scrollbar(right, orient="vertical", command=self.tree.yview) scrollbar.pack(side='right', fill='y') self.tree.configure(yscrollcommand=scrollbar.set) self.tree.bind('<Button-3>', self.show_menu) self.menu = tk.Menu(root, tearoff=0) self.menu.add_command(label="编辑", command=self.edit_selected) self.menu.add_command(label="批量编辑数量", command=self.batch_edit_qty) self.menu.add_command(label="删除", command=self.delete_selected) # 状态栏 status_frame = ttk.Frame(root) status_frame.pack(side='bottom', fill='x') # 左侧状态信息 self.status = tk.StringVar(value="就绪") status_bar = ttk.Label(status_frame, textvariable=self.status, relief='sunken', anchor='w', padding=(5, 2)) status_bar.pack(side='left', fill='x', expand=True) # 右侧时间和公司信息 self.time_var = tk.StringVar() time_label = ttk.Label(status_frame, textvariable=self.time_var, relief='sunken', anchor='e', padding=(5, 2)) time_label.pack(side='right', fill='x') # 更新时间显示 self.update_time() self.data = [] self.updating = False # 防止自动填充时触发循环事件 self.cached_codes = list(self.pair_dict.keys()) self.cached_names = list(self.pair_dict.values()) # 绑定全局快捷键 self.root.bind('<Control-Return>', lambda event: self.add_and_write()) self.root.bind('<Control-o>', lambda event: self.open_excel()) self.root.bind('<Control-O>', lambda event: self.open_excel()) self.root.bind('<Control-l>', lambda event: self.load_excel()) self.root.bind('<Control-L>', lambda event: self.load_excel()) self.root.bind('<Control-e>', lambda event: self.open_pair_manager()) self.root.bind('<Control-E>', lambda event: self.open_pair_manager()) self.root.bind('<Control-h>', lambda event: self.show_help()) self.root.bind('<Control-H>', lambda event: self.show_help()) self.root.bind('<Delete>', lambda event: self.delete_selected()) self.root.bind('<KP_Delete>', lambda event: self.delete_selected()) # 小键盘Delete # 初始焦点 self.code_combo.focus_set() def create_menu(self): """创建菜单栏""" menubar = Menu(self.root) self.root.config(menu=menubar) # 文件菜单 file_menu = Menu(menubar, tearoff=0) file_menu.add_command(label="打开Excel (Ctrl+O)", command=self.open_excel) file_menu.add_command(label="加载Excel (Ctrl+L)", command=self.load_excel) file_menu.add_separator() file_menu.add_command(label="退出", command=self.root.quit) menubar.add_cascade(label="文件", menu=file_menu) # 日志菜单 log_menu = Menu(menubar, tearoff=0) log_menu.add_command(label="操作日志", command=lambda: self.show_logs()) menubar.add_cascade(label="日志", menu=log_menu) # 帮助菜单 help_menu = Menu(menubar, tearoff=0) help_menu.add_command(label="使用帮助 (Ctrl+H)", command=self.show_help) help_menu.add_command(label="关于", command=self.show_about) menubar.add_cascade(label="帮助", menu=help_menu) def log(self, message): """记录日志""" timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") log_entry = f"[{timestamp}] {message}" self.logs.append(log_entry) # 保持日志数量不超过100条 if len(self.logs) > 100: self.logs.pop(0) def show_logs(self): """显示日志对话框""" LogDialog(self.root, self.logs) def load_pair_data(self): """加载零件配对数据""" if os.path.exists(PAIR_FILE): try: with open(PAIR_FILE, 'r', encoding='utf-8') as f: return json.load(f) except: return {} return {} # ---------- 自动补全功能(优化版)---------- def filter_items(self, items, pattern): """根据输入模式过滤项目""" if not pattern: return items return [item for item in items if pattern.lower() in item.lower()] def on_code_keyrelease(self, event): """零件编号输入框按键释放事件(优化版)""" # 忽略方向键和回车键 if event.keysym in ('Up', 'Down', 'Return', 'Escape'): return # 取消之前的定时器(如果存在) if self.code_timer: self.root.after_cancel(self.code_timer) # 设置新的定时器(900毫秒后执行) self.code_timer = self.root.after(900, self.process_code_input) def process_code_input(self): """处理零件编号输入(延迟执行)""" code = self.code_var.get().strip() # 过滤并更新下拉列表 filtered_codes = self.filter_items(self.cached_codes, code) self.code_combo['values'] = filtered_codes # 如果输入框不为空,显示下拉列表 if code: self.code_combo.event_generate('<Down>') # 自动填充零件名称 if not self.updating and code in self.pair_dict: self.updating = True self.name_var.set(self.pair_dict[code]) self.updating = False def on_name_keyrelease(self, event): """零件名称输入框按键释放事件(优化版)""" # 忽略方向键和回车键 if event.keysym in ('Up', 'Down', 'Return', 'Escape'): return # 取消之前的定时器(如果存在) if self.name_timer: self.root.after_cancel(self.name_timer) # 设置新的定时器(500毫秒后执行) self.name_timer = self.root.after(500, self.process_name_input) def process_name_input(self): """处理零件名称输入(延迟执行)""" name = self.name_var.get().strip() # 过滤并更新下拉列表 filtered_names = self.filter_items(self.cached_names, name) self.name_combo['values'] = filtered_names # 如果输入框不为空,显示下拉列表 if name: self.name_combo.event_generate('<Down>') # 自动填充零件编号 if not self.updating: for code, n in self.pair_dict.items(): if n == name: self.updating = True self.code_var.set(code) self.updating = False break def on_code_selected(self, event): """零件编号下拉选项选择事件""" code = self.code_var.get().strip() if code in self.pair_dict: self.name_var.set(self.pair_dict[code]) def on_name_selected(self, event): """零件名称下拉选项选择事件""" name = self.name_var.get().strip() for code, n in self.pair_dict.items(): if n == name: self.code_var.set(code) break # ---------- 输入框变化事件 ---------- def on_code_var_changed(self, *args): """零件编号变化时的事件处理""" if self.updating: return code = self.code_var.get().strip() # 如果零件编号被清空,则同时清空零件名称 if not code: self.name_var.set('') def on_name_var_changed(self, *args): """零件名称变化时的事件处理""" if self.updating: return name = self.name_var.get().strip() # 如果零件名称被清空,则同时清空零件编号 if not name: self.code_var.set('') # ---------- 配对管理 ---------- def open_pair_manager(self): """打开零件配对管理窗口""" manager_window = tk.Toplevel(self.root) PairManager(manager_window) # 当管理窗口关闭后重新加载配对数据 manager_window.wait_window() self.pair_dict = self.load_pair_data() # 更新缓存 self.cached_codes = list(self.pair_dict.keys()) self.cached_names = list(self.pair_dict.values()) # 更新下拉列表 self.code_combo['values'] = self.cached_codes self.name_combo['values'] = self.cached_names self.log("零件配对数据已更新") # ---------- 工具 ---------- def set_status(self, msg): self.status.set(msg) self.root.update_idletasks() def update_time(self): """更新时间显示""" current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") self.time_var.set(f"广州图森机械设备有限公司 | {current_time}") self.root.after(1000, self.update_time) # 每秒更新一次 def add_and_write(self): code = self.code_var.get().strip() name = self.name_var.get().strip() qty_str = self.qty_var.get().strip() if not all([code, name, qty_str]): messagebox.showwarning("提示", "请完整填写零件信息!") return try: qty = int(qty_str) except ValueError: messagebox.showwarning("提示", "数量必须是整数!") return # 检查是否已存在相同零件 existing_index = None for i, item in enumerate(self.data): if item[0] == code and item[1] == name: existing_index = i break if existing_index is not None: # 更新现有零件的数量 old_qty = self.data[existing_index][2] new_qty = old_qty + qty self.data[existing_index] = (code, name, new_qty) self.set_status(f"已更新:{code} 数量从 {old_qty} 增加到 {new_qty}") self.log(f"更新零件: {code} ({name}) 数量 {old_qty} → {new_qty}") else: # 添加新零件 self.data.append((code, name, qty)) self.set_status(f"已添加:{code} × {qty}") self.log(f"添加零件: {code} ({name}) × {qty}") self.refresh_tree() self.write_excel() self.clear_inputs() def clear_inputs(self): self.code_var.set('') self.name_var.set('') self.qty_var.set('') self.code_combo.focus_set() # 焦点回到零件编号输入框 def refresh_tree(self): for item in self.tree.get_children(): self.tree.delete(item) for row in self.data: self.tree.insert('', 'end', values=row) def write_excel(self): wb = Workbook() ws = wb.active ws.title = "零件登记" # 标题 title = self.device_model.get().strip() or "零件登记" ws.merge_cells('A1:C1') ws['A1'] = title ws['A1'].font = TITLE_FONT ws['A1'].alignment = TITLE_ALIGN ws.row_dimensions[1].height = 45 * 0.75 # 表头 headers = ['零件编号', '零件名称', '数量'] for idx, h in enumerate(headers, 1): cell = ws.cell(row=2, column=idx, value=h) cell.font = BODY_FONT cell.alignment = BODY_ALIGN cell.border = all_border # 数据(数量列写入为数字) for r, (code, name, qty) in enumerate(self.data, 3): ws.cell(row=r, column=1, value=code).font = BODY_FONT ws.cell(row=r, column=2, value=name).font = BODY_FONT ws.cell(row=r, column=3, value=qty).font = BODY_FONT for c in range(1, 4): ws.cell(row=r, column=c).alignment = BODY_ALIGN ws.cell(row=r, column=c).border = all_border # 固定列宽 ws.column_dimensions['A'].width = 15 ws.column_dimensions['B'].width = 20 ws.column_dimensions['C'].width = 10 try: wb.save(EXCEL_FILE) self.log(f"Excel文件已保存: {EXCEL_FILE}") except PermissionError: messagebox.showerror("错误", "文件被占用,请关闭后重试!") self.log("保存Excel失败: 文件被占用") def open_excel(self): if not os.path.isfile(EXCEL_FILE): messagebox.showwarning("提示", f"未找到 {EXCEL_FILE}\n请先添加数据生成文件。") return system = platform.system() try: if system == 'Windows': os.startfile(EXCEL_FILE) elif system == 'Darwin': subprocess.run(['open', EXCEL_FILE]) else: subprocess.run(['xdg-open', EXCEL_FILE]) self.set_status(f"已打开 Excel:{EXCEL_FILE}") self.log(f"打开Excel文件: {EXCEL_FILE}") except Exception as e: messagebox.showerror("错误", f"无法打开文件:\n{e}") self.log(f"打开Excel失败: {str(e)}") # ---------- 加载Excel文件 ---------- def load_excel(self): """加载已有的Excel文件到预览区""" # 询问用户加载方式 if self.data: choice = messagebox.askquestion("加载选项", "请选择加载方式:\n\n'是' - 覆盖当前数据\n'否' - 追加到当前数据\n'取消' - 中止操作", icon='question', type='yesnocancel') if choice == 'cancel': return else: choice = 'yes' # 没有数据时默认覆盖 # 弹出文件选择对话框 file_path = filedialog.askopenfilename( title="选择零件登记表", filetypes=[("Excel文件", "*.xlsx"), ("所有文件", "*.*")] ) if not file_path: return # 用户取消选择 try: # 加载工作簿 wb = load_workbook(file_path) ws = wb.active # 如果是追加模式,保留当前数据 if choice == 'no': new_data = self.data.copy() self.log(f"开始追加Excel数据: {file_path}") else: # 覆盖模式 - 清空当前数据 new_data = [] # 读取标题(设备型号) if ws['A1'].value: self.device_model.set(ws['A1'].value) self.log(f"开始覆盖加载Excel数据: {file_path}") # 读取数据行(从第三行开始) for row in ws.iter_rows(min_row=3, values_only=True): # 跳过空行 if not any(row[:3]): continue # 确保数据格式正确 code = str(row[0]) if row[0] else "" name = str(row[1]) if row[1] else "" qty = row[2] if row[2] is not None else 0 # 如果是数字类型,直接使用 if isinstance(qty, (int, float)): qty = int(qty) else: try: qty = int(qty) except: qty = 0 # 添加到数据列表 new_data.append((code, name, qty)) # 更新数据 self.data = new_data # 刷新Treeview self.refresh_tree() # 更新状态 self.set_status(f"已加载文件: {os.path.basename(file_path)},共 {len(self.data)} 条记录") # 保存到默认位置 self.write_excel() self.log(f"成功加载Excel: {file_path},共 {len(self.data)} 条记录") except Exception as e: messagebox.showerror("加载错误", f"无法加载Excel文件:\n{str(e)}") self.log(f"加载Excel失败: {str(e)}") # ---------- 右键菜单功能 ---------- def show_menu(self, event): if self.tree.identify_row(event.y): self.menu.post(event.x_root, event.y_root) def edit_selected(self): """编辑单个选中的零件""" selected = self.tree.selection() if not selected or len(selected) > 1: return item = selected[0] values = self.tree.item(item, 'values') idx = self.tree.index(item) # 创建编辑对话框 dialog = tk.Toplevel(self.root) dialog.title("编辑零件") dialog.geometry("300x200") dialog.resizable(False, False) # 设置为模态窗口,禁用主窗口 dialog.grab_set() dialog.transient(self.root) # 零件编号 ttk.Label(dialog, text="零件编号:").place(x=20, y=20) code_var = tk.StringVar(value=values[0]) code_entry = ttk.Entry(dialog, textvariable=code_var, width=20) code_entry.place(x=100, y=20) # 零件名称 ttk.Label(dialog, text="零件名称:").place(x=20, y=60) name_var = tk.StringVar(value=values[1]) name_entry = ttk.Entry(dialog, textvariable=name_var, width=20) name_entry.place(x=100, y=60) # 数量 ttk.Label(dialog, text="数量:").place(x=20, y=100) qty_var = tk.StringVar(value=values[2]) qty_entry = ttk.Entry(dialog, textvariable=qty_var, width=10) qty_entry.place(x=100, y=100) def save_changes(): code = code_var.get().strip() name = name_var.get().strip() qty_str = qty_var.get().strip() if not all([code, name, qty_str]): messagebox.showwarning("输入错误", "所有字段都必须填写!") return try: qty = int(qty_str) except ValueError: messagebox.showwarning("输入错误", "数量必须是整数!") return # 记录原始值 original_code, original_name, original_qty = self.data[idx] # 更新数据 self.data[idx] = (code, name, qty) self.refresh_tree() self.write_excel() self.set_status(f"已更新: {code}") self.log(f"编辑零件: {original_code}({original_name})×{original_qty} → {code}({name})×{qty}") dialog.destroy() ttk.Button(dialog, text="保存", command=save_changes).place(x=120, y=140) ttk.Button(dialog, text="取消", command=dialog.destroy).place(x=200, y=140) # 居中对话框 dialog.update_idletasks() width = dialog.winfo_width() height = dialog.winfo_height() x = (dialog.winfo_screenwidth() // 2) - (width // 2) y = (dialog.winfo_screenheight() // 2) - (height // 2) dialog.geometry(f'+{x}+{y}') code_entry.focus_set() def batch_edit_qty(self): """批量编辑选中零件的数量""" selected = self.tree.selection() if not selected: return # 创建批量编辑对话框 dialog = tk.Toplevel(self.root) dialog.title("批量编辑数量") dialog.geometry("300x150") dialog.resizable(False, False) # 设置为模态窗口,禁用主窗口 dialog.grab_set() dialog.transient(self.root) # 显示选中零件数量 ttk.Label(dialog, text=f"选中 {len(selected)} 个零件").pack(pady=10) # 新数量输入框 ttk.Label(dialog, text="新数量:").pack() qty_var = tk.StringVar() qty_entry = ttk.Entry(dialog, textvariable=qty_var, width=10) qty_entry.pack() def apply_changes(): qty_str = qty_var.get().strip() if not qty_str: messagebox.showwarning("输入错误", "数量不能为空!") return try: new_qty = int(qty_str) except ValueError: messagebox.showwarning("输入错误", "数量必须是整数!") return # 更新所有选中零件的数量 for item in selected: idx = self.tree.index(item) code, name, old_qty = self.data[idx] self.data[idx] = (code, name, new_qty) self.log(f"批量更新: {code}({name}) 数量 {old_qty} → {new_qty}") self.refresh_tree() self.write_excel() self.set_status(f"已批量更新 {len(selected)} 个零件的数量为 {new_qty}") dialog.destroy() btn_frame = ttk.Frame(dialog) btn_frame.pack(pady=10) ttk.Button(btn_frame, text="应用", command=apply_changes).pack(side='left', padx=10) ttk.Button(btn_frame, text="取消", command=dialog.destroy).pack(side='left', padx=10) # 居中对话框 dialog.update_idletasks() width = dialog.winfo_width() height = dialog.winfo_height() x = (dialog.winfo_screenwidth() // 2) - (width // 2) y = (dialog.winfo_screenheight() // 2) - (height // 2) dialog.geometry(f'+{x}+{y}') qty_entry.focus_set() def delete_selected(self): """删除选中的零件""" selected = self.tree.selection() if not selected: return # 获取所有要删除的零件信息 to_delete = [] for item in selected: idx = self.tree.index(item) to_delete.append(self.data[idx]) if not messagebox.askyesno("确认", f"确定要删除选中的 {len(selected)} 个零件吗?"): return # 从后往前删除,避免索引变化问题 for item in sorted(selected, reverse=True): idx = self.tree.index(item) deleted_part = self.data.pop(idx) self.tree.delete(item) self.log(f"删除零件: {deleted_part[0]}({deleted_part[1]}) × {deleted_part[2]}") # 更新状态 if to_delete: deleted_codes = [item[0] for item in to_delete] if len(deleted_codes) > 3: codes_str = f"{', '.join(deleted_codes[:3])} 等共 {len(deleted_codes)} 个零件" else: codes_str = ", ".join(deleted_codes) self.set_status(f"已删除: {codes_str}") self.write_excel() # ---------- 帮助功能 ---------- def show_help(self): """显示帮助对话框""" HelpDialog(self.root) self.log("打开帮助文档") def show_about(self): """显示关于对话框""" about = tk.Toplevel(self.root) about.title("关于") about.geometry("360x250") about.resizable(False, False) # 设置为模态窗口,禁用主窗口 about.grab_set() about.transient(self.root) # 主框架 main_frame = ttk.Frame(about) main_frame.pack(fill='both', expand=True, padx=20, pady=20) # 公司名称 ttk.Label(main_frame, text="广州图森机械设备有限公司", font=('微软雅黑', 14, 'bold')).pack(pady=(10, 5)) # 软件名称和版本 ttk.Label(main_frame, text="零件登记工具", font=('微软雅黑', 12)).pack(pady=5) ttk.Label(main_frame, text="版本: 2.0", font=('微软雅黑', 10)).pack() # 分隔线 ttk.Separator(main_frame, orient='horizontal').pack(fill='x', pady=10) # 网站链接 website_frame = ttk.Frame(main_frame) website_frame.pack(pady=5) ttk.Label(website_frame, text="公司网站:", font=('微软雅黑', 9)).pack(side='left') # 创建可点击的链接 link = ttk.Label(website_frame, text="www.tusen.cn", font=('微软雅黑', 9, 'underline'), foreground="blue", cursor="hand2") link.pack(side='left', padx=5) link.bind("<Button-1>", lambda e: webbrowser.open("http://www.tusen.cn/")) # 版权信息 ttk.Label(main_frame, text="© 2025 广州图森机械设备有限公司", font=('微软雅黑', 9)).pack(side='bottom', pady=10) # 居中窗口 about.update_idletasks() width = about.winfo_width() height = about.winfo_height() x = (about.winfo_screenwidth() // 2) - (width // 2) y = (about.winfo_screenheight() // 2) - (height // 2) about.geometry(f'+{x}+{y}') # ---------- 其他功能 ---------- def clear_data(self): """清空当前数据""" if not self.data: return if messagebox.askyesno("确认", "确定要清空所有零件数据吗?"): self.log(f"清空所有零件数据,共 {len(self.data)} 条记录") self.data = [] self.refresh_tree() self.write_excel() self.set_status("已清空所有数据") # ---------------- 运行 ---------------- if __name__ == '__main__': root = tk.Tk() app = PartRegApp(root) root.mainloop() 这是我写的一个工具,现在的功能基本已经完善好了,你再帮我分析看下,我这个工具还需要哪些功能呢?或者哪些地方还需要优化呢?给我个建议看看
最新发布
08-15
# 主要优化点: # 1. 增加了多图片批量处理功能 # 2. 添加了历史记录保存和加载功能 # 3. 实现了结果导出功能(TXT/PDF/HTML) # 4. 优化了UI布局和主题系统 # 5. 增加了模型参数配置预设 # 6. 添加了系统托盘支持 # 7. 实现了结果自动保存 import sys import os import base64 import json import requests import webbrowser from datetime import datetime from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QTextEdit, QFileDialog, QGroupBox, QSlider, QDoubleSpinBox, QProgressBar, QSplitter, QMessageBox, QListWidget, QListWidgetItem, QSystemTrayIcon, QMenu, QAction, QComboBox, QTabWidget, QScrollArea, QCheckBox) from PyQt5.QtGui import (QPixmap, QFont, QPalette, QColor, QTextCursor, QIcon, QTextDocumentWriter, QTextDocument) from PyQt5.QtCore import Qt, QSize, QThread, pyqtSignal, QTimer, QSettings # 配置OLLAMA API设置 OLLAMA_HOST = "http://localhost:11434" HISTORY_FILE = "history.json" SETTINGS_FILE = "settings.ini" class EnhancedListWidgetItem(QListWidgetItem): """增强的列表项,支持图标和状态显示""" def __init__(self, text, is_title=False, icon=None, parent=None): super().__init__(text, parent) self.is_title = is_title if icon: self.setIcon(icon) if is_title: self.setSizeHint(QSize(200, 35)) font = QFont("Microsoft YaHei UI", 10, QFont.Bold) self.setFont(font) self.setForeground(QColor("#64ffda")) self.setFlags(Qt.NoItemFlags) self.setBackground(QColor("#112240")) class ModelLoaderThread(QThread): # ...保持原有实现... class ImageAnalysisThread(QThread): # ...保持原有实现... class ExportThread(QThread): export_finished = pyqtSignal(str, bool) def __init__(self, content, file_path, format_type, parent=None): super().__init__(parent) self.content = content self.file_path = file_path self.format_type = format_type def run(self): try: if self.format_type == "html": with open(self.file_path, "w", encoding="utf-8") as f: f.write(self.content) elif self.format_type == "txt": with open(self.file_path, "w", encoding="utf-8") as f: f.write(self.content) elif self.format_type == "pdf": doc = QTextDocument() doc.setHtml(self.content) writer = QTextDocumentWriter(self.file_path) writer.write(doc) self.export_finished.emit(self.file_path, True) except Exception as e: self.export_finished.emit(str(e), False) class MultiModalApp(QMainWindow): def __init__(self): super().__init__() self.image_paths = [] self.current_image_index = 0 self.history = [] self.settings = QSettings(SETTINGS_FILE, QSettings.IniFormat) self.initUI() self.load_settings() self.setWindowTitle("增强版多模态大模型图像解读系统") self.setGeometry(100, 100, 1920, 1000) # 初始化系统托盘 self.init_tray_icon() QTimer.singleShot(500, self.load_models) def initUI(self): # 创建暗色主题样式表 self.setStyleSheet(""" /* 主窗口样式 */ QMainWindow { background-color: #0a192f; } /* 分组框样式 */ QGroupBox { border: 2px solid #64ffda; border-radius: 10px; margin-top: 1ex; color: #ccd6f6; font-weight: bold; } /* 标签样式 */ QLabel { color: #ccd6f6; } /* 按钮样式 */ QPushButton { background-color: #112240; color: #64ffda; border: 1px solid #64ffda; border-radius: 5px; padding: 5px 10px; font-weight: bold; } /* 文本框样式 */ QTextEdit { background-color: #0a192f; color: #a8b2d1; border: 1px solid #64ffda; border-radius: 5px; padding: 5px; font-size: 12pt; } /* 选项卡样式 */ QTabWidget::pane { border: 1px solid #64ffda; border-radius: 5px; background: #0a192f; } QTabBar::tab { background: #112240; color: #ccd6f6; padding: 8px; border: 1px solid #64ffda; border-bottom: none; border-top-left-radius: 4px; border-top-right-radius: 4px; } QTabBar::tab:selected { background: #233554; color: #64ffda; } """) # 设置主窗口布局 central_widget = QWidget() self.setCentralWidget(central_widget) main_layout = QVBoxLayout(central_widget) # 标题区域 title_layout = QHBoxLayout() self.title_label = QLabel("增强版多模态大模型图像解读系统") self.title_label.setStyleSheet("font-size: 24pt; font-weight: bold; color: #64ffda;") title_layout.addWidget(self.title_label) # 添加主题切换按钮 self.theme_button = QPushButton("切换主题") self.theme_button.clicked.connect(self.toggle_theme) title_layout.addWidget(self.theme_button) main_layout.addLayout(title_layout) # 主内容区域 splitter = QSplitter(Qt.Horizontal) # 左侧控制面板 left_panel = QWidget() left_layout = QVBoxLayout(left_panel) # 图片预览区域(改为选项卡形式) self.image_tabs = QTabWidget() self.image_tabs.setTabsClosable(True) self.image_tabs.tabCloseRequested.connect(self.close_image_tab) left_layout.addWidget(self.image_tabs) # 控制面板区域 control_tabs = QTabWidget() # 模型控制选项卡 model_tab = QWidget() self.setup_model_tab(model_tab) control_tabs.addTab(model_tab, "模型设置") # 参数控制选项卡 param_tab = QWidget() self.setup_param_tab(param_tab) control_tabs.addTab(param_tab, "参数设置") # 预设控制选项卡 preset_tab = QWidget() self.setup_preset_tab(preset_tab) control_tabs.addTab(preset_tab, "预设管理") left_layout.addWidget(control_tabs) # 右侧结果面板 right_panel = QWidget() right_layout = QVBoxLayout(right_panel) # 结果展示区域 result_tabs = QTabWidget() # 分析结果选项卡 self.result_tab = QWidget() self.setup_result_tab(self.result_tab) result_tabs.addTab(self.result_tab, "分析结果") # 历史记录选项卡 self.history_tab = QWidget() self.setup_history_tab(self.history_tab) result_tabs.addTab(self.history_tab, "历史记录") right_layout.addWidget(result_tabs) # 添加面板到分割器 splitter.addWidget(left_panel) splitter.addWidget(right_panel) splitter.setSizes([800, 1100]) main_layout.addWidget(splitter) # 状态栏 self.status_bar = self.statusBar() self.status_bar.setStyleSheet("background-color: #112240; color: #64ffda;") self.progress_bar = QProgressBar() self.progress_bar.setRange(0, 100) self.status_bar.addPermanentWidget(self.progress_bar) self.status_bar.showMessage("系统已就绪") def setup_model_tab(self, tab): layout = QVBoxLayout(tab) # 模型选择区域 model_group = QGroupBox("模型选择") model_layout = QVBoxLayout() self.model_list = QListWidget() self.model_list.setMaximumHeight(150) self.model_list.addItem("正在加载模型...") self.refresh_models_button = QPushButton("刷新模型列表") self.refresh_models_button.clicked.connect(self.load_models) model_layout.addWidget(self.model_list) model_layout.addWidget(self.refresh_models_button) model_group.setLayout(model_layout) layout.addWidget(model_group) # 图片操作区域 image_group = QGroupBox("图片操作") image_layout = QVBoxLayout() self.load_button = QPushButton("加载图片") self.load_button.clicked.connect(self.load_image) self.load_multiple_button = QPushButton("批量加载图片") self.load_multiple_button.clicked.connect(self.load_multiple_images) self.clear_images_button = QPushButton("清除所有图片") self.clear_images_button.clicked.connect(self.clear_all_images) image_layout.addWidget(self.load_button) image_layout.addWidget(self.load_multiple_button) image_layout.addWidget(self.clear_images_button) image_group.setLayout(image_layout) layout.addWidget(image_group) # 分析控制区域 control_group = QGroupBox("分析控制") control_layout = QVBoxLayout() self.analyze_button = QPushButton("分析当前图片") self.analyze_button.clicked.connect(self.analyze_image) self.analyze_button.setEnabled(False) self.analyze_all_button = QPushButton("批量分析所有图片") self.analyze_all_button.clicked.connect(self.analyze_all_images) self.analyze_all_button.setEnabled(False) self.stop_button = QPushButton("停止分析") self.stop_button.clicked.connect(self.stop_analysis) self.stop_button.setEnabled(False) control_layout.addWidget(self.analyze_button) control_layout.addWidget(self.analyze_all_button) control_layout.addWidget(self.stop_button) control_group.setLayout(control_layout) layout.addWidget(control_group) def setup_param_tab(self, tab): layout = QVBoxLayout(tab) # 温度控制 temp_group = QGroupBox("温度控制") temp_layout = QHBoxLayout() self.temp_slider = QSlider(Qt.Horizontal) self.temp_slider.setRange(0, 100) self.temp_slider.setValue(50) self.temp_value = QDoubleSpinBox() self.temp_value.setRange(0.0, 1.0) self.temp_value.setSingleStep(0.1) self.temp_value.setValue(0.5) self.temp_value.setDecimals(1) self.temp_slider.valueChanged.connect(lambda val: self.temp_value.setValue(val / 100)) self.temp_value.valueChanged.connect(lambda val: self.temp_slider.setValue(int(val * 100))) temp_layout.addWidget(self.temp_slider) temp_layout.addWidget(self.temp_value) temp_group.setLayout(temp_layout) layout.addWidget(temp_group) # Token控制 token_group = QGroupBox("Token控制") token_layout = QHBoxLayout() self.token_spin = QDoubleSpinBox() self.token_spin.setRange(100, 5000) self.token_spin.setValue(1000) self.token_spin.setSingleStep(100) token_layout.addWidget(QLabel("最大Token:")) token_layout.addWidget(self.token_spin) token_group.setLayout(token_layout) layout.addWidget(token_group) # 提示词区域 prompt_group = QGroupBox("提示词设置") prompt_layout = QVBoxLayout() self.prompt_edit = QTextEdit() self.prompt_edit.setPlainText("请用中文详细描述这张图片的内容,要求描述清晰、有条理,分段落呈现,各段首行按要求缩进2个汉字。") prompt_layout.addWidget(self.prompt_edit) prompt_group.setLayout(prompt_layout) layout.addWidget(prompt_group) def setup_preset_tab(self, tab): layout = QVBoxLayout(tab) # 预设管理 preset_group = QGroupBox("预设管理") preset_layout = QVBoxLayout() self.preset_combo = QComboBox() self.preset_combo.addItems(["默认预设", "详细描述", "创意写作", "技术分析"]) button_layout = QHBoxLayout() self.load_preset_button = QPushButton("加载预设") self.load_preset_button.clicked.connect(self.load_preset) self.save_preset_button = QPushButton("保存预设") self.save_preset_button.clicked.connect(self.save_preset) self.delete_preset_button = QPushButton("删除预设") self.delete_preset_button.clicked.connect(self.delete_preset) button_layout.addWidget(self.load_preset_button) button_layout.addWidget(self.save_preset_button) button_layout.addWidget(self.delete_preset_button) preset_layout.addWidget(self.preset_combo) preset_layout.addLayout(button_layout) preset_group.setLayout(preset_layout) layout.addWidget(preset_group) # 自动保存设置 auto_save_group = QGroupBox("自动保存设置") auto_save_layout = QVBoxLayout() self.auto_save_check = QCheckBox("自动保存分析结果") self.auto_save_check.setChecked(True) self.auto_save_path_button = QPushButton("选择保存路径") self.auto_save_path_button.clicked.connect(self.select_auto_save_path) self.auto_save_path_label = QLabel("默认保存位置: 程序目录/results") self.auto_save_path_label.setWordWrap(True) auto_save_layout.addWidget(self.auto_save_check) auto_save_layout.addWidget(self.auto_save_path_button) auto_save_layout.addWidget(self.auto_save_path_label) auto_save_group.setLayout(auto_save_layout) layout.addWidget(auto_save_group) def setup_result_tab(self, tab): layout = QVBoxLayout(tab) # 结果展示区域 result_group = QGroupBox("分析结果") result_layout = QVBoxLayout() self.result_edit = QTextEdit() self.result_edit.setReadOnly(True) # 结果操作按钮 button_layout = QHBoxLayout() self.save_result_button = QPushButton("保存结果") self.save_result_button.clicked.connect(self.save_result) self.copy_result_button = QPushButton("复制结果") self.copy_result_button.clicked.connect(self.copy_result) self.clear_result_button = QPushButton("清除结果") self.clear_result_button.clicked.connect(self.clear_results) button_layout.addWidget(self.save_result_button) button_layout.addWidget(self.copy_result_button) button_layout.addWidget(self.clear_result_button) result_layout.addWidget(self.result_edit) result_layout.addLayout(button_layout) result_group.setLayout(result_layout) layout.addWidget(result_group) def setup_history_tab(self, tab): layout = QVBoxLayout(tab) # 历史记录区域 history_group = QGroupBox("历史记录") history_layout = QVBoxLayout() self.history_list = QListWidget() self.history_list.itemDoubleClicked.connect(self.load_history_item) # 历史操作按钮 button_layout = QHBoxLayout() self.load_history_button = QPushButton("加载历史") self.load_history_button.clicked.connect(self.load_history) self.clear_history_button = QPushButton("清除历史") self.clear_history_button.clicked.connect(self.clear_history) button_layout.addWidget(self.load_history_button) button_layout.addWidget(self.clear_history_button) history_layout.addWidget(self.history_list) history_layout.addLayout(button_layout) history_group.setLayout(history_layout) layout.addWidget(history_group) def init_tray_icon(self): """初始化系统托盘图标""" self.tray_icon = QSystemTrayIcon(self) self.tray_icon.setIcon(QIcon(":/icons/app_icon.png")) tray_menu = QMenu() show_action = QAction("显示窗口", self) show_action.triggered.connect(self.show_normal) tray_menu.addAction(show_action) exit_action = QAction("退出", self) exit_action.triggered.connect(self.close) tray_menu.addAction(exit_action) self.tray_icon.setContextMenu(tray_menu) self.tray_icon.show() # 托盘图标点击事件 self.tray_icon.activated.connect(self.tray_icon_clicked) def show_normal(self): """从托盘恢复窗口显示""" self.showNormal() self.activateWindow() def tray_icon_clicked(self, reason): """处理托盘图标点击事件""" if reason == QSystemTrayIcon.DoubleClick: self.show_normal() def toggle_theme(self): """切换主题""" if self.theme_button.text() == "切换主题": # 切换到浅色主题 self.setStyleSheet(""" QMainWindow { background-color: #f5f5f5; } QGroupBox { border: 2px solid #2c3e50; border-radius: 10px; margin-top: 1ex; color: #2c3e50; font-weight: bold; } QLabel { color: #2c3e50; } QPushButton { background-color: #ecf0f1; color: #2c3e50; border: 1px solid #bdc3c7; border-radius: 5px; padding: 5px 10px; font-weight: bold; } QTextEdit { background-color: #ffffff; color: #2c3e50; border: 1px solid #bdc3c7; border-radius: 5px; padding: 5px; font-size: 12pt; } """) self.theme_button.setText("切换回暗色") self.title_label.setStyleSheet("font-size: 24pt; font-weight: bold; color: #2c3e50;") self.status_bar.setStyleSheet("background-color: #ecf0f1; color: #2c3e50;") else: # 切换回暗色主题 self.setStyleSheet(""" QMainWindow { background-color: #0a192f; } QGroupBox { border: 2px solid #64ffda; border-radius: 10px; margin-top: 1ex; color: #ccd6f6; font-weight: bold; } QLabel { color: #ccd6f6; } QPushButton { background-color: #112240; color: #64ffda; border: 1px solid #64ffda; border-radius: 5px; padding: 5px 10px; font-weight: bold; } QTextEdit { background-color: #0a192f; color: #a8b2d1; border: 1px solid #64ffda; border-radius: 5px; padding: 5px; font-size: 12pt; } """) self.theme_button.setText("切换主题") self.title_label.setStyleSheet("font-size: 24pt; font-weight: bold; color: #64ffda;") self.status_bar.setStyleSheet("background-color: #112240; color: #64ffda;") def load_settings(self): """加载程序设置""" self.settings.beginGroup("MainWindow") self.restoreGeometry(self.settings.value("geometry", self.saveGeometry())) self.restoreState(self.settings.value("windowState", self.saveState())) self.settings.endGroup() # 加载历史记录 self.load_history() def save_settings(self): """保存程序设置""" self.settings.beginGroup("MainWindow") self.settings.setValue("geometry", self.saveGeometry()) self.settings.setValue("windowState", self.saveState()) self.settings.endGroup() # 保存历史记录 self.save_history() def load_image(self): """加载单张图片""" file_path, _ = QFileDialog.getOpenFileName( self, "选择图片", "", "图片文件 (*.png *.jpg *.jpeg *.bmp *.gif)" ) if file_path: self.add_image_tab(file_path) def load_multiple_images(self): """批量加载多张图片""" file_paths, _ = QFileDialog.getOpenFileNames( self, "选择多张图片", "", "图片文件 (*.png *.jpg *.jpeg *.bmp *.gif)" ) if file_paths: for file_path in file_paths: self.add_image_tab(file_path) def add_image_tab(self, file_path): """添加图片选项卡""" try: pixmap = QPixmap(file_path) if pixmap.isNull(): raise Exception("无法加载图片文件,可能格式不支持或文件已损坏") scroll_area = QScrollArea() scroll_area.setWidgetResizable(True) image_label = QLabel() image_label.setPixmap(pixmap.scaled( scroll_area.width(), scroll_area.height(), Qt.KeepAspectRatio, Qt.SmoothTransformation )) image_label.setAlignment(Qt.AlignCenter) scroll_area.setWidget(image_label) tab_index = self.image_tabs.addTab( scroll_area, os.path.basename(file_path) ) self.image_tabs.setCurrentIndex(tab_index) self.image_paths.append(file_path) self.analyze_button.setEnabled(True) self.analyze_all_button.setEnabled(True) self.status_bar.showMessage(f"已加载图片: {os.path.basename(file_path)}") # 更新当前图片索引 self.current_image_index = tab_index except Exception as e: self.status_bar.showMessage(f"错误: {str(e)}") self.show_error_dialog("图片加载错误", f"无法加载图片:\n{str(e)}") def close_image_tab(self, index): """关闭图片选项卡""" if index < len(self.image_paths): self.image_paths.pop(index) self.image_tabs.removeTab(index) if not self.image_paths: self.analyze_button.setEnabled(False) self.analyze_all_button.setEnabled(False) def clear_all_images(self): """清除所有图片""" self.image_tabs.clear() self.image_paths.clear() self.analyze_button.setEnabled(False) self.analyze_all_button.setEnabled(False) self.status_bar.showMessage("已清除所有图片") def analyze_image(self): """分析当前图片""" current_index = self.image_tabs.currentIndex() if current_index < 0 or current_index >= len(self.image_paths): self.status_bar.showMessage("错误: 没有可分析的图片") return self.current_image_index = current_index self._analyze_image(self.image_paths[current_index]) def analyze_all_images(self): """批量分析所有图片""" if not self.image_paths: self.status_bar.showMessage("错误: 没有可分析的图片") return # 保存当前索引 saved_index = self.current_image_index # 逐个分析图片 for i, image_path in enumerate(self.image_paths): self.current_image_index = i self.image_tabs.setCurrentIndex(i) self._analyze_image(image_path) # 等待分析完成 while self.analysis_thread and self.analysis_thread.isRunning(): QApplication.processEvents() # 恢复原始索引 self.current_image_index = saved_index self.image_tabs.setCurrentIndex(saved_index) def _analyze_image(self, image_path): """实际执行图片分析的内部方法""" selected_items = self.model_list.selectedItems() if not selected_items: self.status_bar.showMessage("错误: 请选择模型") return model_name = selected_items[0].data(Qt.UserRole) self.result_edit.clear() self.progress_bar.setValue(0) self.progress_bar.setFormat("正在分析图片...") self.set_buttons_enabled(False) self.stop_button.setEnabled(True) temperature = self.temp_value.value() max_tokens = int(self.token_spin.value()) prompt = self.prompt_edit.toPlainText().strip() self.analysis_thread = ImageAnalysisThread( model_name, image_path, temperature, max_tokens, prompt ) self.analysis_thread.analysis_complete.connect(self.handle_analysis_result) self.analysis_thread.progress_updated.connect(self.update_progress) self.analysis_thread.error_occurred.connect(self.handle_analysis_error) self.analysis_thread.stream_data.connect(self.analysis_stream_data) self.analysis_thread.finished.connect(self.analysis_finished) self.analysis_thread.start() def handle_analysis_result(self, result): """处理分析结果""" timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") selected_items = self.model_list.selectedItems() model_name = selected_items[0].text()[2:] if selected_items else "未知模型" # 获取当前图片文件名 image_name = os.path.basename(self.image_paths[self.current_image_index]) # 格式化结果 formatted_result = self.format_result(result) result_html = f""" <div style='color:#64ffda; font-size:14pt; font-weight:bold; margin-bottom:10px;'> 图片分析结果: {image_name} </div> <div style='color:#ccd6f6; font-size:12pt; line-height:1.6;'>{formatted_result}</div> <div style='margin-top: 20px; color: #8892b0; font-size: 10pt; border-top: 1px solid #233554; padding-top: 10px;'> 模型: <span style='color: #64ffda;'>{model_name}</span> | 时间: <span style='color: #64ffda;'>{timestamp}</span> </div> """ self.result_edit.setHtml(result_html) self.status_bar.showMessage(f"图片分析完成: {image_name}") # 添加到历史记录 self.add_to_history(image_name, model_name, timestamp, result_html) # 自动保存结果 if self.auto_save_check.isChecked(): self.auto_save_result(image_name, result_html) def add_to_history(self, image_name, model_name, timestamp, content): """添加到历史记录""" history_item = { "image": image_name, "model": model_name, "time": timestamp, "content": content } self.history.insert(0, history_item) self.update_history_list() def update_history_list(self): """更新历史记录列表""" self.history_list.clear() for item in self.history[:50]: # 最多显示50条历史记录 list_item = QListWidgetItem(f"{item['image']} - {item['time']}") list_item.setData(Qt.UserRole, item) self.history_list.addItem(list_item) def load_history_item(self, item): """加载历史记录项""" history_data = item.data(Qt.UserRole) self.result_edit.setHtml(history_data["content"]) def load_history(self): """从文件加载历史记录""" try: if os.path.exists(HISTORY_FILE): with open(HISTORY_FILE, "r", encoding="utf-8") as f: self.history = json.load(f) self.update_history_list() except Exception as e: print(f"加载历史记录失败: {str(e)}") def save_history(self): """保存历史记录到文件""" try: with open(HISTORY_FILE, "w", encoding="utf-8") as f: json.dump(self.history, f, ensure_ascii=False, indent=2) except Exception as e: print(f"保存历史记录失败: {str(e)}") def clear_history(self): """清除历史记录""" self.history.clear() self.history_list.clear() def auto_save_result(self, image_name, content): """自动保存结果""" try: save_dir = self.settings.value("AutoSave/path", "results") if not os.path.exists(save_dir): os.makedirs(save_dir) # 生成文件名 timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") base_name = os.path.splitext(image_name)[0] file_name = f"{base_name}_{timestamp}.html" file_path = os.path.join(save_dir, file_name) # 保存文件 with open(file_path, "w", encoding="utf-8") as f: f.write(content) self.status_bar.showMessage(f"结果已自动保存到: {file_path}") except Exception as e: print(f"自动保存失败: {str(e)}") def select_auto_save_path(self): """选择自动保存路径""" save_dir = QFileDialog.getExistingDirectory( self, "选择自动保存目录", self.settings.value("AutoSave/path", "results") ) if save_dir: self.settings.setValue("AutoSave/path", save_dir) self.auto_save_path_label.setText(f"保存位置: {save_dir}") def save_result(self): """保存结果到文件""" if not self.result_edit.toPlainText(): self.status_bar.showMessage("错误: 没有可保存的内容") return file_path, _ = QFileDialog.getSaveFileName( self, "保存结果", "", "HTML文件 (*.html);;文本文件 (*.txt);;PDF文件 (*.pdf)" ) if file_path: format_type = "html" if file_path.endswith(".txt"): format_type = "txt" elif file_path.endswith(".pdf"): format_type = "pdf" content = self.result_edit.toHtml() if format_type != "txt" else self.result_edit.toPlainText() self.export_thread = ExportThread(content, file_path, format_type) self.export_thread.export_finished.connect(self.handle_export_finished) self.export_thread.start() self.status_bar.showMessage("正在导出结果...") def handle_export_finished(self, message, success): """处理导出完成""" if success: self.status_bar.showMessage(f"结果已保存到: {message}") if message.endswith(".html"): webbrowser.open(message) else: self.status_bar.showMessage(f"导出失败: {message}") self.show_error_dialog("导出错误", f"无法保存结果:\n{message}") def copy_result(self): """复制结果到剪贴板""" self.result_edit.selectAll() self.result_edit.copy() self.status_bar.showMessage("结果已复制到剪贴板") def clear_results(self): """清除结果""" self.result_edit.clear() self.status_bar.showMessage("已清除结果") def load_preset(self): """加载预设""" preset_name = self.preset_combo.currentText() if preset_name == "默认预设": self.temp_value.setValue(0.5) self.token_spin.setValue(1000) self.prompt_edit.setPlainText("请用中文详细描述这张图片的内容,要求描述清晰、有条理,分段落呈现,各段首行按要求缩进2个汉字。") elif preset_name == "详细描述": self.temp_value.setValue(0.3) self.token_spin.setValue(1500) self.prompt_edit.setPlainText("请用中文详细描述这张图片中的每一个细节,包括但不限于场景、人物、物体、颜色、空间关系等。要求描述系统全面,层次分明,每段描述一个方面。") elif preset_name == "创意写作": self.temp_value.setValue(0.7) self.token_spin.setValue(2000) self.prompt_edit.setPlainText("请根据这张图片创作一个富有想象力的故事或诗歌。要求内容生动有趣,语言优美,可以适当发挥想象力,但不要偏离图片内容太远。") elif preset_name == "技术分析": self.temp_value.setValue(0.2) self.token_spin.setValue(1200) self.prompt_edit.setPlainText("请从技术角度分析这张图片,包括但不限于构图、色彩、光线、透视等专业要素。要求分析专业准确,使用适当的专业术语。") self.status_bar.showMessage(f"已加载预设: {preset_name}") def save_preset(self): """保存当前设置为预设""" preset_name, ok = QInputDialog.getText( self, "保存预设", "请输入预设名称:", QLineEdit.Normal, self.preset_combo.currentText() ) if ok and preset_name: # 检查是否已存在 index = self.preset_combo.findText(preset_name) if index == -1: self.preset_combo.addItem(preset_name) self.preset_combo.setCurrentText(preset_name) self.status_bar.showMessage(f"预设 '{preset_name}' 已保存") def delete_preset(self): """删除预设""" preset_name = self.preset_combo.currentText() if preset_name in ["默认预设", "详细描述", "创意写作", "技术分析"]: QMessageBox.warning(self, "警告", "系统预设不能被删除") return reply = QMessageBox.question( self, "确认删除", f"确定要删除预设 '{preset_name}' 吗?", QMessageBox.Yes | QMessageBox.No ) if reply == QMessageBox.Yes: index = self.preset_combo.currentIndex() self.preset_combo.removeItem(index) self.status_bar.showMessage(f"预设 '{preset_name}' 已删除") def closeEvent(self, event): """关闭窗口事件""" if self.analysis_thread and self.analysis_thread.isRunning(): self.analysis_thread.stop() if self.model_loader_thread and self.model_loader_thread.isRunning(): self.model_loader_thread.terminate() self.model_loader_thread.wait(2000) self.save_settings() event.accept() if __name__ == "__main__": def exception_handler(exctype, value, traceback): error_msg = f"程序发生未捕获的异常:\n\n类型: {exctype.__name__}\n\n描述: {value}" print(error_msg) try: app = QApplication.instance() if app is not None: msg = QMessageBox()
08-11
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值