import tkinter as tk
from tkinter import ttk, filedialog, messagebox
import pandas as pd
from openpyxl import load_workbook
from openpyxl.styles import Font
import os
class AdvancedExcelComparator:
def __init__(self, root):
self.root = root
self.root.title("树懒君产品报价数据比对工具")
self.root.geometry("600x400")
# 初始化变量
self.base_file = tk.StringVar()
self.target_file = tk.StringVar()
self.base_sheet = tk.StringVar(value="基础数据")
self.target_sheet = tk.StringVar(value="目标数据")
self.status_text = tk.StringVar(value="准备就绪")
# 创建界面组件
self.create_widgets()
def create_widgets(self):
# 基础文件选择区域
base_frame = ttk.LabelFrame(self.root, text="基础数据文件")
base_frame.pack(pady=5, padx=10, fill="x")
ttk.Label(base_frame, text="文件路径:").grid(row=0, column=0, padx=5)
ttk.Entry(base_frame, textvariable=self.base_file, width=50).grid(row=0, column=1)
ttk.Button(base_frame, text="浏览", command=lambda: self.select_file(self.base_file)).grid(row=0, column=2, padx=5)
# 目标文件选择区域
target_frame = ttk.LabelFrame(self.root, text="目标数据文件")
target_frame.pack(pady=5, padx=10, fill="x")
ttk.Label(target_frame, text="文件路径:").grid(row=0, column=0, padx=5)
ttk.Entry(target_frame, textvariable=self.target_file, width=50).grid(row=0, column=1)
ttk.Button(target_frame, text="浏览", command=lambda: self.select_file(self.target_file)).grid(row=0, column=2, padx=5)
# 参数设置区域
param_frame = ttk.LabelFrame(self.root, text="工作表设置")
param_frame.pack(pady=10, padx=10, fill="x")
ttk.Label(param_frame, text="基础表名称:").grid(row=0, column=0, padx=5)
ttk.Entry(param_frame, textvariable=self.base_sheet).grid(row=0, column=1, sticky="ew")
ttk.Label(param_frame, text="目标表名称:").grid(row=1, column=0, padx=5)
ttk.Entry(param_frame, textvariable=self.target_sheet).grid(row=1, column=1, sticky="ew")
# 操作按钮
btn_frame = ttk.Frame(self.root)
btn_frame.pack(pady=10)
ttk.Button(btn_frame, text="开始处理", command=self.process_data).pack(side="left", padx=5)
ttk.Button(btn_frame, text="清除记录", command=self.reset_all).pack(side="left", padx=5)
# 状态栏
status_frame = ttk.Frame(self.root)
status_frame.pack(fill="x", padx=10, pady=5)
ttk.Label(status_frame, textvariable=self.status_text).pack(side="left")
def select_file(self, target_var):
file_path = filedialog.askopenfilename(
filetypes=[("Excel文件", "*.xlsx"), ("所有文件", "*.*")]
)
if file_path:
target_var.set(file_path)
self.status_text.set(f"已选择文件: {os.path.basename(file_path)}")
def reset_all(self):
self.base_file.set("")
self.target_file.set("")
self.base_sheet.set("基础数据")
self.target_sheet.set("目标数据")
self.status_text.set("状态已重置")
def process_data(self):
if not all([self.base_file.get(), self.target_file.get()]):
messagebox.showerror("错误", "请先选择基础文件和目标文件!")
return
try:
# 读取基础数据
base_df = pd.read_excel(
self.base_file.get(),
sheet_name=self.base_sheet.get(),
usecols=["产品型号", "目录单价(RMB)"],
dtype={'产品型号': 'str', '目录单价(RMB)': 'float'}
).dropna(subset=['产品型号', '目录单价(RMB)'])
price_map = base_df.set_index('产品型号')['目录单价(RMB)'].to_dict()
# 处理目标文件
wb = load_workbook(self.target_file.get())
if self.target_sheet.get() not in wb.sheetnames:
raise ValueError(f"目标表 {self.target_sheet.get()} 不存在")
target_ws = wb[self.target_sheet.get()]
red_font = Font(color="FF0000", bold=True)
# 智能定位列索引
header_row = next(target_ws.iter_rows(min_row=2, max_row=2)) # 假设标题在第二行
headers = [cell.value for cell in header_row]
try:
mpn_idx = headers.index("产品型号")
price_idx = headers.index("目录单价(RMB)")
except ValueError as e:
missing = str(e).split("'")[1]
raise ValueError(f"缺少必要列: {missing}")
# 处理数据行
modified_count = 0
for row in target_ws.iter_rows(min_row=3): # 从第三行开始
mpn_cell = row[mpn_idx]
price_cell = row[price_idx]
if not mpn_cell.value or not price_cell.value:
continue
current_mpn = str(mpn_cell.value).strip()
try:
current_price = float(price_cell.value)
expected_price = price_map.get(current_mpn)
except (ValueError, TypeError):
continue
if expected_price is None:
continue # 基础表中不存在该产品型号
if abs(current_price - expected_price) >= 0.01: # 允许1分钱误差
price_cell.font = red_font
modified_count += 1
# 另存为处理后的文件
save_path = filedialog.asksaveasfilename(
defaultextension=".xlsx",
filetypes=[("Excel文件", "*.xlsx")],
initialfile=f"processed_{os.path.basename(self.target_file.get())}"
)
if save_path:
wb.save(save_path)
messagebox.showinfo("完成",
f"成功处理!\n"
f"标记差异数: {modified_count}\n"
f"保存路径: {save_path}")
self.status_text.set(f"处理完成,保存至: {os.path.basename(save_path)}")
except Exception as e:
messagebox.showerror("处理错误",
f"错误类型: {type(e).__name__}\n"
f"错误详情: {str(e)}")
self.status_text.set("处理过程中发生错误")
raise
if __name__ == "__main__":
root = tk.Tk()
app = AdvancedExcelComparator(root)
root.mainloop()