import pandas as pd
import tkinter as tk
from tkinter import ttk, messagebox, filedialog
import os
import json
import threading
import openpyxl
from openpyxl.utils.dataframe import dataframe_to_rows
from tkinter.font import Font
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.create_widgets()
# 初始化数据
self.excel_path = ""
self.df = None
self.check_vars = {} # 存储每个功能点的复选框变量,key为item_id
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=12)
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("Title.TLabel", font=self.title_font, background="#4a76b5",
foreground="white", padding=10)
style.configure("Status.TLabel", font=self.normal_font, background="#333",
foreground="#fff", 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.TCheckbutton", background="white", font=self.normal_font)
# 输入框样式
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": []
}
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:
return default_config
return default_config
def save_config(self):
"""保存配置文件"""
config_path = "excel_config.json"
with open(config_path, 'w', encoding='utf-8') as f:
json.dump(self.config, f, ensure_ascii=False, indent=2)
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=32, height=32)
self.icon_img.put("#4a76b5", (0, 0, 32, 32))
icon_label = ttk.Label(title_frame, image=self.icon_img, background="#4a76b5")
icon_label.pack(side="left", padx=(15, 10))
except:
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, 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=10, 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, padx=(0, 5), pady=5)
ttk.Button(btn_frame, text="取消全选", command=self.deselect_all, style="Danger.TButton").pack(side="left", fill="x", expand=True, padx=(5, 0), 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)
# 创建带滚动条的表格
columns = ("选择", "ID", "功能点", "状态")
self.tree = ttk.Treeview(data_card, columns=columns, show="headings", height=20)
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))
# 配置列
col_widths = {"选择": 50, "ID": 120, "功能点": 600, "状态": 120}
for col in columns:
self.tree.heading(col, text=col, anchor="w")
self.tree.column(col, width=col_widths[col], minwidth=50, stretch=tk.YES if col == "功能点" else tk.NO)
# 添加标签样式
self.tree.tag_configure("confirmed", background="#d4edda")
self.tree.tag_configure("pending", background="#f8d7da")
# 绑定列调整事件
self.tree.bind("<Configure>", self.adjust_columns)
# 添加详情文本框
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=5)
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_bar = ttk.Label(self.master, textvariable=self.status_var, style="Status.TLabel")
status_bar.pack(side="bottom", fill="x")
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 enable_column_dragging(self):
"""启用列拖动功能"""
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)
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'):
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):
"""移动列位置"""
# 获取当前列顺序
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=col)
self.tree.column(col, anchor="w")
# 调整列宽
self.adjust_columns(None)
def adjust_columns(self, event):
"""根据窗口大小自动调整列宽"""
if not self.tree.winfo_exists():
return
width = self.tree.winfo_width()
if width < 100: # 防止宽度过小
return
# 设置列宽比例:选择 5%,ID 15%,功能点 70%,状态 10%
self.tree.column("选择", width=int(width * 0.05))
self.tree.column("ID", width=int(width * 0.15))
self.tree.column("功能点", width=int(width * 0.7))
self.tree.column("状态", width=int(width * 0.1))
# 调整多列确认的列宽
if self.confirmation_columns:
for col in self.confirmation_columns:
self.tree.column(col, width=int(width * 0.1))
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列以显示确认列"""
# 构建列列表
columns = ["选择", "ID", "功能点", "状态"] + self.confirmation_columns
# 重新配置Treeview
self.tree.configure(columns=columns)
self.tree["show"] = "headings"
# 设置列标题
for col in columns:
self.tree.heading(col, text=col)
# 设置最小宽度和拉伸属性
if col == "功能点":
self.tree.column(col, stretch=tk.YES)
else:
self.tree.column(col, stretch=tk.NO, width=100)
# 调整列宽
self.adjust_columns(None)
def remove_confirmation_column(self):
"""移除确认列"""
selection = self.selected_cols_listbox.curselection()
if selection:
index = selection[0]
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())
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)
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
# 获取当前配置
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.status_var.set("正在加载数据...")
self.master.update()
def load_task():
try:
# 读取Excel文件
if sheet_name:
self.df = pd.read_excel(
file_path,
sheet_name=sheet_name,
header=header_row
)
else:
self.df = pd.read_excel(
file_path,
header=header_row
)
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.tree.delete(*self.tree.get_children())
self.check_vars = {}
# 添加数据到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"
# 构建行数据
values = [
"", # 选择列 - 留空,我们将使用自定义复选框
row[id_col],
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,))
# 创建复选框变量
var = tk.BooleanVar(value=(status_icon == "✓"))
self.check_vars[item_id] = var
# 添加复选框到选择列
self.create_checkbox(item_id, var)
# 更新状态
self.status_var.set(f"成功加载: {len(self.df)} 条记录")
except Exception as e:
self.status_var.set("加载失败")
messagebox.showerror("加载错误", f"读取Excel失败: {str(e)}")
# 在新线程中加载数据
threading.Thread(target=load_task, daemon=True).start()
def create_checkbox(self, item_id, var):
"""为Treeview中的行创建复选框 (修复方法)"""
# 创建Frame作为容器
checkbox_frame = tk.Frame(self.tree)
# 创建复选框
cb = ttk.Checkbutton(
checkbox_frame,
variable=var,
style="Custom.TCheckbutton",
command=lambda: self.toggle_checkbox(item_id, var)
)
cb.pack()
# 正确嵌入到Treeview单元格
self.tree.set(item_id, column="选择", value="") # 清空单元格文本
# 获取"选择"列的索引
columns = self.tree["columns"]
col_index = columns.index("选择")
column_id = f"#{col_index+1}" # 列标识从1开始
# 使用正确的方法嵌入控件
self.tree.window_create(item_id, column=column_id, window=checkbox_frame)
# 更新复选框状态
self.update_checkbox_state(item_id, var)
def toggle_checkbox(self, item_id, var):
"""切换复选框状态"""
# 更新行的状态显示
self.update_checkbox_state(item_id, var)
def update_checkbox_state(self, item_id, var):
"""根据复选框状态更新行状态"""
current_values = list(self.tree.item(item_id, "values"))
if var.get():
current_values[3] = "✓ 待确认"
self.tree.item(item_id, tags=("confirmed",))
else:
current_values[3] = "✗ 未确认"
self.tree.item(item_id, tags=("pending",))
self.tree.item(item_id, values=current_values)
def confirm_selected(self):
"""确认选中的功能点"""
selected_items = []
for item_id in self.tree.get_children():
if self.check_vars[item_id].get():
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:
self.check_vars[item_id].set(True)
current_values = list(self.tree.item(item_id, "values"))
# 更新状态列
current_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:
current_values[i] = custom_status
self.df.at[row_idx, col] = custom_status
# 更新Treeview
self.tree.item(item_id, values=tuple(current_values), tags=("confirmed",))
# 更新复选框状态(触发颜色变化)
self.update_checkbox_state(item_id, self.check_vars[item_id])
self.status_var.set(f"已确认 {len(selected_items)} 个功能点")
# 自动保存
self.auto_save()
def select_all(self):
"""全选功能点"""
for item_id in self.tree.get_children():
var = self.check_vars[item_id]
var.set(True)
self.update_checkbox_state(item_id, var)
self.status_var.set("已全选所有功能点")
def deselect_all(self):
"""取消全选功能点"""
for item_id in self.tree.get_children():
var = self.check_vars[item_id]
var.set(False)
self.update_checkbox_state(item_id, var)
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.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中的状态值(去掉图标)
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("数据已自动保存")
except Exception as e:
self.status_var.set(f"自动保存失败: {str(e)}")
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请确保文件未被其他程序打开")
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:
pass
app = ExcelControlPanel(root)
root.mainloop()
if __name__ == "__main__":
main()
还是会报加载错误 读取Excel失败: ‘Treeview’ object has no attribute ‘window_create’,请更换一种方式进行编辑
最新发布