import logging
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 datetime
# 配置日志系统
def setup_logging():
# 创建日志目录
log_dir = "logs"
if not os.path.exists(log_dir):
os.makedirs(log_dir)
# 创建带时间戳的日志文件
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
log_file = os.path.join(log_dir, f"excel_tool_{timestamp}.log")
# 配置日志
logging.basicConfig(
level=logging.DEBUG,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
handlers=[
logging.FileHandler(log_file),
logging.StreamHandler()
]
)
return logging.getLogger('ExcelControlPanel')
# 初始化日志
logger = setup_logging()
class ScrollableFrame(ttk.Frame):
"""自定义可滚动框架实现"""
def __init__(self, parent, *args, **kwargs):
super().__init__(parent, *args, **kwargs)
logger.debug("初始化可滚动框架")
# 创建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)
logger.debug("可滚动框架初始化完成")
def on_frame_configure(self, event):
"""当内部框架大小改变时更新滚动区域"""
logger.debug(f"框架大小改变: {event.width}x{event.height}")
self.canvas.configure(scrollregion=self.canvas.bbox("all"))
def on_canvas_configure(self, event):
"""当Canvas大小改变时调整内部框架宽度"""
logger.debug(f"Canvas大小改变: {event.width}x{event.height}")
self.canvas.itemconfig(self.canvas_frame, width=event.width)
def on_mousewheel(self, event):
"""鼠标滚轮滚动支持"""
logger.debug(f"鼠标滚轮事件: delta={event.delta}")
self.canvas.yview_scroll(int(-1*(event.delta/120)), "units")
class ExcelControlPanel:
def __init__(self, master):
logger.info("初始化Excel控制面板")
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", [])
logger.debug(f"加载配置: {self.config}")
# 创建界面元素
self.create_widgets()
# 初始化数据
self.excel_path = ""
self.df = None
self.check_states = {} # 存储每个功能点的复选框状态，key为item_id
self.current_sheet = ""
self.header_row = 0
# 启用列拖动
self.enable_column_dragging()
# 初始化后更新Treeview列
self.update_treeview_columns()
logger.info("Excel控制面板初始化完成")
def set_styles(self):
"""设置全局样式和字体"""
logger.debug("设置全局样式")
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=10)
# 配置样式
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)
style.configure("Treeview.Heading", font=self.subtitle_font, background="#4a76b5", foreground="white")
style.configure("Treeview", font=self.normal_font, rowheight=30, 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):
"""加载配置文件"""
logger.info("加载配置文件")
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": []
}
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]
logger.debug(f"从文件加载配置: {config}")
return config
except Exception as e:
logger.error(f"加载配置文件失败: {str(e)}")
return default_config
logger.debug("使用默认配置")
return default_config
def save_config(self):
"""保存配置文件"""
logger.info("保存配置文件")
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)
logger.debug(f"配置保存成功: {self.config}")
except Exception as e:
logger.error(f"保存配置文件失败: {str(e)}")
def create_widgets(self):
"""创建现代化界面元素"""
logger.info("创建界面组件")
# 主容器
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 as e:
logger.warning(f"创建图标失败: {str(e)}")
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, style="Custom.TCombobox")
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)
# === 修改1：使用Frame作为Treeview的容器 ===
tree_container = ttk.Frame(data_card)
tree_container.pack(fill="both", expand=True, padx=5, pady=5)
# === 修改2：创建Treeview并直接放入容器 ===
columns = ["selection", "id", "description", "status"] + [f"col_{i}" for i in range(len(self.confirmation_columns))]
self.tree = ttk.Treeview(tree_container, columns=columns, show="headings", height=20)
logger.debug(f"创建Treeview，列: {columns}")
# 创建带滚动条的表格
# 使用固定的列标识符
columns = ["selection", "id", "description", "status"] + [f"col_{i}" for i in range(len(self.confirmation_columns))]
self.tree = ttk.Treeview(data_card, columns=columns, show="headings", height=20)
logger.debug(f"创建Treeview，列: {columns}")
# 设置列标题文本
self.tree.heading("selection", text="选择")
self.tree.heading("id", text="ID")
self.tree.heading("description", text="功能点")
self.tree.heading("status", text="状态")
# 设置确认列标题
for i, col in enumerate(self.confirmation_columns):
self.tree.heading(f"col_{i}", text=col)
vsb = ttk.Scrollbar(data_card, orient="vertical", command=self.tree.yview)
hsb = ttk.Scrollbar(data_card, orient="horizontal", command=self.tree.xview)
self.tree.configure(yscrollcommand=vsb.set, xscrollcommand=hsb.set)
# 添加分割条和详情文本框
paned_window = tk.PanedWindow(data_card, orient=tk.VERTICAL, sashrelief=tk.RAISED, sashwidth=4)
paned_window.pack(fill="both", expand=True, padx=5, pady=5)
# 将Treeview放入PanedWindow的上部分
tree_frame = ttk.Frame(paned_window)
paned_window.add(tree_frame)
self.tree.pack(in_=tree_frame, side="top", fill="both", expand=True)
vsb.pack(in_=tree_frame, side="right", fill="y", padx=(0, 5), pady=5)
hsb.pack(in_=tree_frame, side="bottom", fill="x", padx=5, pady=(0, 5))
# 配置列属性
self.tree.column("selection", width=50, anchor="center")
self.tree.column("id", width=100, anchor="center")
self.tree.column("description", width=300, anchor="w")
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.Frame(paned_window, height=100)
paned_window.add(detail_frame)
ttk.Label(detail_frame, text="功能点详情:").pack(anchor="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)
self.detail_text.pack(side="left", fill="both", expand=True, padx=5, pady=5)
scroll_detail.pack(side="right", fill="y", padx=(0, 5), pady=5)
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)
logger.info("界面组件创建完成")
def on_tree_configure(self, event):
"""Treeview大小改变时调整列宽"""
logger.debug(f"Treeview大小改变: {event.width}x{event.height}")
self.adjust_columns()
def on_tree_select(self, event):
"""当Treeview中选中行时，显示功能点详情"""
selected_items = self.tree.selection()
if not selected_items:
return
item = selected_items[0]
logger.debug(f"选中行: {item}")
# 获取功能点描述，在第三列（索引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)
logger.debug(f"Treeview点击: region={region}, x={event.x}, y={event.y}")
if region == "cell":
column = self.tree.identify_column(event.x)
item = self.tree.identify_row(event.y)
logger.debug(f"点击单元格: column={column}, item={item}")
# 只处理"选择"列（第一列）
if column == "#1" and item:
# 获取当前状态
values = list(self.tree.item(item, "values"))
current_state = values[0]
# 切换状态
if current_state == "☐":
new_state = "☑"
else:
new_state = "☐"
# 更新Treeview
values[0] = new_state
self.tree.item(item, values=values)
logger.debug(f"更新行状态: item={item}, new_state={new_state}")
# 更新状态存储
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)
if is_checked:
values[3] = "✓ 待确认"
self.tree.item(item_id, tags=("confirmed",))
else:
values[3] = "✗ 未确认"
self.tree.item(item_id, tags=("pending",))
self.tree.item(item_id, values=values)
logger.debug(f"更新行状态显示: item_id={item_id}, is_checked={is_checked}")
def enable_column_dragging(self):
"""启用列拖动功能"""
logger.debug("启用列拖动功能")
def on_header_click(event):
# 记录开始拖动的列
region = self.tree.identify("region", event.x, event.y)
if region == "heading":
self.drag_start_col = self.tree.identify_column(event.x)
logger.debug(f"开始拖动列: {self.drag_start_col}")
def on_header_drag(event):
# 处理拖动中的列
if hasattr(self, 'drag_start_col'):
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_start_col:
# 移动列
self.move_column(self.drag_start_col, end_col)
self.drag_start_col = end_col
def on_header_release(event):
# 结束拖动
if hasattr(self, 'drag_start_col'):
logger.debug(f"结束拖动列: {self.drag_start_col}")
del self.drag_start_col
# 绑定事件
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):
"""移动列位置"""
logger.debug(f"移动列: from={from_col}, to={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
# 重新配置列
for i, col in enumerate(columns):
self.tree.heading(col, text=self.tree.heading(col, "text"))
self.tree.column(col, anchor="w")
# 调整列宽
self.adjust_columns()
def adjust_columns(self, event=None):
"""根据窗口大小自动调整列宽"""
if not self.tree.winfo_exists():
return
width = self.tree.winfo_width()
logger.debug(f"调整列宽: width={width}")
if width < 100: # 防止宽度过小
return
# 计算可用宽度
available_width = width - 20 # 减去滚动条宽度
# 设置列宽比例
column_weights = {
"selection": 0.05,
"id": 0.15,
"description": 0.7,
"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)
logger.debug(f"设置列宽: {col} = {col_width}")
# 设置确认列宽
if self.confirmation_columns:
confirm_col_width = int(available_width * 0.1)
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)
logger.debug(f"设置确认列宽: {col_id} = {confirm_col_width}")
def add_confirmation_column(self):
"""添加确认列"""
col = self.col_selector.get()
logger.debug(f"添加确认列: {col}")
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列以显示确认列 - 修复列配置问题"""
logger.debug("更新Treeview列")
# 构建列标识符
columns = ["selection", "id", "description", "status"] + [f"col_{i}" for i in range(len(self.confirmation_columns))]
# 重新配置Treeview
self.tree.configure(columns=columns)
self.tree["show"] = "headings"
logger.debug(f"Treeview列更新为: {columns}")
# 设置列标题
self.tree.heading("selection", text="选择")
self.tree.heading("id", text="ID")
self.tree.heading("description", text="功能点")
self.tree.heading("status", text="状态")
# 设置确认列标题
for i, col in enumerate(self.confirmation_columns):
self.tree.heading(f"col_{i}", text=col)
def remove_confirmation_column(self):
"""移除确认列"""
selection = self.selected_cols_listbox.curselection()
if selection:
index = selection[0]
col = self.confirmation_columns[index]
logger.debug(f"移除确认列: {col}")
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())
logger.debug(f"浏览文件: initial_dir={initial_dir}")
file_path = filedialog.askopenfilename(
initialdir=initial_dir,
filetypes=[("Excel文件", "*.xlsx;*.xls")]
)
if file_path:
logger.info(f"选择文件: {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()
logger.debug(f"选择工作表: file_path={file_path}")
if not file_path or not os.path.exists(file_path):
logger.error("无效的文件路径")
messagebox.showerror("错误", "请先选择有效的Excel文件")
return
try:
# 获取所有sheet名称
xl = pd.ExcelFile(file_path)
sheet_names = xl.sheet_names
logger.debug(f"工作表列表: {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]
logger.info(f"选择工作表: {self.current_sheet}")
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:
logger.error(f"选择工作表失败: {str(e)}")
messagebox.showerror("错误", f"读取Excel失败: {str(e)}")
def load_data(self):
file_path = self.path_entry.get()
logger.info(f"开始加载数据: {file_path}")
if not file_path or not os.path.exists(file_path):
logger.error("无效的文件路径")
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 = {}
logger.debug("Treeview已清空")
# 获取当前配置
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索引
logger.debug(f"加载配置: id_col={id_col}, desc_col={desc_col}, status_col={status_col}, "
f"sheet_name={sheet_name}, header_row={header_row}")
# 直接调用加载任务
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:
logger.info("开始加载Excel数据")
# 确保使用正确的sheet_name
if not sheet_name and self.config.get("sheet_name"):
sheet_name = self.config["sheet_name"] # 从配置获取工作表名
logger.debug(f"使用配置中的工作表: {sheet_name}")
# 读取Excel文件
if sheet_name:
logger.info(f"读取工作表: {sheet_name}")
self.df = pd.read_excel(
file_path,
sheet_name=sheet_name,
header=header_row
)
else:
logger.info("读取第一个工作表")
# 如果没有指定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)
logger.debug(f"设置当前工作表为: {self.current_sheet}")
self.excel_path = file_path
logger.debug(f"DataFrame形状: {self.df.shape}")
# 检查列是否存在
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行）"
)
logger.error(f"列不存在: {missing_cols}")
logger.debug(f"可用列: {available_cols}")
raise ValueError(error_msg)
# 如果状态列不存在，则创建
if status_col not in self.df.columns:
logger.warning(f"状态列 '{status_col}' 不存在，将创建新列")
self.df[status_col] = "否" # 默认未确认
# 更新列选择器
self.col_selector["values"] = list(self.df.columns)
logger.debug(f"更新列选择器: {list(self.df.columns)}")
# 更新Treeview列
self.update_treeview_columns()
# === 新增：数据清洗和验证 ===
# 处理ID列NaN值
if id_col in self.df.columns:
# 检查ID列数据类型
logger.debug(f"ID列原始数据类型: {self.df[id_col].dtype}")
# 转换并填充NaN
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)
logger.debug(f"处理后的ID列数据类型: {self.df[id_col].dtype}")
logger.debug(f"ID列前5个值: {self.df[id_col].head().tolist()}")
# 处理描述列NaN值
if desc_col in self.df.columns:
self.df[desc_col] = self.df[desc_col].fillna("无描述").astype(str)
# 检查缺失值
if id_col in self.df.columns and self.df[id_col].isna().any():
logger.warning(f"ID列仍有 {self.df[id_col].isna().sum()} 个缺失值")
# 添加数据到Treeview
logger.info(f"开始添加数据到Treeview，共 {len(self.df)} 条记录")
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"
# 构建行数据 - 使用固定列标识符
values = [
"☐", # selection列
id_value, # 确保ID是字符串
row[desc_col],
status_text
]
# 添加多列确认数据
for col in self.confirmation_columns:
if col in row:
values.append(row[col])
else:
values.append("")
# 插入行
item_id = self.tree.insert("", "end", values=values, tags=(tag,))
logger.debug(f"添加行 {i}: ID={row[id_col]}, 描述={row[desc_col][:20]}...")
# 存储复选框状态
self.check_states[item_id] = (status_icon == "✓")
# 每50行更新一次界面
if i % 50 == 0:
self.tree.update()
self.master.update_idletasks()
logger.debug(f"已处理 {i+1} 行")
# 强制刷新界面
self.tree.update_idletasks()
self.master.update_idletasks()
logger.info(f"成功添加 {len(self.df)} 行到Treeview")
# 更新状态
self.status_var.set(f"成功加载: {len(self.df)} 条记录")
logger.info(f"成功加载: {len(self.df)} 条记录")
# 调整列宽并刷新界面
self.adjust_columns()
logger.debug("列宽调整完成")
# 记录Treeview状态
logger.debug(f"Treeview子项数量: {len(self.tree.get_children())}")
if len(self.tree.get_children()) > 0:
first_item = self.tree.get_children()[0]
values = self.tree.item(first_item, 'values')
logger.debug(f"第一行数据: {values}")
except Exception as e:
# 显示详细的错误信息
error_msg = f"读取Excel失败: {str(e)}\n\n{traceback.format_exc()}"
logger.error(f"加载失败: {error_msg}")
self.status_var.set("加载失败")
messagebox.showerror("加载错误", error_msg)
def confirm_selected(self):
"""确认选中的功能点"""
logger.info("开始确认选中项")
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:
logger.warning("没有选中任何功能点")
messagebox.showinfo("提示", "请先选择功能点")
return
custom_status = self.custom_status_var.get().strip() or "OK"
logger.debug(f"自定义状态: {custom_status}")
for item_id in selected_items:
values = list(self.tree.item(item_id, "values"))
# 更新状态列
values[3] = 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, start=4):
if col in self.df.columns:
values[i] = custom_status
self.df.at[row_idx, col] = custom_status
# 更新Treeview
self.tree.item(item_id, values=tuple(values), tags=("confirmed",))
logger.debug(f"更新行: {item_id}, 状态={custom_status}")
self.status_var.set(f"已确认 {len(selected_items)} 个功能点")
logger.info(f"已确认 {len(selected_items)} 个功能点")
# 自动保存
self.auto_save()
def select_all(self):
"""全选功能点"""
logger.info("全选功能点")
for item_id in self.tree.get_children():
values = list(self.tree.item(item_id, "values"))
values[0] = "☑" # 设置为选中状态
self.tree.item(item_id, values=values)
self.check_states[item_id] = True
self.update_row_status(item_id)
self.status_var.set("已全选所有功能点")
logger.info("已全选所有功能点")
def deselect_all(self):
"""取消全选功能点"""
logger.info("取消全选功能点")
for item_id in self.tree.get_children():
values = list(self.tree.item(item_id, "values"))
values[0] = "☐" # 设置为未选中状态
self.tree.item(item_id, values=values)
self.check_states[item_id] = False
self.update_row_status(item_id)
self.status_var.set("已取消全选所有功能点")
logger.info("已取消全选所有功能点")
def save_current_config(self):
"""保存当前配置"""
logger.info("保存当前配置")
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.save_config()
messagebox.showinfo("成功", "配置已保存")
logger.info("配置已保存")
def auto_save(self):
"""自动保存功能"""
if self.df is None or not self.excel_path:
logger.warning("自动保存失败: 没有数据或文件路径")
return
try:
logger.info("开始自动保存")
# 获取当前配置
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中的状态值（去掉图标）
status_value = self.tree.item(item_id, "values")[3]
if status_value.startswith(("✓", "✗")):
status_value = status_value[2:].strip()
self.df.at[i, status_col] = status_value
# 保存回Excel - 使用openpyxl直接操作工作簿
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("数据已自动保存")
logger.info(f"数据已自动保存到: {self.excel_path}")
except Exception as e:
self.status_var.set(f"自动保存失败: {str(e)}")
logger.error(f"自动保存失败: {str(e)}")
def save_to_excel(self):
if self.df is None:
logger.warning("保存到Excel失败: 没有加载的数据")
messagebox.showerror("错误", "没有加载的数据")
return
try:
logger.info("保存到Excel")
# 执行保存
self.auto_save()
messagebox.showinfo("成功", f"数据已保存到:\n{self.excel_path}")
self.status_var.set("数据保存成功")
logger.info("数据保存成功")
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__":
main()
这是我修改完部分的代码，请检查，并且在合适的位置告诉我应该怎么改
最新发布
博客提到老王提出了解决问题的方式，即添加属性限定。这一方式可能在信息技术相关场景中用于解决特定问题。
扫一扫
1478
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
