513. Find Bottom Left Tree Value

本文介绍了一种使用BFS(广度优先搜索)算法来查找树的最底层第一个节点的方法,通过维护每层的第一个节点来实现目标。

题目连接

这道题让我们求树的最底层的最左边的元素。

解答:

       像这种求树的一层中的某个元素的题一般用bps。我们用val记录每一层的第一个元素,用bps遍历到最后,val就是最后一层的第一个元素。

      

  public int findBottomLeftValue(TreeNode root) {
        LinkedList<TreeNode> q = new LinkedList<>();
        int val = 0;
        q.offer(root);
        while (!q.isEmpty()) {
            val = q.peekFirst().val;
            int size = q.size();
            for (int i = 0; i < size; i++) {
                TreeNode cur = q.poll();
                if (cur.left != null) {
                    q.offer(cur.left);
                }
                if (cur.right != null) {
                    q.offer(cur.right);
                }
            }
        }
        return val;
    }

# -*- coding: utf-8 -*- import pandas as pd from datetime import datetime import tkinter as tk from tkinter import ttk, filedialog, messagebox import os import traceback import re from openpyxl import load_workbook from concurrent.futures import ThreadPoolExecutor from tkinter import font as tkfont class EnhancedVersionUpdaterApp: def __init__(self, root): self.root = root self.root.title("Excel批量修改工具") self.root.geometry("1300x900") # 设置全局字体 default_font = tkfont.nametofont("TkDefaultFont") default_font.configure(size=10) self.root.option_add("*Font", default_font) # 使用线程池提高性能 self.executor = ThreadPoolExecutor(max_workers=4) self.running_tasks = 0 # 初始化变量 self.file_path = "" self.old_project = "" self.new_project = "" self.old_date = "" self.new_date = datetime.now().strftime("%Y-%m-%d") self.old_responsible = "" self.new_responsible = "" self.project_updates = [] self.date_updates = [] self.responsible_updates = [] # 特殊sheet配置 self.special_sheets = { '変更履歴': {'process': False}, 'history': {'process': False}, 'log': {'process': False}, '封面': {'process': True, 'update_time': False}, '表紙': {'process': True, 'update_time': False}, 'cover': {'process': True, 'update_time': False} } # 时间格式正则表达式 self.date_patterns = [ r'\d{4}-\d{2}-\d{2}', # YYYY-MM-DD r'\d{4}/\d{2}/\d{2}', # YYYY/MM/DD r'\d{4}年\d{2}月\d{2}日', # 中文日期 r'\d{2}-\d{2}-\d{4}', # MM-DD-YYYY r'\d{2}/\d{2}/\d{4}' # MM/DD/YYYY ] self.create_enhanced_ui() self.setup_style() def setup_style(self): """设置界面样式""" style = ttk.Style() style.configure('TFrame', background='white') style.configure('TLabel', background='white') style.configure('Treeview', rowheight=25) style.configure('Treeview.Heading', font=('Arial', 10, 'bold')) style.configure('TNotebook.Tab', padding=[10, 5]) style.configure('TButton', padding=5) style.map('TButton', background=[('active', '#e6e6e6'), ('!active', '#f0f0f0')], foreground=[('active', 'black'), ('!active', 'black')]) def create_enhanced_ui(self): """创建增强版用户界面""" # 主框架 main_frame = ttk.Frame(self.root) main_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) # 顶部工具栏 toolbar_frame = ttk.Frame(main_frame) toolbar_frame.pack(fill=tk.X, pady=(0, 10)) # 文件选择区域 file_frame = ttk.LabelFrame(toolbar_frame, text="文件选择", padding=10) file_frame.pack(side=tk.LEFT, fill=tk.X, expand=True) ttk.Label(file_frame, text="Excel文件路径:").pack(side=tk.LEFT) self.file_entry = ttk.Entry(file_frame, width=60) self.file_entry.pack(side=tk.LEFT, padx=5, fill=tk.X, expand=True) ttk.Button(file_frame, text="浏览", command=self.browse_file).pack(side=tk.LEFT) # 进度条 self.progress_var = tk.DoubleVar() self.progress_bar = ttk.Progressbar(toolbar_frame, variable=self.progress_var, maximum=100) self.progress_bar.pack(side=tk.RIGHT, fill=tk.X, expand=True, padx=(10, 0)) self.progress_label = ttk.Label(toolbar_frame, text="就绪") self.progress_label.pack(side=tk.RIGHT, padx=5) # 主内容区域 - 使用Notebook实现标签页 notebook = ttk.Notebook(main_frame) notebook.pack(fill=tk.BOTH, expand=True) # 项目变更标签页 project_tab = ttk.Frame(notebook) notebook.add(project_tab, text="项目编号变更") self.create_project_tab(project_tab) # 时间变更标签页 date_tab = ttk.Frame(notebook) notebook.add(date_tab, text="时间变更") self.create_date_tab(date_tab) # 担当变更标签页 responsible_tab = ttk.Frame(notebook) notebook.add(responsible_tab, text="担当变更") self.create_responsible_tab(responsible_tab) # 底部状态栏 self.status_var = tk.StringVar() self.status_var.set("就绪") status_bar = ttk.Label(main_frame, textvariable=self.status_var, relief=tk.SUNKEN) status_bar.pack(fill=tk.X, pady=(5, 0)) def create_project_tab(self, parent): """创建项目变更标签页 - 仅表格可滚动""" # 主框架 main_frame = ttk.Frame(parent) main_frame.pack(fill=tk.BOTH, expand=True) # 输入区域 input_frame = ttk.Frame(main_frame) input_frame.pack(fill=tk.X, pady=5) ttk.Label(input_frame, text="原项目编号:").grid(row=0, column=0, sticky="w", padx=5) self.old_project_entry = ttk.Entry(input_frame, width=30) self.old_project_entry.grid(row=0, column=1, sticky="w", padx=5) ttk.Label(input_frame, text="新项目编号:").grid(row=1, column=0, sticky="w", padx=5) self.new_project_entry = ttk.Entry(input_frame, width=30) self.new_project_entry.grid(row=1, column=1, sticky="w", padx=5) ttk.Button(input_frame, text="查找项目", command=lambda: self.executor.submit(self.load_project_changes)).grid(row=0, column=2, rowspan=2, padx=10) # 表格容器框架 table_container = ttk.Frame(main_frame) table_container.pack(fill=tk.BOTH, expand=True, pady=5) # 创建Treeview和滚动条 tree_scroll_y = ttk.Scrollbar(table_container) tree_scroll_y.pack(side=tk.RIGHT, fill=tk.Y) tree_scroll_x = ttk.Scrollbar(table_container, orient=tk.HORIZONTAL) tree_scroll_x.pack(side=tk.BOTTOM, fill=tk.X) self.project_tree = ttk.Treeview( table_container, columns=("Sheet", "位置", "原项目", "新项目", "状态"), show="headings", height=15, yscrollcommand=tree_scroll_y.set, xscrollcommand=tree_scroll_x.set ) self.project_tree.pack(fill=tk.BOTH, expand=True) # 配置滚动条 tree_scroll_y.config(command=self.project_tree.yview) tree_scroll_x.config(command=self.project_tree.xview) # 配置列 for col, width in [("Sheet", 200), ("位置", 100), ("原项目", 200), ("新项目", 200), ("状态", 100)]: self.project_tree.column(col, width=width, anchor="w") self.project_tree.heading(col, text=col) # 操作按钮区域 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill=tk.X, pady=5) ttk.Button(btn_frame, text="标记更新", command=lambda: self.update_project_status("待更新")).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="标记不更新", command=lambda: self.update_project_status("不更新")).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="执行更新", command=lambda: self.executor.submit(self.apply_project_updates)).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="导出报告", command=self.export_project_report).pack(side=tk.RIGHT, padx=5) def create_date_tab(self, parent): """创建时间变更标签页 - 仅表格可滚动""" # 主框架 main_frame = ttk.Frame(parent) main_frame.pack(fill=tk.BOTH, expand=True) # 输入区域 input_frame = ttk.Frame(main_frame) input_frame.pack(fill=tk.X, pady=5) ttk.Label(input_frame, text="原时间:").grid(row=0, column=0, sticky="w", padx=5) self.old_date_entry = ttk.Entry(input_frame, width=30) self.old_date_entry.grid(row=0, column=1, sticky="w", padx=5) ttk.Label(input_frame, text="新时间:").grid(row=1, column=0, sticky="w", padx=5) self.new_date_entry = ttk.Entry(input_frame, width=30) self.new_date_entry.insert(0, self.new_date) self.new_date_entry.grid(row=1, column=1, sticky="w", padx=5) ttk.Button(input_frame, text="查找时间", command=lambda: self.executor.submit(self.load_date_changes)).grid(row=0, column=2, rowspan=2, padx=10) # 表格容器框架 table_container = ttk.Frame(main_frame) table_container.pack(fill=tk.BOTH, expand=True, pady=5) # 创建Treeview和滚动条 tree_scroll_y = ttk.Scrollbar(table_container) tree_scroll_y.pack(side=tk.RIGHT, fill=tk.Y) tree_scroll_x = ttk.Scrollbar(table_container, orient=tk.HORIZONTAL) tree_scroll_x.pack(side=tk.BOTTOM, fill=tk.X) self.date_tree = ttk.Treeview( table_container, columns=("Sheet", "位置", "原时间", "新时间", "状态"), show="headings", height=15, yscrollcommand=tree_scroll_y.set, xscrollcommand=tree_scroll_x.set ) self.date_tree.pack(fill=tk.BOTH, expand=True) # 配置滚动条 tree_scroll_y.config(command=self.date_tree.yview) tree_scroll_x.config(command=self.date_tree.xview) # 配置列 for col, width in [("Sheet", 200), ("位置", 100), ("原时间", 200), ("新时间", 200), ("状态", 100)]: self.date_tree.column(col, width=width, anchor="w") self.date_tree.heading(col, text=col) # 操作按钮区域 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill=tk.X, pady=5) ttk.Button(btn_frame, text="标记更新", command=lambda: self.update_date_status("待更新")).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="标记不更新", command=lambda: self.update_date_status("不更新")).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="执行更新", command=lambda: self.executor.submit(self.apply_date_updates)).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="导出报告", command=self.export_date_report).pack(side=tk.RIGHT, padx=5) def create_responsible_tab(self, parent): """创建担当变更标签页 - 仅表格可滚动""" # 主框架 main_frame = ttk.Frame(parent) main_frame.pack(fill=tk.BOTH, expand=True) # 输入区域 input_frame = ttk.Frame(main_frame) input_frame.pack(fill=tk.X, pady=5) ttk.Label(input_frame, text="原担当:").grid(row=0, column=0, sticky="w", padx=5) self.old_responsible_entry = ttk.Entry(input_frame, width=30) self.old_responsible_entry.grid(row=0, column=1, sticky="w", padx=5) ttk.Label(input_frame, text="新担当:").grid(row=1, column=0, sticky="w", padx=5) self.new_responsible_entry = ttk.Entry(input_frame, width=30) self.new_responsible_entry.grid(row=1, column=1, sticky="w", padx=5) ttk.Button(input_frame, text="查找担当", command=lambda: self.executor.submit(self.load_responsible_changes)).grid(row=0, column=2, rowspan=2, padx=10) # 表格容器框架 table_container = ttk.Frame(main_frame) table_container.pack(fill=tk.BOTH, expand=True, pady=5) # 创建Treeview和滚动条 tree_scroll_y = ttk.Scrollbar(table_container) tree_scroll_y.pack(side=tk.RIGHT, fill=tk.Y) tree_scroll_x = ttk.Scrollbar(table_container, orient=tk.HORIZONTAL) tree_scroll_x.pack(side=tk.BOTTOM, fill=tk.X) self.responsible_tree = ttk.Treeview( table_container, columns=("Sheet", "位置", "原担当", "新担当", "状态"), show="headings", height=15, yscrollcommand=tree_scroll_y.set, xscrollcommand=tree_scroll_x.set ) self.responsible_tree.pack(fill=tk.BOTH, expand=True) # 配置滚动条 tree_scroll_y.config(command=self.responsible_tree.yview) tree_scroll_x.config(command=self.responsible_tree.xview) # 配置列 for col, width in [("Sheet", 200), ("位置", 100), ("原担当", 200), ("新担当", 200), ("状态", 100)]: self.responsible_tree.column(col, width=width, anchor="w") self.responsible_tree.heading(col, text=col) # 操作按钮区域 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill=tk.X, pady=5) ttk.Button(btn_frame, text="标记更新", command=lambda: self.update_responsible_status("待更新")).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="标记不更新", command=lambda: self.update_responsible_status("不更新")).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="执行更新", command=lambda: self.executor.submit(self.apply_responsible_updates)).pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="导出报告", command=self.export_responsible_report).pack(side=tk.RIGHT, padx=5) def _display_data(self, treeview, data): """显示数据到表格 - 添加自动滚动到顶部功能""" treeview.delete(*treeview.get_children()) for update in data: tags = ("to_update",) if update["status"] == "待更新" else ("no_update",) if update["status"] == "不更新" else () treeview.insert("", "end", values=(update["sheet"], update["cell"], update["old_value"], update["new_value"], update["status"]), tags=tags) treeview.tag_configure("to_update", background="lightyellow") treeview.tag_configure("no_update", background="lightgray") # 自动滚动到顶部 treeview.yview_moveto(0) def browse_file(self): """浏览文件""" file_path = filedialog.askopenfilename( filetypes=[("Excel文件", "*.xlsx *.xls"), ("所有文件", "*.*")], title="选择Excel文件" ) if file_path: self.file_entry.delete(0, tk.END) self.file_entry.insert(0, file_path) self.file_path = file_path self.update_status(f"已选择文件: {os.path.basename(file_path)}") def update_status(self, message): """更新状态栏""" self.status_var.set(message) self.root.update_idletasks() def update_progress(self, value, message=None): """更新进度条""" self.progress_var.set(value) if message: self.progress_label.config(text=message) self.root.update_idletasks() def load_project_changes(self): """加载项目编号变更""" try: self.running_tasks += 1 self.update_progress(0, "正在查找项目编号...") self.file_path = self.file_entry.get() self.old_project = self.old_project_entry.get().strip() self.new_project = self.new_project_entry.get().strip() if not all([self.file_path, self.old_project, self.new_project]): messagebox.showerror("错误", "请填写所有必填字段") return if not os.path.exists(self.file_path): messagebox.showerror("错误", "文件不存在!") return self.project_updates = self._detect_changes( target_value=self.old_project, new_value=self.new_project, value_type="project" ) if not self.project_updates: messagebox.showinfo("提示", "未找到匹配的项目编号") return self._display_data(self.project_tree, self.project_updates) self.update_status(f"找到 {len(self.project_updates)} 处项目编号需要更新") self.update_progress(100, "查找完成") except Exception as e: messagebox.showerror("错误", f"加载失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 def load_date_changes(self): """加载时间变更""" try: self.running_tasks += 1 self.update_progress(0, "正在查找时间...") self.file_path = self.file_entry.get() self.old_date = self.old_date_entry.get().strip() self.new_date = self.new_date_entry.get().strip() if not all([self.file_path, self.old_date, self.new_date]): messagebox.showerror("错误", "请填写所有必填字段") return if not os.path.exists(self.file_path): messagebox.showerror("错误", "文件不存在!") return self.date_updates = self._detect_changes( target_value=self.old_date, new_value=self.new_date, value_type="date" ) if not self.date_updates: messagebox.showinfo("提示", "未找到匹配的时间") return self._display_data(self.date_tree, self.date_updates) self.update_status(f"找到 {len(self.date_updates)} 处时间需要更新") self.update_progress(100, "查找完成") except Exception as e: messagebox.showerror("错误", f"加载失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 def load_responsible_changes(self): """加载担当变更""" try: self.running_tasks += 1 self.update_progress(0, "正在查找担当...") self.file_path = self.file_entry.get() self.old_responsible = self.old_responsible_entry.get().strip() self.new_responsible = self.new_responsible_entry.get().strip() if not all([self.file_path, self.old_responsible, self.new_responsible]): messagebox.showerror("错误", "请填写所有必填字段") return if not os.path.exists(self.file_path): messagebox.showerror("错误", "文件不存在!") return self.responsible_updates = self._detect_changes( target_value=self.old_responsible, new_value=self.new_responsible, value_type="responsible" ) if not self.responsible_updates: messagebox.showinfo("提示", "未找到匹配的担当") return self._display_data(self.responsible_tree, self.responsible_updates) self.update_status(f"找到 {len(self.responsible_updates)} 处担当需要更新") self.update_progress(100, "查找完成") except Exception as e: messagebox.showerror("错误", f"加载失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 def _detect_changes(self, target_value, new_value, value_type): """通用变更检测方法""" updates = [] try: wb = load_workbook(self.file_path) total_sheets = len(wb.sheetnames) for i, sheet_name in enumerate(wb.sheetnames): self.update_progress((i+1)/total_sheets*100, f"正在处理工作表: {sheet_name}") sheet_config = self._get_sheet_config(sheet_name) if not sheet_config['process']: continue if value_type == "date" and not sheet_config.get('update_time', True): continue sheet = wb[sheet_name] if value_type == "date": cells = self._find_date_cells(sheet, target_value) else: cells = self._find_cells_with_value(sheet, target_value) for cell in cells: updates.append({ "sheet": sheet_name, "cell": cell.coordinate, "old_value": cell.value, "new_value": new_value, "status": "待审核" }) except Exception as e: print(f"DEBUG - 读取错误: {str(e)}") return updates def _display_data(self, treeview, data): """显示数据到表格""" treeview.delete(*treeview.get_children()) for update in data: tags = ("to_update",) if update["status"] == "待更新" else ("no_update",) if update["status"] == "不更新" else () treeview.insert("", "end", values=(update["sheet"], update["cell"], update["old_value"], update["new_value"], update["status"]), tags=tags) treeview.tag_configure("to_update", background="lightyellow") treeview.tag_configure("no_update", background="lightgray") def update_project_status(self, status): """更新项目变更状态""" self._update_status(self.project_tree, self.project_updates, status) def update_date_status(self, status): """更新时间变更状态""" self._update_status(self.date_tree, self.date_updates, status) def update_responsible_status(self, status): """更新担当变更状态""" self._update_status(self.responsible_tree, self.responsible_updates, status) def _update_status(self, treeview, data, status): """通用状态更新方法""" selected = treeview.selection() if not selected: messagebox.showwarning("警告", "请先选择记录") return for item in selected: index = treeview.index(item) data[index]["status"] = status self._display_data(treeview, data) def apply_project_updates(self): """执行项目编号变更""" self._apply_updates( updates=self.project_updates, success_message="项目编号更新完成!", treeview=self.project_tree ) def apply_date_updates(self): """执行时间变更""" self._apply_updates( updates=self.date_updates, success_message="时间更新完成!", treeview=self.date_tree ) def apply_responsible_updates(self): """执行担当变更""" self._apply_updates( updates=self.responsible_updates, success_message="担当更新完成!", treeview=self.responsible_tree ) def _apply_updates(self, updates, success_message, treeview): """通用更新应用方法""" if not any(u["status"] == "待更新" for u in updates): messagebox.showwarning("警告", "没有标记为'待更新'的记录") return try: self.running_tasks += 1 self.update_progress(0, "正在更新...") wb = load_workbook(self.file_path) total_updates = len([u for u in updates if u["status"] == "待更新"]) processed = 0 for update in [u for u in updates if u["status"] == "待更新"]: sheet = wb[update["sheet"]] sheet[update["cell"]] = update["new_value"] processed += 1 self.update_progress(processed/total_updates*100, f"正在更新 {update['sheet']} {update['cell']}") wb.save(self.file_path) messagebox.showinfo("成功", success_message) # 重新加载数据 if treeview == self.project_tree: self.load_project_changes() elif treeview == self.date_tree: self.load_date_changes() else: self.load_responsible_changes() except Exception as e: messagebox.showerror("错误", f"更新失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 self.update_progress(100, "更新完成") def export_project_report(self): """导出项目变更报告""" self._export_report(self.project_updates, "项目变更报告") def export_date_report(self): """导出时间变更报告""" self._export_report(self.date_updates, "时间变更报告") def export_responsible_report(self): """导出担当变更报告""" self._export_report(self.responsible_updates, "担当变更报告") def _export_report(self, data, report_name): """通用报告导出方法""" if not data: messagebox.showwarning("警告", f"没有可导出的{report_name}数据") return try: file_path = filedialog.asksaveasfilename( defaultextension=".xlsx", filetypes=[("Excel文件", "*.xlsx")], title=f"保存{report_name}" ) if not file_path: return df = pd.DataFrame([{ "工作表": item["sheet"], "单元格位置": item["cell"], "原内容": item["old_value"], "新内容": item["new_value"], "状态": item["status"] } for item in data]) df.to_excel(file_path, index=False) messagebox.showinfo("成功", f"{report_name}已导出到: {file_path}") except Exception as e: messagebox.showerror("错误", f"导出失败: {str(e)}") def _get_sheet_config(self, sheet_name): """获取sheet配置""" sheet_lower = sheet_name.lower() for kw in self.special_sheets: if kw.lower() in sheet_lower: config = self.special_sheets[kw].copy() config['is_special'] = True return config return {'process': True, 'update_time': True, 'is_special': False} def _find_cells_with_value(self, sheet, target_value): """查找包含目标值的所有单元格""" found_cells = [] pattern = re.compile(rf'.*{re.escape(str(target_value))}.*', re.IGNORECASE) for row in sheet.iter_rows(): for cell in row: if cell.value and pattern.search(str(cell.value)): found_cells.append(cell) return found_cells def _find_date_cells(self, sheet, target_date=None): """查找所有包含日期的单元格""" date_cells = [] for row in sheet.iter_rows(): for cell in row: if cell.value and self._is_date(cell.value): if target_date: try: cell_date = pd.to_datetime(cell.value).strftime('%Y-%m-%d') if target_date in str(cell_date): date_cells.append(cell) except: if target_date in str(cell.value): date_cells.append(cell) else: date_cells.append(cell) return date_cells def _is_date(self, value): """判断值是否为日期""" try: pd.to_datetime(value) return True except: # 检查是否是字符串形式的日期 if isinstance(value, str): for pattern in self.date_patterns: if re.fullmatch(pattern, value.strip()): return True return False def on_closing(self): """关闭窗口时的处理""" if self.running_tasks > 0: if messagebox.askokcancel("警告", "有任务正在运行,确定要退出吗?"): self.executor.shutdown(wait=False) self.root.destroy() else: self.root.destroy() if __name__ == "__main__": root = tk.Tk() app = EnhancedVersionUpdaterApp(root) root.protocol("WM_DELETE_WINDOW", app.on_closing) root.mainloop() 1、请增加一些关于文档检查的功能 2、请美化一下界面 3、请加快一下运行速度,并减少对内存的占用
08-14
import tkinter as tk from tkinter import filedialog, ttk, messagebox import pandas as pd import os import math from datetime import datetime import matplotlib.pyplot as plt from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg from matplotlib import ticker import numpy as np # 设置matplotlib中文字体 plt.rcParams["font.family"] = ["SimHei", "WenQuanYi Micro Hei", "Heiti TC"] class YToXCalculator: def __init__(self, root): self.root = root self.root.title("F0计算工具(多格式支持)") self.root.geometry("1400x900") self.root.resizable(True, True) # 设置中文字体 self.style = ttk.Style() self.style.configure("TLabel", font=("SimHei", 9)) self.style.configure("TButton", font=("SimHei", 9)) self.style.configure("TEntry", font=("SimHei", 9)) self.style.configure("Treeview", font=("SimHei", 9)) # 数据存储 self.file_data = {} self.current_file = None self.x_values = None self.curve_data = {} # 存储所有曲线,异常曲线值为None self.curve_vars = {} self.results = [] # 起始列设置(默认从第1列开始) self.start_col = tk.IntVar(value=1) # TXT文件分隔符设置 self.delimiter = tk.StringVar(value="\t") # 默认制表符分隔 # 计算范围设置 self.calc_start_x = tk.StringVar(value="") self.calc_end_x = tk.StringVar(value="") # 错误提示控制 self.show_errors = tk.BooleanVar(value=True) # 默认显示错误提示 # 图表相关变量 self.fig = None self.canvas = None self.ax = None self.lines = [] self.selected_curve = None self.coord_text = None self.range_rect = None # 用于显示计算范围的矩形 self.cursor_line_x = None # X光标线 self.cursor_line_y = None # Y光标线 self.cursor_point = None # 光标在曲线上的点 # 创建界面 self.create_widgets() def create_widgets(self): # 主窗口分割 self.main_paned = ttk.PanedWindow(self.root, orient=tk.VERTICAL) self.main_paned.pack(fill=tk.BOTH, expand=True, padx=2, pady=2) # 上方控制区域 self.controls_frame = ttk.Frame(self.main_paned) self.main_paned.add(self.controls_frame, weight=5) # 下方内容区域 self.content_paned = ttk.PanedWindow(self.main_paned, orient=tk.VERTICAL) self.main_paned.add(self.content_paned, weight=16) # 图表区域 self.chart_frame = ttk.LabelFrame(self.content_paned, text="数据图表") self.content_paned.add(self.chart_frame, weight=14) # 结果区域 self.result_frame = ttk.LabelFrame(self.content_paned, text="计算结果") self.content_paned.add(self.result_frame, weight=2) # 构建控制区域 self.build_controls_frame() # 初始化结果表格 self.init_result_table() # 状态栏 self.status_var = tk.StringVar(value="就绪") status_bar = ttk.Label(self.root, textvariable=self.status_var, relief=tk.SUNKEN, anchor=tk.W, font=("SimHei", 9)) status_bar.pack(side=tk.BOTTOM, fill=tk.X) def build_controls_frame(self): # 调整网格布局为3列 self.controls_frame.grid_columnconfigure(0, weight=1) self.controls_frame.grid_columnconfigure(1, weight=1) self.controls_frame.grid_columnconfigure(2, weight=0) self.controls_frame.grid_rowconfigure(0, weight=1) self.controls_frame.grid_rowconfigure(1, weight=1) self.controls_frame.grid_rowconfigure(2, weight=1) self.controls_frame.grid_rowconfigure(3, weight=1) self.controls_frame.grid_rowconfigure(4, weight=1) # 数据文件夹选择 folder_frame = ttk.LabelFrame(self.controls_frame, text="数据文件夹", padding=(2, 1)) folder_frame.grid(row=0, column=0, sticky="ew", padx=2, pady=0) ttk.Label(folder_frame, text="文件夹:").pack(side=tk.LEFT, padx=2) self.folder_path_var = tk.StringVar() ttk.Entry(folder_frame, textvariable=self.folder_path_var, width=30).pack(side=tk.LEFT, padx=2, fill=tk.X) ttk.Button(folder_frame, text="浏览", command=self.browse_folder).pack(side=tk.LEFT, padx=2) # 文件切换 file_frame = ttk.LabelFrame(self.controls_frame, text="文件切换", padding=(2, 1)) file_frame.grid(row=0, column=1, sticky="ew", padx=2, pady=0) ttk.Label(file_frame, text="选择文件:").pack(side=tk.LEFT, padx=2) self.file_combobox = ttk.Combobox(file_frame, state="disabled", width=30) self.file_combobox.pack(side=tk.LEFT, padx=2, fill=tk.X) self.file_combobox.bind("<<ComboboxSelected>>", self.on_file_selected) # 起始列选择框和错误提示控制 start_col_frame = ttk.LabelFrame(self.controls_frame, text="数据设置", padding=(2, 1)) start_col_frame.grid(row=0, column=2, sticky="ew", padx=2, pady=0) # 起始列设置 ttk.Label(start_col_frame, text="起始列:").pack(side=tk.LEFT, padx=2) ttk.Entry(start_col_frame, textvariable=self.start_col, width=5).pack(side=tk.LEFT, padx=2) ttk.Label(start_col_frame, text="(从1开始)").pack(side=tk.LEFT, padx=1) # 错误提示复选框 ttk.Checkbutton(start_col_frame, text="显示错误提示", variable=self.show_errors).pack(side=tk.LEFT, padx=5) # TXT文件设置 txt_frame = ttk.LabelFrame(self.controls_frame, text="TXT文件设置", padding=(2, 1)) txt_frame.grid(row=1, column=0, columnspan=3, sticky="ew", padx=2, pady=0) ttk.Label(txt_frame, text="分隔符:").pack(side=tk.LEFT, padx=5) self.delimiter_combobox = ttk.Combobox(txt_frame, values=["制表符", "逗号", "空格", "其他"], width=8, state="readonly") self.delimiter_combobox.current(0) # 默认选择制表符 self.delimiter_combobox.pack(side=tk.LEFT, padx=2) self.delimiter_combobox.bind("<<ComboboxSelected>>", self.on_delimiter_changed) ttk.Label(txt_frame, text="自定义:").pack(side=tk.LEFT, padx=2) self.custom_delimiter = ttk.Entry(txt_frame, textvariable=self.delimiter, width=3) self.custom_delimiter.pack(side=tk.LEFT, padx=2) ttk.Label(txt_frame, text="(仅当选择'其他'时生效)").pack(side=tk.LEFT, padx=2) # 图表控制(跨3列) chart_ctrl_frame = ttk.LabelFrame(self.controls_frame, text="图表控制", padding=(2, 1)) chart_ctrl_frame.grid(row=2, column=0, columnspan=3, sticky="ew", padx=2, pady=0) ttk.Label(chart_ctrl_frame, text="显示曲线:").pack(side=tk.LEFT, padx=2) # 曲线复选框滚动区域 scroll_frame = ttk.Frame(chart_ctrl_frame) scroll_frame.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=2) canvas_check = tk.Canvas(scroll_frame, height=30) hscroll = ttk.Scrollbar(scroll_frame, orient=tk.HORIZONTAL, command=canvas_check.xview) self.curve_check_frame = ttk.Frame(canvas_check) self.curve_check_frame.bind( "<Configure>", lambda e: canvas_check.configure(scrollregion=canvas_check.bbox("all")) ) canvas_check.create_window((0, 0), window=self.curve_check_frame, anchor=tk.NW) canvas_check.configure(xscrollcommand=hscroll.set) canvas_check.pack(side=tk.TOP, fill=tk.X, expand=True) hscroll.pack(side=tk.BOTTOM, fill=tk.X) ttk.Button(chart_ctrl_frame, text="反选", command=self.invert_selection).pack(side=tk.LEFT, padx=2) # 坐标轴设置(跨3列) axis_frame = ttk.LabelFrame(self.controls_frame, text="坐标轴设置", padding=(2, 1)) axis_frame.grid(row=3, column=0, columnspan=3, sticky="ew", padx=2, pady=0) # X轴范围 ttk.Label(axis_frame, text="X:").pack(side=tk.LEFT, padx=2) self.x_min_var = tk.StringVar() ttk.Entry(axis_frame, textvariable=self.x_min_var, width=10).pack(side=tk.LEFT, padx=1) ttk.Label(axis_frame, text="-").pack(side=tk.LEFT) self.x_max_var = tk.StringVar() ttk.Entry(axis_frame, textvariable=self.x_max_var, width=10).pack(side=tk.LEFT, padx=1) # Y轴范围 ttk.Label(axis_frame, text="Y:").pack(side=tk.LEFT, padx=5) self.y_min_var = tk.StringVar() ttk.Entry(axis_frame, textvariable=self.y_min_var, width=10).pack(side=tk.LEFT, padx=1) ttk.Label(axis_frame, text="-").pack(side=tk.LEFT) self.y_max_var = tk.StringVar() ttk.Entry(axis_frame, textvariable=self.y_max_var, width=10).pack(side=tk.LEFT, padx=1) ttk.Button(axis_frame, text="应用", command=self.apply_axis_limits).pack(side=tk.LEFT, padx=5) ttk.Button(axis_frame, text="重置", command=self.reset_axis_limits).pack(side=tk.LEFT, padx=2) # 计算范围和操作区(跨3列) calc_frame = ttk.LabelFrame(self.controls_frame, text="计算设置", padding=(2, 1)) calc_frame.grid(row=4, column=0, columnspan=3, sticky="ew", padx=2, pady=0) # 计算范围设置 ttk.Label(calc_frame, text="计算范围 X:").pack(side=tk.LEFT, padx=5) ttk.Entry(calc_frame, textvariable=self.calc_start_x, width=10).pack(side=tk.LEFT, padx=1) ttk.Label(calc_frame, text="-").pack(side=tk.LEFT) ttk.Entry(calc_frame, textvariable=self.calc_end_x, width=10).pack(side=tk.LEFT, padx=1) # 衰减值设置 ttk.Label(calc_frame, text="衰减值(dB):").pack(side=tk.LEFT, padx=5) self.db_var = tk.StringVar(value="-0.3") ttk.Entry(calc_frame, textvariable=self.db_var, width=8).pack(side=tk.LEFT, padx=2) # 光标跟踪 ttk.Label(calc_frame, text="跟踪曲线:").pack(side=tk.LEFT, padx=5) self.track_curve_combobox = ttk.Combobox(calc_frame, state="disabled", width=10) self.track_curve_combobox.pack(side=tk.LEFT, padx=2) self.track_curve_combobox.bind("<<ComboboxSelected>>", self.on_track_curve_selected) self.coord_display = ttk.Label(calc_frame, text="X=?, Y=?", width=30) self.coord_display.pack(side=tk.LEFT, padx=5) # 操作按钮 ttk.Button(calc_frame, text="计算", command=self.calculate).pack(side=tk.LEFT, padx=2) ttk.Button(calc_frame, text="导出", command=self.export_results).pack(side=tk.LEFT, padx=2) ttk.Button(calc_frame, text="绘图", command=self.plot_chart).pack(side=tk.LEFT, padx=2) def on_delimiter_changed(self, event): """处理分隔符选择变化""" selection = self.delimiter_combobox.get() if selection == "制表符": self.delimiter.set("\t") self.custom_delimiter.configure(state="disabled") elif selection == "逗号": self.delimiter.set(",") self.custom_delimiter.configure(state="disabled") elif selection == "空格": self.delimiter.set(" ") self.custom_delimiter.configure(state="disabled") elif selection == "其他": self.custom_delimiter.configure(state="enabled") def init_result_table(self): # 创建结果表格 columns = ("curve", "ymax", "target_y", "factor", "x1", "x2", "f0") self.result_tree = ttk.Treeview(self.result_frame, columns=columns, show="headings", height=3) # 设置列标题 self.result_tree.heading("curve", text="曲线编号") self.result_tree.heading("ymax", text="Y最大值") self.result_tree.heading("target_y", text="目标Y值") self.result_tree.heading("factor", text="计算系数(%)") self.result_tree.heading("x1", text="上升段X值") self.result_tree.heading("x2", text="下降段X值") self.result_tree.heading("f0", text="F0 (Hz)") # 调整列宽 self.result_tree.column("curve", width=80, anchor=tk.CENTER) self.result_tree.column("ymax", width=100, anchor=tk.CENTER) self.result_tree.column("target_y", width=100, anchor=tk.CENTER) self.result_tree.column("factor", width=100, anchor=tk.CENTER) self.result_tree.column("x1", width=120, anchor=tk.CENTER) self.result_tree.column("x2", width=120, anchor=tk.CENTER) self.result_tree.column("f0", width=100, anchor=tk.CENTER) # 添加滚动条 vscrollbar = ttk.Scrollbar(self.result_frame, orient=tk.VERTICAL, command=self.result_tree.yview) self.result_tree.configure(yscroll=vscrollbar.set) hscrollbar = ttk.Scrollbar(self.result_frame, orient=tk.HORIZONTAL, command=self.result_tree.xview) self.result_tree.configure(xscroll=hscrollbar.set) # 布局 hscrollbar.pack(side=tk.BOTTOM, fill=tk.X) vscrollbar.pack(side=tk.RIGHT, fill=tk.Y) self.result_tree.pack(fill=tk.BOTH, expand=True) def browse_folder(self): """浏览并选择包含数据文件的文件夹""" folder_path = filedialog.askdirectory(title="选择数据文件夹") if folder_path: self.folder_path_var.set(folder_path) self.load_folder_data(folder_path) def load_folder_data(self, folder_path): """加载文件夹中所有支持格式的文件数据""" try: self.status_var.set(f"正在加载文件夹: {os.path.basename(folder_path)}") self.root.update() # 清空之前的数据 self.file_data = {} self.file_combobox['values'] = [] self.file_combobox['state'] = "disabled" self.clear_chart() self.clear_results() self.track_curve_combobox['values'] = [] self.track_curve_combobox['state'] = "disabled" # 清空计算范围 self.calc_start_x.set("") self.calc_end_x.set("") # 获取文件夹中所有支持格式的文件 supported_files = [] for file in os.listdir(folder_path): if file.endswith(('.xls', '.xlsx', '.txt')): supported_files.append(file) if not supported_files: if self.show_errors.get(): messagebox.showwarning("警告", "所选文件夹中未找到支持的文件(.xls, .xlsx, .txt)") self.status_var.set("就绪") return # 获取起始列设置(转换为0-based索引) start_col = self.start_col.get() - 1 if start_col < 0: if self.show_errors.get(): messagebox.showwarning("警告", "起始列不能小于1,已自动调整为1") start_col = 0 self.start_col.set(1) # 加载每个文件的数据 for file in supported_files: file_path = os.path.join(folder_path, file) try: # 根据文件扩展名选择合适的读取方法和引擎 ext = os.path.splitext(file)[1].lower() if ext == '.xlsx': # .xlsx格式专用openpyxl引擎 data = pd.read_excel(file_path, header=None, engine='openpyxl') elif ext == '.xls': # .xls格式专用xlrd引擎 data = pd.read_excel(file_path, header=None, engine='xlrd') elif ext == '.txt': # 读取TXT文件,使用指定的分隔符 delimiter = self.delimiter.get() # 处理可能的空分隔符 if not delimiter.strip(): delimiter = None # 让pandas自动检测空格分隔 data = pd.read_csv( file_path, header=None, sep=delimiter, engine='python', skip_blank_lines=True ) # 验证数据格式 if len(data) < 2: if self.show_errors.get(): messagebox.showwarning("警告", f"{file} 至少需要包含两行数据,已跳过") continue # 验证起始列是否有效 if start_col >= data.shape[1]: if self.show_errors.get(): messagebox.showwarning("警告", f"{file} 没有足够的列,已使用最后一列作为起始列") start_col = max(0, data.shape[1] - 1) # 第一行为X值(从起始列开始) x_values = data.iloc[0, start_col:].dropna().values # 后续行为曲线数据(从起始列开始) curve_data = {} for i in range(1, len(data)): curve_name = f"曲线{i}" try: y_values = data.iloc[i, start_col:].dropna().values if len(y_values) == len(x_values): curve_data[curve_name] = y_values else: # 数据异常,保留曲线位置但值为None curve_data[curve_name] = None if self.show_errors.get(): messagebox.showwarning("警告", f"{file} 中{curve_name}的Y值数量与X值数量不匹配") except Exception as e: # 数据异常,保留曲线位置但值为None curve_data[curve_name] = None if self.show_errors.get(): messagebox.showwarning("警告", f"{file} 中{curve_name}数据异常: {str(e)}") if curve_data: # 只保存有有效曲线数据的文件 self.file_data[file] = { 'x_values': x_values, 'curve_data': curve_data } except Exception as e: if self.show_errors.get(): messagebox.showwarning("警告", f"加载 {file} 时出错: {str(e)}, 已跳过") except Exception as e: if self.show_errors.get(): messagebox.showerror("错误", f"加载文件夹时出错: {str(e)}") self.status_var.set("加载文件夹失败") return if not self.file_data: if self.show_errors.get(): messagebox.showwarning("警告", "未成功加载任何文件的数据") self.status_var.set("就绪") return # 更新下拉菜单 self.file_combobox['values'] = list(self.file_data.keys()) self.file_combobox['state'] = "readonly" # 自动选择第一个文件 if self.file_data: first_file = next(iter(self.file_data.keys())) self.file_combobox.current(0) self.on_file_selected(None) self.status_var.set(f"已加载 {len(self.file_data)} 个文件的数据(从第{self.start_col.get()}列开始)") if self.show_errors.get(): messagebox.showinfo("成功", f"文件夹加载成功,共加载 {len(self.file_data)} 个文件的数据(从第{self.start_col.get()}列开始)") def on_file_selected(self, event): """当选择不同文件时更新数据和界面""" selected_file = self.file_combobox.get() if not selected_file or selected_file not in self.file_data: return self.current_file = selected_file file_info = self.file_data[selected_file] # 更新当前文件的数据 self.x_values = file_info['x_values'] self.curve_data = file_info['curve_data'] # 包含所有曲线,异常曲线值为None # 清除之前的复选框 for widget in self.curve_check_frame.winfo_children(): widget.destroy() # 创建新的曲线复选框(包含所有曲线,无论是否异常) self.curve_vars = {} for curve_name in self.curve_data.keys(): # 异常曲线默认不勾选 default_state = self.curve_data[curve_name] is not None self.curve_vars[curve_name] = tk.BooleanVar(value=default_state) # 异常曲线显示为灰色 if self.curve_data[curve_name] is None: check = ttk.Checkbutton(self.curve_check_frame, text=curve_name, variable=self.curve_vars[curve_name], state=tk.DISABLED) check.pack(side=tk.LEFT, padx=3, pady=1) # 添加灰色样式 check.configure(style="Disabled.TCheckbutton") self.style.configure("Disabled.TCheckbutton", foreground="gray") else: ttk.Checkbutton(self.curve_check_frame, text=curve_name, variable=self.curve_vars[curve_name]).pack(side=tk.LEFT, padx=3, pady=1) # 更新跟踪曲线下拉菜单(只包含有效曲线) valid_curves = [name for name, data in self.curve_data.items() if data is not None] self.track_curve_combobox['values'] = valid_curves self.track_curve_combobox['state'] = "readonly" if valid_curves else "disabled" if valid_curves: self.track_curve_combobox.current(0) self.selected_curve = valid_curves[0] else: self.selected_curve = None # 清除之前的结果和图表 self.clear_results() self.clear_chart() # 填充轴范围的默认值 if self.x_values is not None and len(self.x_values) > 0: self.x_min_var.set(f"{min(self.x_values):.2f}") self.x_max_var.set(f"{max(self.x_values):.2f}") # 设置默认计算范围为全范围 self.calc_start_x.set(f"{min(self.x_values):.2f}") self.calc_end_x.set(f"{max(self.x_values):.2f}") # 计算Y值的默认范围 valid_y_values = [] for y_values in self.curve_data.values(): if y_values is not None: valid_y_values.extend(y_values) if valid_y_values: self.y_min_var.set(f"{min(valid_y_values):.4f}") self.y_max_var.set(f"{max(valid_y_values):.4f}") # 统计异常曲线数量 total_curves = len(self.curve_data) invalid_curves = sum(1 for data in self.curve_data.values() if data is None) status_text = f"已加载文件: {selected_file},共 {total_curves} 条曲线" if invalid_curves > 0: status_text += f"(其中 {invalid_curves} 条曲线数据异常)" self.status_var.set(status_text) def on_track_curve_selected(self, event): """选择要跟踪的曲线""" self.selected_curve = self.track_curve_combobox.get() if self.ax and self.selected_curve in self.curve_data and self.curve_data[self.selected_curve] is not None: # 重新绘制光标以更新到新选择的曲线 self.set_cursor_behavior() self.status_var.set(f"已选择跟踪曲线: {self.selected_curve}") def invert_selection(self): """反选所有曲线复选框(只对有效曲线生效)""" for curve_name, var in self.curve_vars.items(): if self.curve_data[curve_name] is not None and var.winfo_exists(): # 只处理有效曲线 var.set(not var.get()) self.plot_chart() def apply_axis_limits(self): """应用用户设置的坐标轴范围""" if not self.ax: if self.show_errors.get(): messagebox.showwarning("警告", "请先绘制图表") return try: # 应用X轴范围 if self.x_min_var.get() and self.x_max_var.get(): x_min = float(self.x_min_var.get()) x_max = float(self.x_max_var.get()) if x_min < x_max: self.ax.set_xlim(x_min, x_max) else: if self.show_errors.get(): messagebox.showwarning("警告", "X轴起始点必须小于截止点") return # 应用Y轴范围 if self.y_min_var.get() and self.y_max_var.get(): y_min = float(self.y_min_var.get()) y_max = float(self.y_max_var.get()) if y_min < y_max: self.ax.set_ylim(y_min, y_max) else: if self.show_errors.get(): messagebox.showwarning("警告", "Y轴起始点必须小于截止点") return # 更新计算范围的显示 self.update_range_visualization() self.canvas.draw() self.status_var.set("已应用自定义坐标轴范围") except ValueError: if self.show_errors.get(): messagebox.showerror("错误", "请输入有效的数字作为坐标轴范围") except Exception as e: if self.show_errors.get(): messagebox.showerror("错误", f"应用坐标轴范围时出错: {str(e)}") def reset_axis_limits(self): """重置坐标轴范围为自动计算""" if not self.ax: if self.show_errors.get(): messagebox.showwarning("警告", "请先绘制图表") return try: self.ax.relim() self.ax.autoscale_view() x_min, x_max = self.ax.get_xlim() y_min, y_max = self.ax.get_ylim() self.x_min_var.set(f"{x_min:.2f}") self.x_max_var.set(f"{x_max:.2f}") self.y_min_var.set(f"{y_min:.4f}") self.y_max_var.set(f"{y_max:.4f}") # 更新计算范围的显示 self.update_range_visualization() self.canvas.draw() self.status_var.set("已重置坐标轴范围为自动计算") except Exception as e: if self.show_errors.get(): messagebox.showerror("错误", f"重置坐标轴范围时出错: {str(e)}") def update_range_visualization(self): """在图表上可视化显示计算范围""" if not self.ax or self.x_values is None or len(self.x_values) == 0: return try: # 移除之前的范围矩形 if self.range_rect: self.range_rect.remove() self.range_rect = None # 获取计算范围 start_x = float(self.calc_start_x.get()) end_x = float(self.calc_end_x.get()) if start_x >= end_x: return # 获取Y轴范围 y_min, y_max = self.ax.get_ylim() # 创建半透明矩形表示计算范围 self.range_rect = self.ax.fill_between( [start_x, end_x], y_min, y_max, color='gray', alpha=0.2 ) except (ValueError, TypeError): # 如果计算范围无效,不显示矩形 pass def find_x_for_y_in_range(self, x_values, y_values, target_y, start_x, end_x): """在指定X范围内查找目标Y值对应的两个X值""" # 筛选出在计算范围内的数据点 mask = (x_values >= start_x) & (x_values <= end_x) filtered_x = x_values[mask] filtered_y = y_values[mask] if len(filtered_x) < 2: return None, None # 范围内数据点不足 上升段_x = None 下降段_x = None # 遍历筛选后的点对,寻找上升段 for i in range(len(filtered_y) - 1): y1, y2 = filtered_y[i], filtered_y[i + 1] x1, x2 = filtered_x[i], filtered_x[i + 1] if y1 < target_y < y2: 上升段_x = x1 + (target_y - y1) * (x2 - x1) / (y2 - y1) break # 遍历筛选后的点对,寻找下降段 for i in range(len(filtered_y) - 1): y1, y2 = filtered_y[i], filtered_y[i + 1] x1, x2 = filtered_x[i], filtered_x[i + 1] if y1 > target_y > y2: 下降段_x = x1 + (target_y - y1) * (x2 - x1) / (y2 - y1) break return 上升段_x, 下降段_x def calculate(self): """计算当前选中文件在指定范围内的结果""" if not self.current_file or not self.curve_data: if self.show_errors.get(): messagebox.showwarning("警告", "请先加载数据文件") return try: db_value = float(self.db_var.get()) start_x = float(self.calc_start_x.get()) end_x = float(self.calc_end_x.get()) if start_x >= end_x: if self.show_errors.get(): messagebox.showerror("错误", "起始计算点必须小于截止计算点") return # 检查计算范围是否在数据范围内 if self.x_values is not None and (start_x < min(self.x_values) or end_x > max(self.x_values)): if self.show_errors.get(): messagebox.showwarning("警告", "计算范围超出数据范围,将使用有效部分进行计算") except ValueError: if self.show_errors.get(): messagebox.showerror("错误", "请输入有效的数字作为衰减值和计算范围") return self.clear_results() self.status_var.set( f"正在计算 {self.current_file} 在范围 {start_x} - {end_x} 内的衰减值 {db_value} dB 对应的结果") self.root.update() for curve_name, y_values in self.curve_data.items(): # 处理异常曲线 if y_values is None: self.result_tree.insert("", tk.END, values=(curve_name, "N/A", "N/A", "N/A", "数据异常", "数据异常", "N/A")) continue # 筛选计算范围内的数据点 mask = (self.x_values >= start_x) & (self.x_values <= end_x) filtered_x = self.x_values[mask] filtered_y = y_values[mask] if len(filtered_x) < 2: self.result_tree.insert("", tk.END, values=(curve_name, "N/A", "N/A", "N/A", "范围数据不足", "范围数据不足", "N/A")) continue # 在计算范围内寻找Y最大值 y_max = max(filtered_y) factor = math.pow(10, db_value / 20) target_y = y_max * factor # 在计算范围内查找目标Y值对应的X值 x1, x2 = self.find_x_for_y_in_range(self.x_values, y_values, target_y, start_x, end_x) f0 = None if x1 is not None and x2 is not None: try: f0 = math.sqrt(x1 * x2) except: f0 = "计算错误" y_max_str = f"{y_max:.4f}" target_y_str = f"{target_y:.4f}" factor_percent = factor * 100 factor_str = f"{factor_percent:.2f}%" x1_str = f"{x1:.4f}" if x1 is not None else "未找到" x2_str = f"{x2:.4f}" if x2 is not None else "未找到" f0_str = f"{f0:.4f}" if f0 and f0 != "计算错误" else f0 if f0 else "未找到" self.result_tree.insert("", tk.END, values=(curve_name, y_max_str, target_y_str, factor_str, x1_str, x2_str, f0_str)) self.results.append({ "文件名": self.current_file, "曲线编号": curve_name, "Y最大值": y_max_str, "目标Y值": target_y_str, "计算系数": factor_str, "上升段X值": x1_str, "下降段X值": x2_str, "F0 (Hz)": f0_str, "计算范围X": f"{start_x:.2f}-{end_x:.2f}" }) # 统计计算结果 total = len(self.curve_data) valid = sum(1 for data in self.curve_data.values() if data is not None) self.status_var.set(f"计算完成,共 {total} 条曲线({valid} 条有效)") def export_results(self): """导出当前文件的计算结果""" if not self.results: if self.show_errors.get(): messagebox.showwarning("警告", "没有可导出的结果,请先进行计算") return try: timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") current_filename = os.path.splitext(self.current_file)[0] default_filename = f"{current_filename}_计算结果_{timestamp}" file_path = filedialog.asksaveasfilename( defaultextension=".xlsx", filetypes=[ ("Excel文件", "*.xlsx"), ("CSV文件", "*.csv"), ("文本文件", "*.txt") ], initialfile=default_filename ) if not file_path: return df = pd.DataFrame(self.results) ext = os.path.splitext(file_path)[1].lower() if ext == '.xlsx': with pd.ExcelWriter(file_path, engine='openpyxl') as writer: df.to_excel(writer, index=False) worksheet = writer.sheets['Sheet1'] if '计算系数' in df.columns: col_idx = df.columns.get_loc('计算系数') + 1 for row in worksheet.iter_rows(min_col=col_idx, max_col=col_idx): for cell in row: if cell.value and '%' in str(cell.value): try: value = float(str(cell.value).replace('%', '')) / 100 cell.value = value cell.number_format = '0.00%' except: continue elif ext == '.csv': df.to_csv(file_path, index=False, encoding='utf-8-sig') elif ext == '.txt': # 使用制表符分隔导出TXT文件 df.to_csv(file_path, index=False, sep='\t', encoding='utf-8-sig') self.status_var.set(f"结果已导出至: {file_path}") if self.show_errors.get(): messagebox.showinfo("成功", f"结果已成功导出至:\n{file_path}") except Exception as e: if self.show_errors.get(): messagebox.showerror("错误", f"导出文件时出错: {str(e)}") self.status_var.set("导出文件失败") def clear_chart(self): """清除当前图表""" if self.canvas: self.canvas.get_tk_widget().destroy() self.fig = None self.canvas = None self.ax = None self.lines = [] self.coord_text = None self.range_rect = None self.cursor_line_x = None self.cursor_line_y = None self.cursor_point = None def clear_results(self): """清除当前结果""" for item in self.result_tree.get_children(): self.result_tree.delete(item) self.results = [] def on_hover(self, event): """处理鼠标悬停事件,使光标锁定在选定的曲线上""" if not self.ax or not event.inaxes or event.inaxes != self.ax: return if not self.selected_curve or self.selected_curve not in self.curve_data or self.curve_data[ self.selected_curve] is None: return # 获取当前选中曲线的数据 x_values = self.x_values y_values = self.curve_data[self.selected_curve] # 找到与鼠标X位置最接近的点 x_mouse = event.xdata idx = np.argmin(np.abs(x_values - x_mouse)) x_nearest = x_values[idx] y_nearest = y_values[idx] # 更新坐标显示 self.coord_display.config(text=f"X={x_nearest:.4f}, Y={y_nearest:.4f}") # 更新坐标文本框 if self.coord_text: self.coord_text.set_text(f'X: {x_nearest:.4f}\nY: {y_nearest:.4f}') self.coord_text.set_position((x_nearest, y_nearest)) # 更新光标线位置,使其锁定在曲线上 if self.cursor_line_x: self.cursor_line_x.set_xdata([x_nearest, x_nearest]) if self.cursor_line_y: self.cursor_line_y.set_ydata([y_nearest, y_nearest]) # 更新光标点位置 if self.cursor_point: self.cursor_point.set_data([x_nearest], [y_nearest]) self.canvas.draw_idle() def set_cursor_behavior(self): """设置光标行为,使其锁定在选定的曲线上""" if not self.ax or not self.selected_curve or self.selected_curve not in self.curve_data or self.curve_data[ self.selected_curve] is None: return # 移除现有的光标元素 if self.cursor_line_x: self.cursor_line_x.remove() if self.cursor_line_y: self.cursor_line_y.remove() if self.cursor_point: self.cursor_point.remove() # 创建新的光标线(十字线) self.cursor_line_x = self.ax.axvline(color='red', linewidth=1, linestyle='--', alpha=0.7) self.cursor_line_y = self.ax.axhline(color='red', linewidth=1, linestyle='--', alpha=0.7) # 创建光标点(在曲线上的标记) self.cursor_point, = self.ax.plot([], [], 'ro', markersize=6, alpha=0.8) # 创建坐标文本框 if not self.coord_text: self.coord_text = self.ax.text(0.05, 0.95, '', transform=self.ax.transAxes, verticalalignment='top', bbox=dict(boxstyle='round', facecolor='wheat', alpha=0.5)) else: self.coord_text.set_text('') # 绑定鼠标移动事件 self.canvas.mpl_connect('motion_notify_event', self.on_hover) def plot_chart(self): """绘制图表,包括计算范围的可视化""" if not self.current_file or not self.curve_data: if self.show_errors.get(): messagebox.showwarning("警告", "请先加载数据文件") return self.clear_chart() # 创建图表 self.fig, self.ax = plt.subplots(figsize=(15, 9), dpi=100) # 获取要显示的曲线(只包含有效且被选中的曲线) curves_to_plot = [ curve for curve, var in self.curve_vars.items() if var.get() and self.curve_data[curve] is not None ] if not curves_to_plot and self.show_errors.get(): messagebox.showwarning("警告", "没有可显示的有效曲线,请检查曲线选择") return # 定义颜色列表 colors = ['b', 'g', 'r', 'c', 'm', 'y', 'k', 'tab:orange', 'tab:pink', 'tab:purple', 'tab:brown', 'tab:gray', 'tab:olive', 'tab:cyan'] # 绘制每条曲线 for i, curve_name in enumerate(curves_to_plot): y_values = self.curve_data[curve_name] color = colors[i % len(colors)] linewidth = 3 if curve_name == self.selected_curve else 2 line, = self.ax.plot(self.x_values, y_values, color=color, linewidth=linewidth, marker='', linestyle='-', label=curve_name) self.lines.append(line) # 设置X轴为对数坐标且显示为整数格式 self.ax.set_xscale('log') self.ax.xaxis.set_major_formatter(ticker.StrMethodFormatter('{x:.0f}')) self.ax.xaxis.set_minor_formatter(ticker.StrMethodFormatter('{x:.0f}')) # 应用用户设置的轴范围 try: if self.x_min_var.get() and self.x_max_var.get(): x_min = float(self.x_min_var.get()) x_max = float(self.x_max_var.get()) if x_min < x_max: self.ax.set_xlim(x_min, x_max) if self.y_min_var.get() and self.y_max_var.get(): y_min = float(self.y_min_var.get()) y_max = float(self.y_max_var.get()) if y_min < y_max: self.ax.set_ylim(y_min, y_max) except: pass # 显示计算范围 self.update_range_visualization() # 设置轴标签和标题 self.ax.set_xlabel('X值 (对数坐标)', fontsize=9, labelpad=5) self.ax.set_ylabel('Y值', fontsize=9) self.ax.set_title(f'{self.current_file} 的XY曲线关系图', fontsize=10, pad=10) # 添加网格线 self.ax.grid(True, which='both', linestyle='--', alpha=0.7) # 添加图例 self.ax.legend(loc='upper right', fontsize=8, bbox_to_anchor=(1.08, 1.15), borderaxespad=0.1) # 优化刻度 self.ax.tick_params(axis='x', rotation=0, pad=5, labelsize=9) # 整体布局优化 plt.subplots_adjust( top=0.9, # 顶部留白 bottom=0.18, # 底部留白 left=0.07, right=0.90 # 右侧留白 ) # 将图表嵌入Tkinter并充满空间 self.canvas = FigureCanvasTkAgg(self.fig, master=self.chart_frame) canvas_widget = self.canvas.get_tk_widget() canvas_widget.pack(fill=tk.BOTH, expand=True) self.canvas.draw() # 设置光标和交互行为(锁定到选定曲线) self.set_cursor_behavior() # 显示图表状态 total = len(self.curve_data) plotted = len(curves_to_plot) self.status_var.set(f"已绘制图表,共 {total} 条曲线,显示 {plotted} 条有效曲线") if __name__ == "__main__": root = tk.Tk() app = YToXCalculator(root) root.mainloop() 我让你优化我这个代码,输出优化后的全代码,你给我的什么乱七八糟
最新发布
09-16
# -*- coding: utf-8 -*- import pandas as pd from datetime import datetime import tkinter as tk from tkinter import ttk, filedialog, messagebox, scrolledtext import os import traceback import re from openpyxl import load_workbook from concurrent.futures import ThreadPoolExecutor from tkinter import font as tkfont import psutil import time import threading import math from PIL import Image, ImageTk class EnhancedVersionUpdaterApp: def __init__(self, root): self.root = root self.root.title("Excel批量修改工具 - 专业版") self.root.geometry("1400x950") self.root.configure(bg="#f0f2f5") # 设置主题颜色 self.primary_color = "#4e73df" self.secondary_color = "#858796" self.success_color = "#1cc88a" self.warning_color = "#f6c23e" self.danger_color = "#e74a3b" self.light_color = "#f8f9fc" self.dark_color = "#5a5c69" # 设置全局字体 default_font = tkfont.nametofont("TkDefaultFont") default_font.configure(family="Segoe UI", size=10) self.root.option_add("*Font", default_font) # 使用线程池提高性能 self.executor = ThreadPoolExecutor(max_workers=4) self.running_tasks = 0 # 初始化变量 self.file_path = "" self.old_project = "" self.new_project = "" self.old_date = "" self.new_date = datetime.now().strftime("%Y-%m-%d") self.old_responsible = "" self.new_responsible = "" self.project_updates = [] self.date_updates = [] self.responsible_updates = [] self.doc_checks = [] # 特殊sheet配置 self.special_sheets = { '変更履歴': {'process': False}, 'history': {'process': False}, 'log': {'process': False}, '封面': {'process': True, 'update_time': False}, '表紙': {'process': True, 'update_time': False}, 'cover': {'process': True, 'update_time': False} } # 时间格式正则表达式 self.date_patterns = [ re.compile(r'\d{4}-\d{2}-\d{2}'), # YYYY-MM-DD re.compile(r'\d{4}/\d{2}/\d{2}'), # YYYY/MM/DD re.compile(r'\d{4}年\d{2}月\d{2}日'), # 中文日期 re.compile(r'\d{2}-\d{2}-\d{4}'), # MM-DD-YYYY re.compile(r'\d{2}/\d{2}/\d{4}') # MM/DD/YYYY ] self.create_enhanced_ui() self.setup_style() # 启动内存监控线程 self.memory_usage = 0 self.memory_thread = threading.Thread(target=self.monitor_memory, daemon=True) self.memory_thread.start() def monitor_memory(self): """监控内存使用情况""" while True: process = psutil.Process(os.getpid()) self.memory_usage = process.memory_info().rss / (1024 * 1024) # MB time.sleep(2) def setup_style(self): """设置界面样式""" style = ttk.Style() # 配置主题 style.theme_use('clam') # 通用样式 style.configure('TFrame', background=self.light_color) style.configure('TLabel', background=self.light_color, foreground=self.dark_color) style.configure('TLabelframe', background=self.light_color, relief="flat", borderwidth=0) style.configure('TLabelframe.Label', background=self.light_color, foreground=self.primary_color, font=('Segoe UI', 10, 'bold')) # 按钮样式 style.configure('TButton', background="#f8f9fc", foreground=self.dark_color, borderwidth=1, relief="solid", padding=6, font=('Segoe UI', 9)) style.map('TButton', background=[('active', '#e6e6e6'), ('!active', '#f8f9fc')], foreground=[('active', self.dark_color), ('!active', self.dark_color)]) # 特殊按钮样式 style.configure('Primary.TButton', background=self.primary_color, foreground="white") style.map('Primary.TButton', background=[('active', '#2e59d9'), ('!active', self.primary_color)], foreground=[('active', 'white'), ('!active', 'white')]) style.configure('Success.TButton', background=self.success_color, foreground="white") style.map('Success.TButton', background=[('active', '#17a673'), ('!active', self.success_color)], foreground=[('active', 'white'), ('!active', 'white')]) style.configure('Warning.TButton', background=self.warning_color, foreground="white") style.map('Warning.TButton', background=[('active', '#dda20a'), ('!active', self.warning_color)], foreground=[('active', 'white'), ('!active', 'white')]) # 进度条样式 style.configure("Custom.Horizontal.TProgressbar", thickness=15, troughcolor=self.light_color, background=self.primary_color, lightcolor=self.primary_color, darkcolor=self.primary_color) # Treeview样式 style.configure("Treeview", background="white", foreground=self.dark_color, rowheight=28, fieldbackground="white", font=('Segoe UI', 9)) style.configure("Treeview.Heading", background=self.primary_color, foreground="white", font=('Segoe UI', 10, 'bold'), padding=(5, 5)) style.map("Treeview.Heading", background=[('active', '#2e59d9')]) # Notebook样式 style.configure("TNotebook", background=self.light_color) style.configure("TNotebook.Tab", background=self.light_color, foreground=self.dark_color, padding=(12, 6), font=('Segoe UI', 10, 'bold')) style.map("TNotebook.Tab", background=[('selected', 'white'), ('!selected', self.light_color)], foreground=[('selected', self.primary_color), ('!selected', self.secondary_color)]) def create_enhanced_ui(self): """创建增强版用户界面""" # 主框架 main_frame = ttk.Frame(self.root) main_frame.pack(fill=tk.BOTH, expand=True, padx=15, pady=15) # 标题栏 title_frame = ttk.Frame(main_frame) title_frame.pack(fill=tk.X, pady=(0, 15)) ttk.Label(title_frame, text="Excel批量修改工具", font=('Segoe UI', 18, 'bold'), foreground=self.primary_color).pack(side=tk.LEFT) # 内存指示器 self.memory_label = ttk.Label(title_frame, text="内存使用: 0 MB", font=('Segoe UI', 9)) self.memory_label.pack(side=tk.RIGHT, padx=10) # 文件选择区域 file_frame = ttk.LabelFrame(main_frame, text="文件操作", padding=10) file_frame.pack(fill=tk.X, pady=(0, 15)) file_input_frame = ttk.Frame(file_frame) file_input_frame.pack(fill=tk.X, pady=5) ttk.Label(file_input_frame, text="Excel文件路径:").pack(side=tk.LEFT) self.file_entry = ttk.Entry(file_input_frame, width=70) self.file_entry.pack(side=tk.LEFT, padx=5, fill=tk.X, expand=True) ttk.Button(file_input_frame, text="浏览文件", command=self.browse_file, style='Primary.TButton').pack(side=tk.LEFT) # 状态和进度区域 status_frame = ttk.Frame(file_frame) status_frame.pack(fill=tk.X, pady=(10, 0)) # 进度条 self.progress_var = tk.DoubleVar() self.progress_bar = ttk.Progressbar(status_frame, variable=self.progress_var, maximum=100, style="Custom.Horizontal.TProgressbar") self.progress_bar.pack(side=tk.LEFT, fill=tk.X, expand=True, padx=(0, 10)) # 进度标签 self.progress_label = ttk.Label(status_frame, text="就绪", width=15) self.progress_label.pack(side=tk.RIGHT) # 主内容区域 - 使用Notebook实现标签页 notebook = ttk.Notebook(main_frame) notebook.pack(fill=tk.BOTH, expand=True) # 项目变更标签页 project_tab = ttk.Frame(notebook) notebook.add(project_tab, text="项目编号变更") self.create_project_tab(project_tab) # 时间变更标签页 date_tab = ttk.Frame(notebook) notebook.add(date_tab, text="时间变更") self.create_date_tab(date_tab) # 担当变更标签页 responsible_tab = ttk.Frame(notebook) notebook.add(responsible_tab, text="担当变更") self.create_responsible_tab(responsible_tab) # 文档检查标签页 doc_check_tab = ttk.Frame(notebook) notebook.add(doc_check_tab, text="文档检查") self.create_doc_check_tab(doc_check_tab) # 底部状态栏 status_bar = ttk.Frame(main_frame, relief=tk.SUNKEN, padding=(5, 3)) status_bar.pack(fill=tk.X, pady=(10, 0)) self.status_var = tk.StringVar() self.status_var.set("就绪 - 等待操作") ttk.Label(status_bar, textvariable=self.status_var, anchor=tk.W, font=('Segoe UI', 9)).pack(fill=tk.X) # 定期更新内存使用情况 self.update_memory_display() def update_memory_display(self): """更新内存显示""" self.memory_label.config(text=f"内存使用: {self.memory_usage:.2f} MB") self.root.after(2000, self.update_memory_display) def create_project_tab(self, parent): """创建项目变更标签页""" # 主框架 main_frame = ttk.Frame(parent) main_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) # 输入区域 input_frame = ttk.Frame(main_frame) input_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Label(input_frame, text="原项目编号:").grid(row=0, column=0, sticky="w", padx=5, pady=5) self.old_project_entry = ttk.Entry(input_frame, width=30) self.old_project_entry.grid(row=0, column=1, sticky="w", padx=5, pady=5) ttk.Label(input_frame, text="新项目编号:").grid(row=1, column=0, sticky="w", padx=5, pady=5) self.new_project_entry = ttk.Entry(input_frame, width=30) self.new_project_entry.grid(row=1, column=1, sticky="w", padx=5, pady=5) ttk.Button(input_frame, text="查找项目", command=lambda: self.executor.submit(self.load_project_changes), style='Primary.TButton').grid(row=0, column=2, rowspan=2, padx=10, pady=5) # 表格容器框架 table_frame = ttk.LabelFrame(main_frame, text="变更列表") table_frame.pack(fill=tk.BOTH, expand=True, pady=(0, 10)) # 创建Treeview和滚动条 tree_container = ttk.Frame(table_frame) tree_container.pack(fill=tk.BOTH, expand=True, padx=5, pady=5) tree_scroll_y = ttk.Scrollbar(tree_container) tree_scroll_y.pack(side=tk.RIGHT, fill=tk.Y) tree_scroll_x = ttk.Scrollbar(tree_container, orient=tk.HORIZONTAL) tree_scroll_x.pack(side=tk.BOTTOM, fill=tk.X) columns = ("Sheet", "位置", "原项目", "新项目", "状态") self.project_tree = ttk.Treeview( tree_container, columns=columns, show="headings", height=15, yscrollcommand=tree_scroll_y.set, xscrollcommand=tree_scroll_x.set, selectmode="extended" ) self.project_tree.pack(fill=tk.BOTH, expand=True) # 配置滚动条 tree_scroll_y.config(command=self.project_tree.yview) tree_scroll_x.config(command=self.project_tree.xview) # 配置列 col_widths = {"Sheet": 200, "位置": 100, "原项目": 200, "新项目": 200, "状态": 100} for col in columns: self.project_tree.heading(col, text=col) self.project_tree.column(col, width=col_widths.get(col, 120), anchor="w") # 操作按钮区域 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Button(btn_frame, text="标记更新", command=lambda: self.update_project_status("待更新"), style='Success.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="标记不更新", command=lambda: self.update_project_status("不更新"), style='Warning.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="执行更新", command=lambda: self.executor.submit(self.apply_project_updates), style='Primary.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="导出报告", command=self.export_project_report, style='Primary.TButton').pack(side=tk.RIGHT, padx=5) def create_date_tab(self, parent): """创建时间变更标签页""" # 主框架 main_frame = ttk.Frame(parent) main_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) # 输入区域 input_frame = ttk.Frame(main_frame) input_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Label(input_frame, text="原时间:").grid(row=0, column=0, sticky="w", padx=5, pady=5) self.old_date_entry = ttk.Entry(input_frame, width=30) self.old_date_entry.grid(row=0, column=1, sticky="w", padx=5, pady=5) ttk.Label(input_frame, text="新时间:").grid(row=1, column=0, sticky="w", padx=5, pady=5) self.new_date_entry = ttk.Entry(input_frame, width=30) self.new_date_entry.insert(0, self.new_date) self.new_date_entry.grid(row=1, column=1, sticky="w", padx=5, pady=5) ttk.Button(input_frame, text="查找时间", command=lambda: self.executor.submit(self.load_date_changes), style='Primary.TButton').grid(row=0, column=2, rowspan=2, padx=10, pady=5) # 表格容器框架 table_frame = ttk.LabelFrame(main_frame, text="变更列表") table_frame.pack(fill=tk.BOTH, expand=True, pady=(0, 10)) # 创建Treeview和滚动条 tree_container = ttk.Frame(table_frame) tree_container.pack(fill=tk.BOTH, expand=True, padx=5, pady=5) tree_scroll_y = ttk.Scrollbar(tree_container) tree_scroll_y.pack(side=tk.RIGHT, fill=tk.Y) tree_scroll_x = ttk.Scrollbar(tree_container, orient=tk.HORIZONTAL) tree_scroll_x.pack(side=tk.BOTTOM, fill=tk.X) columns = ("Sheet", "位置", "原时间", "新时间", "状态") self.date_tree = ttk.Treeview( tree_container, columns=columns, show="headings", height=15, yscrollcommand=tree_scroll_y.set, xscrollcommand=tree_scroll_x.set, selectmode="extended" ) self.date_tree.pack(fill=tk.BOTH, expand=True) # 配置滚动条 tree_scroll_y.config(command=self.date_tree.yview) tree_scroll_x.config(command=self.date_tree.xview) # 配置列 col_widths = {"Sheet": 200, "位置": 100, "原时间": 200, "新时间": 200, "状态": 100} for col in columns: self.date_tree.heading(col, text=col) self.date_tree.column(col, width=col_widths.get(col, 120), anchor="w") # 操作按钮区域 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Button(btn_frame, text="标记更新", command=lambda: self.update_date_status("待更新"), style='Success.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="标记不更新", command=lambda: self.update_date_status("不更新"), style='Warning.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="执行更新", command=lambda: self.executor.submit(self.apply_date_updates), style='Primary.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="导出报告", command=self.export_date_report, style='Primary.TButton').pack(side=tk.RIGHT, padx=5) def create_responsible_tab(self, parent): """创建担当变更标签页""" # 主框架 main_frame = ttk.Frame(parent) main_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) # 输入区域 input_frame = ttk.Frame(main_frame) input_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Label(input_frame, text="原担当:").grid(row=0, column=0, sticky="w", padx=5, pady=5) self.old_responsible_entry = ttk.Entry(input_frame, width=30) self.old_responsible_entry.grid(row=0, column=1, sticky="w", padx=5, pady=5) ttk.Label(input_frame, text="新担当:").grid(row=1, column=0, sticky="w", padx=5, pady=5) self.new_responsible_entry = ttk.Entry(input_frame, width=30) self.new_responsible_entry.grid(row=1, column=1, sticky="w", padx=5, pady=5) ttk.Button(input_frame, text="查找担当", command=lambda: self.executor.submit(self.load_responsible_changes), style='Primary.TButton').grid(row=0, column=2, rowspan=2, padx=10, pady=5) # 表格容器框架 table_frame = ttk.LabelFrame(main_frame, text="变更列表") table_frame.pack(fill=tk.BOTH, expand=True, pady=(0, 10)) # 创建Treeview和滚动条 tree_container = ttk.Frame(table_frame) tree_container.pack(fill=tk.BOTH, expand=True, padx=5, pady=5) tree_scroll_y = ttk.Scrollbar(tree_container) tree_scroll_y.pack(side=tk.RIGHT, fill=tk.Y) tree_scroll_x = ttk.Scrollbar(tree_container, orient=tk.HORIZONTAL) tree_scroll_x.pack(side=tk.BOTTOM, fill=tk.X) columns = ("Sheet", "位置", "原担当", "新担当", "状态") self.responsible_tree = ttk.Treeview( tree_container, columns=columns, show="headings", height=15, yscrollcommand=tree_scroll_y.set, xscrollcommand=tree_scroll_x.set, selectmode="extended" ) self.responsible_tree.pack(fill=tk.BOTH, expand=True) # 配置滚动条 tree_scroll_y.config(command=self.responsible_tree.yview) tree_scroll_x.config(command=self.responsible_tree.xview) # 配置列 col_widths = {"Sheet": 200, "位置": 100, "原担当": 200, "新担当": 200, "状态": 100} for col in columns: self.responsible_tree.heading(col, text=col) self.responsible_tree.column(col, width=col_widths.get(col, 120), anchor="w") # 操作按钮区域 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Button(btn_frame, text="标记更新", command=lambda: self.update_responsible_status("待更新"), style='Success.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="标记不更新", command=lambda: self.update_responsible_status("不更新"), style='Warning.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="执行更新", command=lambda: self.executor.submit(self.apply_responsible_updates), style='Primary.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="导出报告", command=self.export_responsible_report, style='Primary.TButton').pack(side=tk.RIGHT, padx=5) def create_doc_check_tab(self, parent): """创建文档检查标签页""" # 主框架 main_frame = ttk.Frame(parent) main_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10) # 按钮区域 btn_frame = ttk.Frame(main_frame) btn_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Button(btn_frame, text="执行文档检查", command=lambda: self.executor.submit(self.perform_document_checks), style='Primary.TButton').pack(side=tk.LEFT, padx=5) ttk.Button(btn_frame, text="导出检查报告", command=self.export_doc_check_report, style='Primary.TButton').pack(side=tk.LEFT, padx=5) # 检查结果区域 result_frame = ttk.LabelFrame(main_frame, text="文档检查结果") result_frame.pack(fill=tk.BOTH, expand=True, pady=(0, 10)) # 创建Treeview和滚动条 tree_container = ttk.Frame(result_frame) tree_container.pack(fill=tk.BOTH, expand=True, padx=5, pady=5) tree_scroll_y = ttk.Scrollbar(tree_container) tree_scroll_y.pack(side=tk.RIGHT, fill=tk.Y) tree_scroll_x = ttk.Scrollbar(tree_container, orient=tk.HORIZONTAL) tree_scroll_x.pack(side=tk.BOTTOM, fill=tk.X) columns = ("检查项", "状态", "详情", "建议") self.doc_tree = ttk.Treeview( tree_container, columns=columns, show="headings", height=15, yscrollcommand=tree_scroll_y.set, xscrollcommand=tree_scroll_x.set ) self.doc_tree.pack(fill=tk.BOTH, expand=True) # 配置滚动条 tree_scroll_y.config(command=self.doc_tree.yview) tree_scroll_x.config(command=self.doc_tree.xview) # 配置列 col_widths = {"检查项": 200, "状态": 80, "详情": 300, "建议": 300} for col in columns: self.doc_tree.heading(col, text=col) self.doc_tree.column(col, width=col_widths.get(col, 150), anchor="w") # 配置标签样式 self.doc_tree.tag_configure("pass", background="#e8f5e9") self.doc_tree.tag_configure("warning", background="#fff8e1") self.doc_tree.tag_configure("error", background="#ffebee") self.doc_tree.tag_configure("info", background="#e3f2fd") def perform_document_checks(self): """执行文档检查""" try: self.running_tasks += 1 self.update_progress(0, "正在检查文档...") self.update_status("正在执行文档检查...") self.file_path = self.file_entry.get() if not self.file_path: messagebox.showerror("错误", "请先选择Excel文件") return if not os.path.exists(self.file_path): messagebox.showerror("错误", "文件不存在!") return self.doc_checks = [] # 1. 文件基本信息检查 file_size = os.path.getsize(self.file_path) / (1024 * 1024) # MB file_ext = os.path.splitext(self.file_path)[1].lower() self.doc_checks.append({ "检查项": "文件格式", "状态": "通过" if file_ext in ['.xlsx', '.xls'] else "错误", "详情": f"文件格式: {file_ext}", "建议": "使用.xlsx格式以获得最佳兼容性" if file_ext != '.xlsx' else "" }) self.doc_checks.append({ "检查项": "文件大小", "状态": "通过" if file_size < 10 else "警告", "详情": f"{file_size:.2f} MB", "建议": "文件过大可能影响性能,建议拆分" if file_size >= 10 else "" }) # 2. 工作簿结构检查 wb = load_workbook(self.file_path, read_only=True) # 工作表数量 sheet_count = len(wb.sheetnames) self.doc_checks.append({ "检查项": "工作表数量", "状态": "通过" if 1 <= sheet_count <= 20 else "警告", "详情": f"{sheet_count} 个工作表", "建议": "工作表数量过多可能导致性能问题" if sheet_count > 20 else "" }) # 隐藏工作表 hidden_sheets = [name for name in wb.sheetnames if wb[name].sheet_state == 'hidden'] self.doc_checks.append({ "检查项": "隐藏工作表", "状态": "警告" if hidden_sheets else "通过", "详情": f"{len(hidden_sheets)} 个隐藏工作表" if hidden_sheets else "无隐藏工作表", "建议": "检查隐藏工作表内容是否必要" if hidden_sheets else "" }) # 工作表保护 protected_sheets = [name for name in wb.sheetnames if wb[name].protection.sheet] self.doc_checks.append({ "检查项": "受保护工作表", "状态": "信息", "详情": f"{len(protected_sheets)} 个受保护工作表" if protected_sheets else "无受保护工作表", "建议": "确保有必要的访问权限" if protected_sheets else "" }) # 3. 内容检查 # 公式错误 formula_errors = {} for sheet_name in wb.sheetnames: sheet = wb[sheet_name] for row in sheet.iter_rows(): for cell in row: if cell.data_type == 'e': # 错误类型 if sheet_name not in formula_errors: formula_errors[sheet_name] = 0 formula_errors[sheet_name] += 1 error_details = ", ".join([f"{k}:{v}" for k, v in formula_errors.items()]) if formula_errors else "无" self.doc_checks.append({ "检查项": "公式错误", "状态": "错误" if formula_errors else "通过", "详情": f"共发现 {sum(formula_errors.values())} 处错误" if formula_errors else "无公式错误", "建议": "修复公式错误以确保数据准确性" if formula_errors else "" }) # 特殊字符 non_printable_chars = [] non_printable_pattern = re.compile(r'[\x00-\x1F\x7F-\x9F]') # 控制字符 for sheet_name in wb.sheetnames: sheet = wb[sheet_name] for row in sheet.iter_rows(): for cell in row: if cell.value and isinstance(cell.value, str) and non_printable_pattern.search(cell.value): non_printable_chars.append(f"{sheet_name}!{cell.coordinate}") self.doc_checks.append({ "检查项": "特殊字符", "状态": "警告" if non_printable_chars else "通过", "详情": f"发现 {len(non_printable_chars)} 个特殊字符" if non_printable_chars else "无特殊字符", "建议": "移除不可打印字符" if non_printable_chars else "" }) # 超链接检查 hyperlinks = {} for sheet_name in wb.sheetnames: sheet = wb[sheet_name] if sheet.hyperlinks: hyperlinks[sheet_name] = len(sheet.hyperlinks) self.doc_checks.append({ "检查项": "超链接", "状态": "信息", "详情": f"共 {sum(hyperlinks.values())} 个超链接" if hyperlinks else "无超链接", "建议": "验证超链接有效性" if hyperlinks else "" }) # 4. 元数据检查 # 文件属性 created_time = datetime.fromtimestamp(os.path.getctime(self.file_path)) modified_time = datetime.fromtimestamp(os.path.getmtime(self.file_path)) self.doc_checks.append({ "检查项": "文件属性", "状态": "信息", "详情": f"创建: {created_time.strftime('%Y-%m-%d')}, 修改: {modified_time.strftime('%Y-%m-%d')}", "建议": "" }) # 显示结果 self._display_doc_checks() self.update_progress(100, "检查完成") self.update_status(f"文档检查完成,共 {len(self.doc_checks)} 项检查") except Exception as e: messagebox.showerror("错误", f"文档检查失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 def _display_doc_checks(self): """显示文档检查结果""" self.doc_tree.delete(*self.doc_tree.get_children()) for check in self.doc_checks: status = check["状态"] tags = ("pass",) if status == "通过" else ("warning",) if status == "警告" else ("error",) if status == "错误" else ("info",) self.doc_tree.insert("", "end", values=(check["检查项"], check["状态"], check["详情"], check["建议"]), tags=tags) def browse_file(self): """浏览文件""" file_path = filedialog.askopenfilename( filetypes=[("Excel文件", "*.xlsx *.xls"), ("所有文件", "*.*")], title="选择Excel文件" ) if file_path: self.file_entry.delete(0, tk.END) self.file_entry.insert(0, file_path) self.file_path = file_path self.update_status(f"已选择文件: {os.path.basename(file_path)}") def update_status(self, message): """更新状态栏""" self.status_var.set(message) self.root.update_idletasks() def update_progress(self, value, message=None): """更新进度条""" self.progress_var.set(value) if message: self.progress_label.config(text=message) self.root.update_idletasks() def load_project_changes(self): """加载项目编号变更""" try: self.running_tasks += 1 self.update_progress(0, "正在查找项目编号...") self.update_status("正在查找项目编号变更...") self.file_path = self.file_entry.get() self.old_project = self.old_project_entry.get().strip() self.new_project = self.new_project_entry.get().strip() if not all([self.file_path, self.old_project, self.new_project]): messagebox.showerror("错误", "请填写所有必填字段") return if not os.path.exists(self.file_path): messagebox.showerror("错误", "文件不存在!") return # 使用只读模式加载工作簿以节省内存 wb = load_workbook(self.file_path, read_only=True) total_sheets = len(wb.sheetnames) self.project_updates = [] for i, sheet_name in enumerate(wb.sheetnames): self.update_progress((i+1)/total_sheets*100, f"处理: {sheet_name}") sheet_config = self._get_sheet_config(sheet_name) if not sheet_config['process']: continue sheet = wb[sheet_name] # 分块处理大型工作表 for row_chunk in self._chunked_rows(sheet, chunk_size=100): for cell in row_chunk: if cell.value and self.old_project in str(cell.value): self.project_updates.append({ "sheet": sheet_name, "cell": cell.coordinate, "old_value": cell.value, "new_value": cell.value.replace(self.old_project, self.new_project), "status": "待审核" }) if not self.project_updates: messagebox.showinfo("提示", "未找到匹配的项目编号") return self._display_data(self.project_tree, self.project_updates) self.update_status(f"找到 {len(self.project_updates)} 处项目编号需要更新") self.update_progress(100, "查找完成") except Exception as e: messagebox.showerror("错误", f"加载失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 def load_date_changes(self): """加载时间变更""" try: self.running_tasks += 1 self.update_progress(0, "正在查找时间...") self.update_status("正在查找时间变更...") self.file_path = self.file_entry.get() self.old_date = self.old_date_entry.get().strip() self.new_date = self.new_date_entry.get().strip() if not all([self.file_path, self.old_date, self.new_date]): messagebox.showerror("错误", "请填写所有必填字段") return if not os.path.exists(self.file_path): messagebox.showerror("错误", "文件不存在!") return # 使用只读模式加载工作簿以节省内存 wb = load_workbook(self.file_path, read_only=True) total_sheets = len(wb.sheetnames) self.project_updates = [] for i, sheet_name in enumerate(wb.sheetnames): self.update_progress((i+1)/total_sheets*100, f"处理: {sheet_name}") sheet_config = self._get_sheet_config(sheet_name) if not sheet_config['process']: continue sheet = wb[sheet_name] # 分块处理大型工作表 for row_chunk in self._chunked_rows(sheet, chunk_size=100): for cell in row_chunk: if cell.value and self.old_project in str(cell.value): self.project_updates.append({ "sheet": sheet_name, "cell": cell.coordinate, "old_value": cell.value, "new_value": cell.value.replace(self.old_project, self.new_project), "status": "待审核" }) if not self.date_updates: messagebox.showinfo("提示", "未找到匹配的时间") return self._display_data(self.date_tree, self.date_updates) self.update_status(f"找到 {len(self.date_updates)} 处时间需要更新") self.update_progress(100, "查找完成") except Exception as e: messagebox.showerror("错误", f"加载失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 def load_responsible_changes(self): """加载担当变更""" try: self.running_tasks += 1 self.update_progress(0, "正在查找担当...") self.update_status("正在查找担当变更...") self.file_path = self.file_entry.get() self.old_responsible = self.old_responsible_entry.get().strip() self.new_responsible = self.new_responsible_entry.get().strip() if not all([self.file_path, self.old_responsible, self.new_responsible]): messagebox.showerror("错误", "请填写所有必填字段") return if not os.path.exists(self.file_path): messagebox.showerror("错误", "文件不存在!") return # 使用只读模式加载工作簿以节省内存 wb = load_workbook(self.file_path, read_only=True) total_sheets = len(wb.sheetnames) self.project_updates = [] for i, sheet_name in enumerate(wb.sheetnames): self.update_progress((i+1)/total_sheets*100, f"处理: {sheet_name}") sheet_config = self._get_sheet_config(sheet_name) if not sheet_config['process']: continue sheet = wb[sheet_name] # 分块处理大型工作表 for row_chunk in self._chunked_rows(sheet, chunk_size=100): for cell in row_chunk: if cell.value and self.old_project in str(cell.value): self.project_updates.append({ "sheet": sheet_name, "cell": cell.coordinate, "old_value": cell.value, "new_value": cell.value.replace(self.old_project, self.new_project), "status": "待审核" }) if not self.responsible_updates: messagebox.showinfo("提示", "未找到匹配的担当") return self._display_data(self.responsible_tree, self.responsible_updates) self.update_status(f"找到 {len(self.responsible_updates)} 处担当需要更新") self.update_progress(100, "查找完成") except Exception as e: messagebox.showerror("错误", f"加载失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 def _display_data(self, treeview, data): """显示数据到表格""" treeview.delete(*treeview.get_children()) for update in data: tags = ("to_update",) if update["status"] == "待更新" else ("no_update",) if update["status"] == "不更新" else () treeview.insert("", "end", values=(update["sheet"], update["cell"], update["old_value"], update["new_value"], update["status"]), tags=tags) treeview.tag_configure("to_update", background="lightyellow") treeview.tag_configure("no_update", background="lightgray") # 自动滚动到顶部 treeview.yview_moveto(0) def _detect_changes(self, target_value, new_value, value_type): """通用变更检测方法""" updates = [] try: wb = load_workbook(self.file_path) total_sheets = len(wb.sheetnames) for i, sheet_name in enumerate(wb.sheetnames): self.update_progress((i+1)/total_sheets*100, f"正在处理工作表: {sheet_name}") sheet_config = self._get_sheet_config(sheet_name) if not sheet_config['process']: continue if value_type == "date" and not sheet_config.get('update_time', True): continue sheet = wb[sheet_name] if value_type == "date": cells = self._find_date_cells(sheet, target_value) else: cells = self._find_cells_with_value(sheet, target_value) for cell in cells: updates.append({ "sheet": sheet_name, "cell": cell.coordinate, "old_value": cell.value, "new_value": new_value, "status": "待审核" }) except Exception as e: print(f"DEBUG - 读取错误: {str(e)}") return updates def update_project_status(self, status): """更新项目变更状态""" self._update_status(self.project_tree, self.project_updates, status) def update_date_status(self, status): """更新时间变更状态""" self._update_status(self.date_tree, self.date_updates, status) def update_responsible_status(self, status): """更新担当变更状态""" self._update_status(self.responsible_tree, self.responsible_updates, status) def _update_status(self, treeview, data, status): """通用状态更新方法""" selected = treeview.selection() if not selected: messagebox.showwarning("警告", "请先选择记录") return for item in selected: index = treeview.index(item) data[index]["status"] = status self._display_data(treeview, data) def apply_project_updates(self): """执行项目编号变更""" self._apply_updates( updates=self.project_updates, success_message="项目编号更新完成!", treeview=self.project_tree ) def apply_date_updates(self): """执行时间变更""" self._apply_updates( updates=self.date_updates, success_message="时间更新完成!", treeview=self.date_tree ) def apply_responsible_updates(self): """执行担当变更""" self._apply_updates( updates=self.responsible_updates, success_message="担当更新完成!", treeview=self.responsible_tree ) def _apply_updates(self, updates, success_message, treeview): """通用更新应用方法""" if not any(u["status"] == "待更新" for u in updates): messagebox.showwarning("警告", "没有标记为'待更新'的记录") return try: self.running_tasks += 1 self.update_progress(0, "正在更新...") wb = load_workbook(self.file_path) total_updates = len([u for u in updates if u["status"] == "待更新"]) processed = 0 for update in [u for u in updates if u["status"] == "待更新"]: sheet = wb[update["sheet"]] sheet[update["cell"]] = update["new_value"] processed += 1 self.update_progress(processed/total_updates*100, f"正在更新 {update['sheet']} {update['cell']}") wb.save(self.file_path) messagebox.showinfo("成功", success_message) # 重新加载数据 if treeview == self.project_tree: self.load_project_changes() elif treeview == self.date_tree: self.load_date_changes() else: self.load_responsible_changes() except Exception as e: messagebox.showerror("错误", f"更新失败: {str(e)}") print(traceback.format_exc()) finally: self.running_tasks -= 1 self.update_progress(100, "更新完成") def export_project_report(self): """导出项目变更报告""" self._export_report(self.project_updates, "项目变更报告") def export_date_report(self): """导出时间变更报告""" self._export_report(self.date_updates, "时间变更报告") def export_responsible_report(self): """导出担当变更报告""" self._export_report(self.responsible_updates, "担当变更报告") def _export_report(self, data, report_name): """通用报告导出方法""" if not data: messagebox.showwarning("警告", f"没有可导出的{report_name}数据") return try: file_path = filedialog.asksaveasfilename( defaultextension=".xlsx", filetypes=[("Excel文件", "*.xlsx")], title=f"保存{report_name}" ) if not file_path: return df = pd.DataFrame([{ "工作表": item["sheet"], "单元格位置": item["cell"], "原内容": item["old_value"], "新内容": item["new_value"], "状态": item["status"] } for item in data]) df.to_excel(file_path, index=False) messagebox.showinfo("成功", f"{report_name}已导出到: {file_path}") except Exception as e: messagebox.showerror("错误", f"导出失败: {str(e)}") def export_doc_check_report(self): """导出文档检查报告""" if not self.doc_checks: messagebox.showwarning("警告", "没有可导出的文档检查数据") return try: file_path = filedialog.asksaveasfilename( defaultextension=".xlsx", filetypes=[("Excel文件", "*.xlsx")], title="保存文档检查报告" ) if not file_path: return df = pd.DataFrame(self.doc_checks) df.to_excel(file_path, index=False) messagebox.showinfo("成功", f"文档检查报告已导出到: {file_path}") except Exception as e: messagebox.showerror("错误", f"导出失败: {str(e)}") def on_closing(self): """关闭窗口时的处理""" if self.running_tasks > 0: if messagebox.askokcancel("警告", "有任务正在运行,确定要退出吗?"): self.executor.shutdown(wait=False) self.root.destroy() else: self.root.destroy() def _get_sheet_config(self, sheet_name): """获取sheet配置""" sheet_lower = sheet_name.lower() for kw in self.special_sheets: if kw.lower() in sheet_lower: config = self.special_sheets[kw].copy() config['is_special'] = True return config return {'process': True, 'update_time': True, 'is_special': False} def _find_cells_with_value(self, sheet, target_value): """查找包含目标值的所有单元格""" found_cells = [] pattern = re.compile(rf'.*{re.escape(str(target_value))}.*', re.IGNORECASE) for row in sheet.iter_rows(): for cell in row: if cell.value and pattern.search(str(cell.value)): found_cells.append(cell) return found_cells def _find_date_cells(self, sheet, target_date=None): """查找所有包含日期的单元格""" date_cells = [] for row in sheet.iter_rows(): for cell in row: if cell.value and self._is_date(cell.value): if target_date: try: cell_date = pd.to_datetime(cell.value).strftime('%Y-%m-%d') if target_date in str(cell_date): date_cells.append(cell) except: if target_date in str(cell.value): date_cells.append(cell) else: date_cells.append(cell) return date_cells def _is_date(self, value): """判断值是否为日期""" try: pd.to_datetime(value) return True except: # 检查是否是字符串形式的日期 if isinstance(value, str): for pattern in self.date_patterns: if re.fullmatch(pattern, value.strip()): return True return False def on_closing(self): """关闭窗口时的处理""" if self.running_tasks > 0: if messagebox.askokcancel("警告", "有任务正在运行,确定要退出吗?"): self.executor.shutdown(wait=False) self.root.destroy() else: self.root.destroy() if __name__ == "__main__": root = tk.Tk() app = EnhancedVersionUpdaterApp(root) root.protocol("WM_DELETE_WINDOW", app.on_closing) root.mainloop() 这是我按照你的要求更改后的代码,但是其中有一些错误,例如AttributeError: 'EnhancedVersionUpdaterApp' object has no attribute 'memory_usage'. Did you mean: 'memory_label'?,请检查类似的问题,并修改完成给我
08-14
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值