import pandas as pd
import tkinter as tk
from tkinter import ttk, messagebox, filedialog
import os
import json
import openpyxl
from openpyxl.utils.dataframe import dataframe_to_rows
from tkinter.font import Font
import traceback
import textwrap
class ScrollableFrame(ttk.Frame):
"""自定义可滚动框架实现"""
def __init__(self, parent, *args, **kwargs):
super().__init__(parent, *args, **kwargs)
# 创建Canvas和滚动条
self.canvas = tk.Canvas(self)
self.scrollbar = ttk.Scrollbar(self, orient="vertical", command=self.canvas.yview)
self.scrollable_frame = ttk.Frame(self.canvas)
# 配置Canvas
self.canvas.configure(yscrollcommand=self.scrollbar.set)
self.canvas_frame = self.canvas.create_window((0, 0), window=self.scrollable_frame, anchor="nw")
# 布局
self.canvas.pack(side="left", fill="both", expand=True)
self.scrollbar.pack(side="right", fill="y")
# 绑定事件
self.scrollable_frame.bind("<Configure>", self.on_frame_configure)
self.canvas.bind("<Configure>", self.on_canvas_configure)
self.canvas.bind_all("<MouseWheel>", self.on_mousewheel)
def on_frame_configure(self, event):
"""当内部框架大小改变时更新滚动区域"""
self.canvas.configure(scrollregion=self.canvas.bbox("all"))
def on_canvas_configure(self, event):
"""当Canvas大小改变时调整内部框架宽度"""
self.canvas.itemconfig(self.canvas_frame, width=event.width)
def on_mousewheel(self, event):
"""鼠标滚轮滚动支持"""
self.canvas.yview_scroll(int(-1*(event.delta/120)), "units")
class ExcelControlPanel:
def __init__(self, master):
self.master = master
master.title("功能点确认系统")
master.geometry("1280x800")
master.configure(bg="#f0f2f5")
# 设置全局样式
self.set_styles()
# 加载配置
self.config = self.load_config()
# 初始化多列确认配置
self.confirmation_columns = self.config.get("confirmation_columns", [])
self.column_order = self.config.get("column_order", [])
# 创建界面元素
self.create_widgets()
# 初始化数据
self.excel_path = ""
self.df = None
self.check_states = {} # 存储每个功能点的复选框状态
self.current_sheet = ""
self.header_row = 0
# 启用列拖动
self.enable_column_dragging()
def set_styles(self):
"""设置全局样式和字体"""
style = ttk.Style()
style.theme_use('clam')
# 自定义字体
self.title_font = Font(family="Microsoft YaHei", size=16, weight="bold")
self.subtitle_font = Font(family="Microsoft YaHei", size=10)
self.normal_font = Font(family="Microsoft YaHei", size=9) # 减小字体大小
# 配置样式
style.configure("TFrame", background="#f0f2f5")
style.configure("TLabel", font=self.normal_font, background="#f0f2f5", foreground="#333")
style.configure("TButton", font=self.normal_font, padding=8)
# Treeview样式 - 增加行高
style.configure("Treeview.Heading", font=self.subtitle_font, background="#4a76b5", foreground="white")
style.configure("Treeview", font=self.normal_font, rowheight=40, background="white", fieldbackground="white") # 增加行高
style.configure("Status.TFrame", background="#e0e0e0")
style.configure("Status.TLabel",
font=self.normal_font,
background="#4a76b5",
foreground="#ffffff",
padding=5)
style.configure("Card.TFrame", background="white", borderwidth=0, relief="solid",
padding=10, bordercolor="#e1e4e8", borderradius=8)
style.configure("Card.TLabelframe", background="white", borderwidth=1, relief="solid",
padding=10, bordercolor="#e1e4e8", borderradius=8)
style.configure("Card.TLabelframe.Label", font=self.subtitle_font, foreground="#2c3e50",
background="white")
# 按钮样式
style.map("Primary.TButton",
background=[("active", "#3a66a5"), ("pressed", "#2a5685")],
foreground=[("active", "white"), ("pressed", "white")])
style.configure("Primary.TButton", background="#4a76b5", foreground="white",
font=self.subtitle_font, borderwidth=0, borderradius=4)
style.map("Success.TButton",
background=[("active", "#28a745"), ("pressed", "#218838")],
foreground=[("active", "white"), ("pressed", "white")])
style.configure("Success.TButton", background="#28a745", foreground="white",
font=self.subtitle_font, borderwidth=0, borderradius=4)
style.map("Danger.TButton",
background=[("active", "#dc3545"), ("pressed", "#c82333")],
foreground=[("active", "white"), ("pressed", "white")])
style.configure("Danger.TButton", background="#dc3545", foreground="white",
font=self.subtitle_font, borderwidth=0, borderradius=4)
# 输入框样式
style.configure("Custom.TEntry", fieldbackground="#f8f9fa", bordercolor="#ced4da")
def load_config(self):
"""加载配置文件"""
config_path = "excel_config.json"
default_config = {
"id_col": "No.",
"desc_col": "レビュー観点(CHN)",
"status_col": "レビュー結果",
"sheet_name": "",
"header_row": 9,
"last_dir": os.getcwd(),
"custom_status": "OK",
"confirmation_columns": [],
"column_order": ["selection", "id", "description", "status"]
}
if os.path.exists(config_path):
try:
with open(config_path, 'r', encoding='utf-8') as f:
config = json.load(f)
# 确保所有键都存在
for key in default_config:
if key not in config:
config[key] = default_config[key]
return config
except Exception as e:
return default_config
return default_config
def save_config(self):
"""保存配置文件"""
config_path = "excel_config.json"
try:
with open(config_path, 'w', encoding='utf-8') as f:
json.dump(self.config, f, ensure_ascii=False, indent=2)
except Exception:
pass
def create_widgets(self):
"""创建现代化界面元素"""
# 主容器
main_container = ttk.Frame(self.master, style="Card.TFrame")
main_container.pack(fill="both", expand=True, padx=20, pady=20)
# 标题栏
title_frame = ttk.Frame(main_container, style="Title.TFrame")
title_frame.pack(fill="x", pady=(0, 20))
# 添加内置图标
try:
# 创建一个简单的内置图标
self.icon_img = tk.PhotoImage(width=8, height=8)
self.icon_img.put("#4a76b5", (0, 0, 8, 8))
icon_label = ttk.Label(title_frame, image=self.icon_img, background="#4a76b5")
icon_label.pack(side="left", padx=(10, 5))
except Exception:
pass
ttk.Label(title_frame, text="功能点确认系统", style="Title.TLabel").pack(side="left", padx=10)
# 主内容区域（卡片式布局）
content_frame = ttk.Frame(main_container)
content_frame.pack(fill="both", expand=True)
# 左侧控制面板（卡片） - 可滚动
control_container = ttk.Frame(content_frame, width=350)
control_container.pack(side="left", fill="y", padx=(0, 20))
# 创建自定义可滚动框架
scrollable_frame = ScrollableFrame(control_container, width=350)
scrollable_frame.pack(fill="both", expand=True)
# 获取内部框架
inner_frame = scrollable_frame.scrollable_frame
# 美化控制面板
control_card = ttk.LabelFrame(inner_frame, text="控制面板", style="Card.TLabelframe")
control_card.pack(fill="both", expand=True, padx=10, pady=10, ipadx=5, ipady=5)
# 文件选择区域
file_frame = ttk.LabelFrame(control_card, text="Excel文件设置")
file_frame.pack(fill="x", padx=10, pady=(0, 15))
ttk.Label(file_frame, text="Excel文件路径:").pack(anchor="w", pady=(0, 5), padx=10)
path_frame = ttk.Frame(file_frame)
path_frame.pack(fill="x", pady=5, padx=10)
self.path_entry = ttk.Entry(path_frame, width=30, style="Custom.TEntry")
self.path_entry.pack(side="left", fill="x", expand=True, padx=(0, 5))
ttk.Button(path_frame, text="浏览", command=self.browse_file, width=8).pack(side="left")
ttk.Button(file_frame, text="加载数据", command=self.load_data, style="Primary.TButton").pack(fill="x", pady=10, padx=10)
# Sheet选择区域
sheet_frame = ttk.LabelFrame(control_card, text="工作表设置")
sheet_frame.pack(fill="x", padx=10, pady=(0, 15))
ttk.Label(sheet_frame, text="当前Sheet:").pack(anchor="w", padx=10, pady=(10, 0))
self.sheet_var = tk.StringVar(value=self.config.get("sheet_name", "未选择"))
sheet_display = ttk.Label(sheet_frame, textvariable=self.sheet_var, font=self.subtitle_font)
sheet_display.pack(anchor="w", pady=(0, 10), padx=10)
ttk.Button(sheet_frame, text="选择工作表", command=self.select_sheet, style="Primary.TButton").pack(fill="x", padx=10, pady=(0, 10))
# 表头行设置
header_frame = ttk.LabelFrame(control_card, text="表头设置")
header_frame.pack(fill="x", padx=10, pady=(0, 15))
ttk.Label(header_frame, text="表头所在行号:").pack(anchor="w", padx=10, pady=(10, 0))
self.header_var = tk.IntVar(value=self.config.get("header_row", 9))
ttk.Entry(header_frame, textvariable=self.header_var, width=10, style="Custom.TEntry").pack(anchor="w", pady=5, padx=10)
# 列名配置区域
col_frame = ttk.LabelFrame(control_card, text="列名配置")
col_frame.pack(fill="x", padx=10, pady=(0, 15))
# ID列
ttk.Label(col_frame, text="ID列名:").pack(anchor="w", padx=10, pady=(10, 0))
self.id_col_var = tk.StringVar(value=self.config["id_col"])
ttk.Entry(col_frame, textvariable=self.id_col_var, style="Custom.TEntry").pack(fill="x", pady=(0, 10), padx=10)
# 功能点列
ttk.Label(col_frame, text="功能点列名:").pack(anchor="w", padx=10, pady=(0, 0))
self.desc_col_var = tk.StringVar(value=self.config["desc_col"])
ttk.Entry(col_frame, textvariable=self.desc_col_var, style="Custom.TEntry").pack(fill="x", pady=(0, 10), padx=10)
# 状态列
ttk.Label(col_frame, text="状态列名:").pack(anchor="w", padx=10, pady=(0, 0))
self.status_col_var = tk.StringVar(value=self.config["status_col"])
ttk.Entry(col_frame, textvariable=self.status_col_var, style="Custom.TEntry").pack(fill="x", pady=(0, 10), padx=10)
# 自定义状态
ttk.Label(col_frame, text="自定义确认状态:").pack(anchor="w", padx=10, pady=(0, 0))
self.custom_status_var = tk.StringVar(value=self.config.get("custom_status", "OK"))
ttk.Entry(col_frame, textvariable=self.custom_status_var, style="Custom.TEntry").pack(fill="x", pady=(0, 10), padx=10)
# 多列确认设置
multi_col_frame = ttk.LabelFrame(control_card, text="多列确认设置")
multi_col_frame.pack(fill="x", padx=10, pady=(0, 15))
# 列选择器
ttk.Label(multi_col_frame, text="选择要确认的列:").pack(anchor="w", padx=10, pady=(10, 0))
col_selector_frame = ttk.Frame(multi_col_frame)
col_selector_frame.pack(fill="x", pady=5, padx=10)
self.col_selector = ttk.Combobox(col_selector_frame, state="readonly", width=15)
self.col_selector.pack(side="left", fill="x", expand=True, padx=(0, 5))
# 添加/移除按钮
ttk.Button(col_selector_frame, text="添加", command=self.add_confirmation_column, width=8).pack(side="left")
# 已选列列表
ttk.Label(multi_col_frame, text="已选确认列 (可拖动调整顺序):").pack(anchor="w", padx=10, pady=(10, 0))
self.selected_cols_listbox = tk.Listbox(multi_col_frame, height=3, font=self.normal_font,
bg="#f8f9fa", highlightthickness=0)
self.selected_cols_listbox.pack(fill="x", pady=5, padx=10)
# 加载已配置的确认列
for col in self.confirmation_columns:
self.selected_cols_listbox.insert(tk.END, col)
# 移除按钮
remove_btn = ttk.Button(multi_col_frame, text="移除选中列", command=self.remove_confirmation_column)
remove_btn.pack(fill="x", pady=(0, 10), padx=10)
# 操作按钮区域
btn_frame = ttk.Frame(control_card)
btn_frame.pack(fill="x", padx=5, pady=10)
ttk.Button(btn_frame, text="保存配置", command=self.save_current_config, style="Success.TButton").pack(fill="x", pady=5)
ttk.Button(btn_frame, text="确认选中项", command=self.confirm_selected, style="Primary.TButton").pack(fill="x", pady=5)
ttk.Button(btn_frame, text="全选", command=self.select_all, style="Primary.TButton").pack(side="left", fill="x", expand=True, pady=5)
ttk.Button(btn_frame, text="取消全选", command=self.deselect_all, style="Danger.TButton").pack(side="left", fill="x", expand=True, pady=5)
ttk.Button(btn_frame, text="保存到Excel", command=self.save_to_excel, style="Success.TButton").pack(fill="x", pady=5)
# 数据显示区域（卡片）
data_card = ttk.LabelFrame(content_frame, text="功能点列表", style="Card.TLabelframe")
data_card.pack(side="right", fill="both", expand=True)
# Treeview容器
tree_frame = ttk.Frame(data_card)
tree_frame.pack(fill="both", expand=True, padx=5, pady=5)
# 创建Treeview
# 使用保存的列顺序或默认顺序
if not self.column_order:
self.column_order = ["selection", "id", "description", "status"] + [f"col_{i}" for i in range(len(self.confirmation_columns))]
self.tree = ttk.Treeview(tree_frame, columns=self.column_order, show="headings", height=20)
# 设置列标题文本
self.set_column_headings()
# 滚动条设置
vsb = ttk.Scrollbar(tree_frame, orient="vertical", command=self.tree.yview)
hsb = ttk.Scrollbar(tree_frame, orient="horizontal", command=self.tree.xview)
self.tree.configure(yscrollcommand=vsb.set, xscrollcommand=hsb.set)
# 使用grid布局管理器
self.tree.grid(row=0, column=0, sticky="nsew")
vsb.grid(row=0, column=1, sticky="ns")
hsb.grid(row=1, column=0, sticky="ew")
# 配置网格权重
tree_frame.columnconfigure(0, weight=1)
tree_frame.rowconfigure(0, weight=1)
# 配置列属性
self.tree.column("selection", width=50, anchor="center")
self.tree.column("id", width=100, anchor="center")
self.tree.column("description", width=400, anchor="w", stretch=True) # 增加功能点列宽度
self.tree.column("status", width=100, anchor="center")
for i in range(len(self.confirmation_columns)):
self.tree.column(f"col_{i}", width=100, anchor="center")
# 添加标签样式
self.tree.tag_configure("confirmed", background="#d4edda")
self.tree.tag_configure("pending", background="#f8d7da")
# 绑定列调整事件
self.tree.bind("<Configure>", self.on_tree_configure)
# 绑定点击事件处理复选框
self.tree.bind("<Button-1>", self.on_tree_click)
# 详情区域
detail_frame = ttk.LabelFrame(data_card, text="功能点详情", style="Card.TLabelframe")
detail_frame.pack(fill="x", padx=5, pady=(0, 5))
# 使用grid布局所有组件
detail_label = ttk.Label(detail_frame, text="功能点详情:")
detail_label.grid(row=0, column=0, sticky="w", padx=5, pady=5)
self.detail_text = tk.Text(detail_frame, wrap="word", font=self.normal_font, height=3)
scroll_detail = ttk.Scrollbar(detail_frame, command=self.detail_text.yview)
self.detail_text.config(yscrollcommand=scroll_detail.set)
# 使用grid布局文本框和滚动条
self.detail_text.grid(row=1, column=0, sticky="nsew", padx=5, pady=5)
scroll_detail.grid(row=1, column=1, sticky="ns", pady=5)
# 配置网格权重
detail_frame.columnconfigure(0, weight=1)
detail_frame.rowconfigure(1, weight=1)
self.detail_text.config(state="disabled")
# 绑定Treeview选择事件
self.tree.bind("<<TreeviewSelect>>", self.on_tree_select)
# 状态栏
self.status_var = tk.StringVar()
self.status_var.set("就绪 - 请选择Excel文件开始")
# 创建状态栏容器框架
status_container = ttk.Frame(self.master, style="Status.TFrame")
status_container.pack(side="bottom", fill="x", padx=0, pady=0)
# 添加分隔线
separator = ttk.Separator(status_container, orient="horizontal")
separator.pack(fill="x", pady=(0, 10))
# 状态标签
status_bar = ttk.Label(
status_container,
textvariable=self.status_var,
style="Status.TLabel",
anchor="w",
padding=(10, 5, 10, 5)
)
status_bar.pack(side="bottom", fill="x", expand=False)
# 修改 set_column_headings 方法
def set_column_headings(self):
"""设置列标题"""
# 定义列标题映射
column_titles = {
"selection": "选择",
"id": "ID",
"description": "功能点",
"status": "状态"
}
# 设置列标题
for col in self.tree["columns"]:
# 特殊列使用预设标题
if col in column_titles:
self.tree.heading(col, text=column_titles[col])
# 确认列使用原列名作为标题
elif col in self.confirmation_columns:
self.tree.heading(col, text=col)
def on_tree_configure(self, event):
"""Treeview大小改变时调整列宽"""
self.adjust_columns()
def on_tree_select(self, event):
"""当Treeview中选中行时，显示功能点详情"""
selected_items = self.tree.selection()
if not selected_items:
return
item = selected_items[0]
# 获取功能点描述，在第三列（索引2）
values = self.tree.item(item, "values")
if len(values) >= 3:
desc = values[2]
self.detail_text.config(state="normal")
self.detail_text.delete(1.0, tk.END)
self.detail_text.insert(tk.END, desc)
self.detail_text.config(state="disabled")
def on_tree_click(self, event):
"""处理Treeview点击事件，切换复选框状态"""
region = self.tree.identify("region", event.x, event.y)
if region == "cell":
column = self.tree.identify_column(event.x)
item = self.tree.identify_row(event.y)
if not item:
return
# 获取列索引
col_idx = int(column[1:]) - 1
columns = self.tree["columns"]
# 确保索引有效
if col_idx >= len(columns):
return
# 获取列标识符
col_id = columns[col_idx]
# 获取列标题文本
col_text = self.tree.heading(col_id, "text")
# 只处理"选择"列
if col_text == "选择":
# 获取当前状态
values = list(self.tree.item(item, "values"))
current_state = values[col_idx]
# 切换状态
if current_state == "☐":
new_state = "☑"
else:
new_state = "☐"
# 更新Treeview
values[col_idx] = new_state
self.tree.item(item, values=values)
# 更新状态存储
self.check_states[item] = (new_state == "☑")
# 更新行的整体状态显示
self.update_row_status(item)
def update_row_status(self, item_id):
"""根据复选框状态更新行状态显示"""
values = list(self.tree.item(item_id, "values"))
is_checked = self.check_states.get(item_id, False)
# 找到状态列的位置
status_col_idx = None
for idx, col_id in enumerate(self.tree["columns"]):
if self.tree.heading(col_id, "text") == "状态":
status_col_idx = idx
break
if status_col_idx is not None and status_col_idx < len(values):
if is_checked:
values[status_col_idx] = "✓ 待确认"
self.tree.item(item_id, tags=("confirmed",))
else:
values[status_col_idx] = "✗ 未确认"
self.tree.item(item_id, tags=("pending",))
self.tree.item(item_id, values=values)
def enable_column_dragging(self):
"""启用列拖动功能"""
self.drag_data = {"item": None, "column": None}
def on_header_click(event):
# 记录开始拖动的列
region = self.tree.identify("region", event.x, event.y)
if region == "heading":
col_id = self.tree.identify_column(event.x)
self.drag_data["column"] = col_id
self.drag_data["start_values"] = {} # 保存所有行的值
def on_header_drag(event):
# 处理拖动中的列
if self.drag_data["column"]:
region = self.tree.identify("region", event.x, event.y)
if region == "heading":
end_col = self.tree.identify_column(event.x)
if end_col != self.drag_data["column"]:
# 移动列
self.move_column(self.drag_data["column"], end_col)
self.drag_data["column"] = end_col
def on_header_release(event):
# 结束拖动
if "column" in self.drag_data:
# 保存列顺序到配置
self.column_order = self.tree["columns"]
self.config["column_order"] = self.column_order
self.save_config()
self.drag_data.pop("column", None)
# 绑定事件
self.tree.bind("<ButtonPress-1>", on_header_click)
self.tree.bind("<B1-Motion>", on_header_drag)
self.tree.bind("<ButtonRelease-1>", on_header_release)
def move_column(self, from_col, to_col):
"""移动列位置及其数据"""
# 获取当前列顺序
columns = list(self.tree["columns"])
# 转换为索引
from_idx = int(from_col.replace("#", "")) - 1
to_idx = int(to_col.replace("#", "")) - 1
# 移动列
if from_idx < len(columns) and to_idx < len(columns):
col = columns.pop(from_idx)
columns.insert(to_idx, col)
# 更新列顺序
self.tree["columns"] = columns
# 重新设置列标题
self.set_column_headings()
# 移动所有行的数据
for item in self.tree.get_children():
values = list(self.tree.item(item, "values"))
if from_idx < len(values) and to_idx < len(values):
# 移动数据
value_to_move = values.pop(from_idx)
values.insert(to_idx, value_to_move)
self.tree.item(item, values=values)
# 调整列宽
self.adjust_columns()
def adjust_columns(self, event=None):
"""根据窗口大小自动调整列宽"""
if not self.tree.winfo_exists():
return
width = self.tree.winfo_width()
if width < 100: # 防止宽度过小
return
# 计算可用宽度
available_width = width - 20 # 减去滚动条宽度
# 设置列宽比例
column_weights = {
"selection": 0.05,
"id": 0.15,
"description": 0.55, # 增加功能点列的比例
"status": 0.1
}
# 设置基础列宽
for col, weight in column_weights.items():
if col in self.tree["columns"]:
col_width = int(available_width * weight)
self.tree.column(col, width=col_width)
# 设置确认列宽
if self.confirmation_columns:
confirm_col_width = int(available_width * 0.08) # 减小确认列宽度
for i in range(len(self.confirmation_columns)):
col_id = f"col_{i}"
if col_id in self.tree["columns"]:
self.tree.column(col_id, width=confirm_col_width)
def add_confirmation_column(self):
"""添加确认列"""
col = self.col_selector.get()
if col and col not in self.confirmation_columns:
self.confirmation_columns.append(col)
self.selected_cols_listbox.insert(tk.END, col)
# 更新Treeview列
self.update_treeview_columns()
def update_treeview_columns(self):
"""更新Treeview列"""
# 构建列标识符 - 使用实际列名而不是 col_i 占位符
base_columns = ["selection", "id", "description", "status"]
confirmation_cols = self.confirmation_columns # 直接使用实际列名
# 创建新的列顺序（如果不存在）
if not self.column_order:
self.column_order = base_columns + confirmation_cols
else:
# 过滤掉无效的列标识符
valid_columns = set(base_columns + confirmation_cols)
self.column_order = [col for col in self.column_order if col in valid_columns]
# 添加缺失的列
for col in base_columns + confirmation_cols:
if col not in self.column_order:
self.column_order.append(col)
# 重新配置Treeview
self.tree.configure(columns=self.column_order)
self.tree["show"] = "headings"
# 设置列标题
self.set_column_headings()
# 设置列属性 - 只在列存在时设置
if "selection" in self.column_order:
self.tree.column("selection", width=50, anchor="center")
if "id" in self.column_order:
self.tree.column("id", width=100, anchor="center")
if "description" in self.column_order:
self.tree.column("description", width=400, anchor="w", stretch=True)
if "status" in self.column_order:
self.tree.column("status", width=100, anchor="center")
# 设置确认列属性
for col in self.confirmation_columns:
if col in self.column_order:
self.tree.column(col, width=100, anchor="center")
# 调整列宽
self.adjust_columns()
def remove_confirmation_column(self):
"""移除确认列"""
selection = self.selected_cols_listbox.curselection()
if selection:
index = selection[0]
col = self.confirmation_columns[index]
self.confirmation_columns.pop(index)
self.selected_cols_listbox.delete(index)
# 更新Treeview列
self.update_treeview_columns()
def browse_file(self):
initial_dir = self.config.get("last_dir", os.getcwd())
file_path = filedialog.askopenfilename(
initialdir=initial_dir,
filetypes=[("Excel文件", "*.xlsx;*.xls")]
)
if file_path:
self.path_entry.delete(0, tk.END)
self.path_entry.insert(0, file_path)
# 更新最后访问目录
self.config["last_dir"] = os.path.dirname(file_path)
self.save_config()
def select_sheet(self):
"""选择工作表"""
file_path = self.path_entry.get()
if not file_path or not os.path.exists(file_path):
messagebox.showerror("错误", "请先选择有效的Excel文件")
return
try:
# 获取所有sheet名称
xl = pd.ExcelFile(file_path)
sheet_names = xl.sheet_names
# 创建选择对话框
sheet_dialog = tk.Toplevel(self.master)
sheet_dialog.title("选择工作表")
sheet_dialog.geometry("400x300")
sheet_dialog.transient(self.master)
sheet_dialog.grab_set()
sheet_dialog.configure(bg="#f5f7fa")
ttk.Label(sheet_dialog, text="请选择工作表:", font=self.subtitle_font, background="#f5f7fa").pack(pady=10)
# 使用Treeview显示工作表
sheet_tree = ttk.Treeview(sheet_dialog, columns=("名称",), show="headings", height=8)
sheet_tree.heading("名称", text="工作表名称")
sheet_tree.column("名称", width=350)
sheet_tree.pack(fill="both", expand=True, padx=20, pady=5)
for name in sheet_names:
sheet_tree.insert("", "end", values=(name,))
# 按钮框架
btn_frame = ttk.Frame(sheet_dialog)
btn_frame.pack(fill="x", padx=20, pady=10)
def on_select():
selected = sheet_tree.selection()
if selected:
self.current_sheet = sheet_tree.item(selected[0], "values")[0]
self.sheet_var.set(self.current_sheet)
# 保存工作表名称到配置
self.config["sheet_name"] = self.current_sheet
self.save_config()
sheet_dialog.destroy()
ttk.Button(btn_frame, text="取消", command=sheet_dialog.destroy).pack(side="right", padx=5)
ttk.Button(btn_frame, text="确定", command=on_select, style="Primary.TButton").pack(side="right")
except Exception as e:
messagebox.showerror("错误", f"读取Excel失败: {str(e)}")
def load_data(self):
file_path = self.path_entry.get()
if not file_path or not os.path.exists(file_path):
messagebox.showerror("错误", "无效的文件路径")
return
# 在状态栏显示加载中
self.status_var.set("正在加载数据...")
self.master.update() # 强制更新界面
# 清空Treeview
for item in self.tree.get_children():
self.tree.delete(item)
self.check_states = {}
# 获取当前配置
id_col = self.id_col_var.get().strip()
desc_col = self.desc_col_var.get().strip()
status_col = self.status_col_var.get().strip()
sheet_name = self.sheet_var.get() or None
header_row = self.header_var.get() - 1 # pandas header是0-based索引
# 直接调用加载任务
self.load_task(file_path, id_col, desc_col, status_col, sheet_name, header_row)
def load_task(self, file_path, id_col, desc_col, status_col, sheet_name, header_row):
try:
# 确保使用正确的sheet_name
if not sheet_name and self.config.get("sheet_name"):
sheet_name = self.config["sheet_name"] # 从配置获取工作表名
# 读取Excel文件
if sheet_name:
self.df = pd.read_excel(
file_path,
sheet_name=sheet_name,
header=header_row
)
else:
# 如果没有指定sheet，尝试读取第一个sheet
self.df = pd.read_excel(
file_path,
header=header_row
)
# 尝试获取第一个sheet的名称
xl = pd.ExcelFile(file_path)
if xl.sheet_names:
self.current_sheet = xl.sheet_names[0]
self.sheet_var.set(self.current_sheet)
self.excel_path = file_path
# 检查列是否存在
missing_cols = []
if id_col not in self.df.columns:
missing_cols.append(f"ID列 '{id_col}'")
if desc_col not in self.df.columns:
missing_cols.append(f"功能点列 '{desc_col}'")
if missing_cols:
# 提供更详细的错误信息
available_cols = "\n".join(self.df.columns)
error_msg = (
f"以下列不存在: {', '.join(missing_cols)}\n\n"
f"可用列名:\n{available_cols}\n\n"
"请检查表头行设置是否正确（默认为第9行）"
)
raise ValueError(error_msg)
# 如果状态列不存在，则创建
if status_col not in self.df.columns:
self.df[status_col] = "否" # 默认未确认
# 更新列选择器
self.col_selector["values"] = list(self.df.columns)
# 更新Treeview列
self.update_treeview_columns()
# 处理ID列NaN值
if id_col in self.df.columns:
if self.df[id_col].dtype == float:
# 处理浮点型NaN
self.df[id_col] = self.df[id_col].fillna(-1).astype(int).astype(str)
self.df[id_col] = self.df[id_col].replace("-1", "缺失ID")
else:
# 处理其他类型的NaN
self.df[id_col] = self.df[id_col].fillna("缺失ID").astype(str)
# 处理描述列NaN值
if desc_col in self.df.columns:
self.df[desc_col] = self.df[desc_col].fillna("无描述").astype(str)
# 添加数据到Treeview
for i, row in self.df.iterrows():
status_value = row.get(status_col, "否")
# 使用图标表示状态
status_icon = "✓" if status_value in ["是", "Y", "y", "Yes", "yes", "OK", "确认"] else "✗"
status_text = f"{status_icon} {status_value}"
tag = "confirmed" if status_icon == "✓" else "pending"
# 构建行数据 - 确保ID是字符串
id_value = str(row[id_col]) if id_col in row else "缺失ID"
# 处理功能点描述 - 添加换行符
desc = row[desc_col]
# 每50个字符插入一个换行符
wrapped_desc = '\n'.join([desc[i:i+50] for i in range(0, len(desc), 50)])
# 构建行数据 - 使用实际列名而不是 col_i
values = {
"selection": "☐",
"id": str(row[id_col]) if id_col in row else "缺失ID",
"description": wrapped_desc,
"status": status_text
}
# 添加多列确认数据
for col in self.confirmation_columns:
values[col] = row[col] if col in row else ""
# 按列顺序构建值列表
ordered_values = [values.get(col, "") for col in self.column_order]
# 插入行
item_id = self.tree.insert("", "end", values=ordered_values, tags=(tag,))
# 存储复选框状态
self.check_states[item_id] = (status_icon == "✓")
# 每50行更新一次界面
if i % 50 == 0:
self.tree.update()
self.master.update_idletasks()
# 强制刷新界面
self.tree.update_idletasks()
self.master.update_idletasks()
# 更新状态
self.status_var.set(f"成功加载: {len(self.df)} 条记录")
# 调整列宽并刷新界面
self.adjust_columns()
except Exception as e:
# 显示详细的错误信息
error_msg = f"读取Excel失败: {str(e)}\n\n{traceback.format_exc()}"
self.status_var.set("加载失败")
messagebox.showerror("加载错误", error_msg)
def confirm_selected(self):
"""确认选中的功能点"""
selected_items = []
for item_id in self.tree.get_children():
if self.check_states.get(item_id, False):
selected_items.append(item_id)
if not selected_items:
messagebox.showinfo("提示", "请先选择功能点")
return
custom_status = self.custom_status_var.get().strip() or "OK"
for item_id in selected_items:
values = list(self.tree.item(item_id, "values"))
# 找到状态列的位置
status_col_idx = None
for idx, col_id in enumerate(self.tree["columns"]):
if self.tree.heading(col_id, "text") == "状态":
status_col_idx = idx
break
# 更新状态列
if status_col_idx is not None and status_col_idx < len(values):
values[status_col_idx] = f"✓ {custom_status}"
# 更新多列确认
if self.confirmation_columns and self.df is not None:
row_idx = self.tree.index(item_id)
for i, col in enumerate(self.confirmation_columns):
col_id = f"col_{i}"
if col_id in self.tree["columns"]:
col_idx = self.tree["columns"].index(col_id)
if col_idx < len(values) and col in self.df.columns:
values[col_idx] = custom_status
self.df.at[row_idx, col] = custom_status
# 更新Treeview
self.tree.item(item_id, values=values, tags=("confirmed",))
self.status_var.set(f"已确认 {len(selected_items)} 个功能点")
# 自动保存
self.auto_save()
def select_all(self):
"""全选功能点"""
for item_id in self.tree.get_children():
values = list(self.tree.item(item_id, "values"))
# 找到选择列的位置
select_col_idx = None
for idx, col_id in enumerate(self.tree["columns"]):
if self.tree.heading(col_id, "text") == "选择":
select_col_idx = idx
break
# 更新选择列
if select_col_idx is not None and select_col_idx < len(values):
values[select_col_idx] = "☑"
self.tree.item(item_id, values=values)
self.check_states[item_id] = True
self.update_row_status(item_id)
self.status_var.set("已全选所有功能点")
def deselect_all(self):
"""取消全选功能点"""
for item_id in self.tree.get_children():
values = list(self.tree.item(item_id, "values"))
# 找到选择列的位置
select_col_idx = None
for idx, col_id in enumerate(self.tree["columns"]):
if self.tree.heading(col_id, "text") == "选择":
select_col_idx = idx
break
# 更新选择列
if select_col_idx is not None and select_col_idx < len(values):
values[select_col_idx] = "☐"
self.tree.item(item_id, values=values)
self.check_states[item_id] = False
self.update_row_status(item_id)
self.status_var.set("已取消全选所有功能点")
def save_current_config(self):
"""保存当前配置"""
self.config["id_col"] = self.id_col_var.get().strip()
self.config["desc_col"] = self.desc_col_var.get().strip()
self.config["status_col"] = self.status_col_var.get().strip()
self.config["sheet_name"] = self.sheet_var.get()
self.config["header_row"] = self.header_var.get()
self.config["custom_status"] = self.custom_status_var.get().strip()
self.config["confirmation_columns"] = self.confirmation_columns
self.config["column_order"] = self.column_order
self.save_config()
messagebox.showinfo("成功", "配置已保存")
def auto_save(self):
"""自动保存功能"""
if self.df is None or not self.excel_path:
return
try:
# 获取当前配置
id_col = self.id_col_var.get().strip()
desc_col = self.desc_col_var.get().strip()
status_col = self.status_col_var.get().strip()
# 更新DataFrame中的确认状态
for i, item_id in enumerate(self.tree.get_children()):
# 获取Treeview中的状态值（去掉图标）
values = self.tree.item(item_id, "values")
# 找到状态列的位置
status_col_idx = None
for idx, col_id in enumerate(self.tree["columns"]):
if self.tree.heading(col_id, "text") == "状态":
status_col_idx = idx
break
if status_col_idx is not None and status_col_idx < len(values):
status_value = values[status_col_idx]
if status_value.startswith(("✓", "✗")):
status_value = status_value[2:].strip()
self.df.at[i, status_col] = status_value
# 保存回Excel
wb = openpyxl.load_workbook(self.excel_path)
if self.current_sheet in wb.sheetnames:
del wb[self.current_sheet]
ws = wb.create_sheet(self.current_sheet)
# 写入数据
for r, row in enumerate(dataframe_to_rows(self.df, index=False, header=True), 1):
ws.append(row)
wb.save(self.excel_path)
self.status_var.set("数据已自动保存")
except Exception:
self.status_var.set("自动保存失败")
def save_to_excel(self):
if self.df is None:
messagebox.showerror("错误", "没有加载的数据")
return
try:
# 执行保存
self.auto_save()
messagebox.showinfo("成功", f"数据已保存到:\n{self.excel_path}")
self.status_var.set("数据保存成功")
except Exception as e:
messagebox.showerror("保存错误", f"写入Excel失败: {str(e)}\n请确保文件未被其他程序打开")
logger.error(f"保存失败: {str(e)}")
def main():
root = tk.Tk()
# 设置应用图标
try:
# 创建一个简单的蓝色方块作为图标
icon_data = """
R0lGODlhEAAQAIAAAP///wAAACH5BAEAAAAALAAAAAAQABAAAAIOhI+py+0Po5y02ouzPgUAOw==
"""
icon_img = tk.PhotoImage(data=icon_data)
root.tk.call('wm', 'iconphoto', root._w, icon_img)
except Exception as e:
logger.warning(f"设置应用图标失败: {str(e)}")
app = ExcelControlPanel(root)
root.mainloop()
if __name__ == "__main__":
# 强制使用UTF-8编码
import sys
import io
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')
main()
错误可能太多了，能否帮我重新设计框架，书写代码，完成这个代码中实现的功能，请给我完整的代码
扫一扫
3527
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
