zyy/cwExcel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
zhuyiyi
2026-07-18 15:04:19 +08:00
commit 7fdcab3405
26 changed files with 2749 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/ui/main_window.py
+676
View File
@@ -0,0 +1,676 @@
"""
主窗口 UI 组件
"""
import os
import sys
import threading
import tkinter as tk
from tkinter import ttk, filedialog, messagebox
import openpyxl
# ---- 配色 ----
C = {
"bg": "#f0f2f5",
"white": "#ffffff",
"text": "#1a1a2e",
"text_muted": "#6b7280",
"accent": "#4f46e5",
"accent_hover": "#4338ca",
"accent_light": "#eef2ff",
"border": "#e5e7eb",
"success": "#059669",
}
# 列宽
COL_W = [220, 1, 220, 1, 220, 1, 200]
COL_X = [0]
for w in COL_W[:-1]:
COL_X.append(COL_X[-1] + w)
ROW_H = 44
CELL_PAD = 8
# ======================================================================
# 自定义下拉组件
# ======================================================================
class Dropdown(tk.Frame):
"""纯自定义下拉选择器 — 不用任何系统 Combobox"""
def __init__(self, parent, values=None, default="", on_select=None,
font=("Microsoft YaHei", 10), width=180, **kw):
super().__init__(parent, bg=C["white"], highlightbackground=C["border"],
highlightthickness=1, cursor="hand2", **kw)
self.values = values or []
self._selected = default or ""
self._on_select = on_select
self._font = font
self._popup = None
self._hover = False
# 文字标签
self._label = tk.Label(
self, text=self._selected or "请选择",
font=font, bg=C["white"], fg=C["text"] if self._selected else C["text_muted"],
anchor=tk.W, padx=8,
)
self._label.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
# 箭头
self._arrow = tk.Label(
self, text="▾", font=("Microsoft YaHei", 8),
bg=C["white"], fg="#9ca3af", padx=6,
)
self._arrow.pack(side=tk.RIGHT, fill=tk.Y)
# 事件绑定
for w in (self, self._label, self._arrow):
w.bind("<Button-1>", self._toggle)
w.bind("<Enter>", self._on_enter)
w.bind("<Leave>", self._on_leave)
def _on_enter(self, _):
self._hover = True
self.configure(highlightbackground=C["accent"])
self._label.configure(bg="#f9fafb")
self._arrow.configure(bg="#f9fafb", fg=C["accent"])
def _on_leave(self, _):
self._hover = False
if not self._popup:
self.configure(highlightbackground=C["border"])
self._label.configure(bg=C["white"])
self._arrow.configure(bg=C["white"], fg="#9ca3af")
def _toggle(self, event=None):
if self._popup:
self._close()
else:
self._open()
def _open(self):
self.configure(highlightbackground=C["accent"])
self._label.configure(bg="#f9fafb")
self._arrow.configure(bg="#f9fafb", fg=C["accent"])
self._popup = tk.Toplevel(self)
self._popup.overrideredirect(True)
self._popup.configure(bg=C["white"])
# 计算位置
x = self.winfo_rootx()
y = self.winfo_rooty() + self.winfo_height() + 1
w = self.winfo_width()
item_h = 32
max_h = min(len(self.values) * item_h, 240)
list_h = max(item_h, max_h)
self._popup.geometry(f"{w}x{list_h}+{x}+{y}")
# 外边框
border = tk.Frame(self._popup, bg=C["border"])
border.pack(fill=tk.BOTH, expand=True)
# 列表
self._listbox = tk.Listbox(
border, font=self._font,
bg=C["white"], fg=C["text"],
selectbackground=C["accent_light"],
selectforeground=C["accent"],
selectborderwidth=0,
activestyle="none",
highlightthickness=0,
borderwidth=0,
relief=tk.FLAT,
height=len(self.values),
)
self._listbox.pack(fill=tk.BOTH, expand=True, padx=1, pady=1)
for i, v in enumerate(self.values):
self._listbox.insert(tk.END, f" {v}")
if v == self._selected:
self._listbox.selection_set(i)
self._listbox.see(i)
self._listbox.bind("<ButtonRelease-1>", self._on_pick)
self._listbox.bind("<Return>", self._on_pick)
self._listbox.bind("<Escape>", lambda e: self._close())
self._listbox.focus_set()
# 点击外部关闭
self._popup.bind("<FocusOut>", lambda e: self.after(100, self._check_focus))
# 绑定全局点击关闭（鼠标点击 popup 外）
self._grab_bind = self.winfo_toplevel().bind(
"<Button-1>", lambda e: self._on_global_click(e), add="+"
)
def _on_global_click(self, event):
if self._popup:
px, py = event.x_root, event.y_root
px2 = self._popup.winfo_rootx()
py2 = self._popup.winfo_rooty()
pw = self._popup.winfo_width()
ph = self._popup.winfo_height()
if not (px2 <= px <= px2 + pw and py2 <= py <= py2 + ph):
self._close()
def _check_focus(self):
if self._popup and not self._popup.focus_displayof():
self._close()
def _close(self, *_):
if self._popup:
self._popup.destroy()
self._popup = None
self._listbox = None
self.configure(highlightbackground=C["border"])
self._label.configure(bg=C["white"])
self._arrow.configure(bg=C["white"], fg="#9ca3af")
# 移除全局绑定
try:
self.winfo_toplevel().unbind("<Button-1>", self._grab_bind)
except Exception:
pass
def _on_pick(self, event):
sel = self._listbox.curselection()
if sel:
idx = sel[0]
self._selected = self.values[idx]
self._label.configure(text=self._selected, fg=C["text"])
if self._on_select:
self._on_select(self._selected)
self._close()
# ── 公开 API ──
def get(self):
return self._selected
def set_values(self, values, default=None):
self.values = values
if default is not None:
self._selected = default
self._label.configure(text=default, fg=C["text"] if default else C["text_muted"])
# 如果已打开则重建
if self._popup:
self._close()
self._open()
# ======================================================================
# 主窗口
# ======================================================================
class MainWindow:
"""应用程序主窗口"""
def __init__(self, root: tk.Tk):
self.root = root
self.root.configure(bg=C["bg"])
self._files_data = []
self._build_ui()
def _build_ui(self):
# ── 选择卡片 ──
card = tk.Frame(self.root, bg=C["white"])
card.pack(fill=tk.X, padx=24, pady=(20, 0))
card_border = tk.Frame(card, bg=C["white"],
highlightbackground=C["border"],
highlightthickness=1)
card_border.pack(fill=tk.X)
tk.Label(
card_border, text="📂 选择 Excel 文件",
font=("Microsoft YaHei", 12, "bold"),
bg=C["white"], fg=C["text"], anchor=tk.W,
).pack(fill=tk.X, padx=20, pady=(16, 10))
input_row = tk.Frame(card_border, bg=C["white"])
input_row.pack(fill=tk.X, padx=20, pady=(0, 10))
entry_border = tk.Frame(
input_row, bg=C["white"],
highlightbackground=C["border"], highlightthickness=1,
)
entry_border.pack(side=tk.LEFT, fill=tk.X, expand=True)
self._path_entry = tk.Entry(
entry_border,
font=("Microsoft YaHei", 10),
bg="#f9fafb", fg=C["text_muted"],
relief=tk.FLAT, borderwidth=0,
readonlybackground="#f9fafb",
)
self._path_entry.pack(fill=tk.X, padx=10, pady=8)
self._path_entry.insert(0, "请选择 Excel 文件（.xlsx / .xls，可多选）")
self._path_entry.configure(state="readonly")
self._browse_btn = tk.Button(
input_row, text="浏览...",
font=("Microsoft YaHei", 10, "bold"),
bg=C["accent"], fg="white", relief=tk.FLAT,
activebackground=C["accent_hover"], activeforeground="white",
cursor="hand2", padx=20, pady=7, borderwidth=0,
command=self._on_select_files,
)
self._browse_btn.pack(side=tk.LEFT, padx=(8, 0))
status_bar = tk.Frame(card_border, bg="#f9fafb")
status_bar.pack(fill=tk.X)
self._status_label = tk.Label(
status_bar, text=" 未选择任何文件",
font=("Microsoft YaHei", 9),
bg="#f9fafb", fg=C["text_muted"], anchor=tk.W,
)
self._status_label.pack(side=tk.LEFT, padx=16, pady=(6, 6))
# ── 表格区域 ──
self._table_area = tk.Frame(self.root, bg=C["bg"])
self._table_area.pack(fill=tk.BOTH, expand=True, padx=24, pady=(12, 0))
# ── 日志面板（底部固定高度，内部独立滚动） ──
self._log_frame = tk.Frame(self.root, bg="white", height=130)
self._log_frame.pack(fill=tk.X, padx=24, pady=(8, 8))
self._log_frame.pack_propagate(False)
log_header = tk.Frame(self._log_frame, bg="white")
log_header.pack(fill=tk.X)
tk.Label(log_header, text="📋 处理日志",
font=("Microsoft YaHei", 9, "bold"),
bg="white", fg=C["text"], anchor=tk.W,
).pack(side=tk.LEFT, padx=10, pady=(6, 2))
log_text_frame = tk.Frame(self._log_frame, bg="white")
log_text_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=(0, 8))
self._log_text = tk.Text(
log_text_frame,
font=("Consolas", 9),
bg="#f9fafb", fg=C["text"],
wrap=tk.WORD, state=tk.DISABLED,
relief=tk.FLAT, borderwidth=0,
height=6,
)
log_scrollbar = tk.Scrollbar(log_text_frame, orient=tk.VERTICAL,
command=self._log_text.yview)
self._log_text.configure(yscrollcommand=log_scrollbar.set)
log_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
self._log_text.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
self._show_empty()
# ==================================================================
# 空状态
# ==================================================================
def _show_empty(self):
for w in self._table_area.winfo_children():
w.destroy()
drop = tk.Frame(self._table_area, bg="#fafafe",
highlightbackground="#c7d2fe",
highlightthickness=2)
drop.pack(fill=tk.BOTH, expand=True)
inner = tk.Frame(drop, bg="#fafafe")
inner.place(relx=0.5, rely=0.5, anchor=tk.CENTER)
tk.Label(inner, text="📂", font=("Segoe UI Emoji", 40),
bg="#fafafe", fg="#c7d2fe").pack(pady=(0, 4))
tk.Label(inner, text="点击上方「浏览...」按钮选择文件",
font=("Microsoft YaHei", 13, "bold"),
bg="#fafafe", fg=C["text_muted"]).pack()
tk.Label(inner, text="支持 .xlsx、.xls 格式，可同时选择多个文件",
font=("Microsoft YaHei", 9),
bg="#fafafe", fg="#c0c4cc").pack(pady=(2, 0))
# ==================================================================
# 读取 xlsx 信息（用内置 zipfile，避免 openpyxl 打包问题）
# ==================================================================
def _read_xlsx_info(self, filepath):
import pandas as pd
sheet_names = []
columns = []
for engine in ["openpyxl", "xlrd"]:
try:
xls = pd.ExcelFile(filepath, engine=engine)
sheet_names = xls.sheet_names
if sheet_names:
df = pd.read_excel(xls, sheet_name=sheet_names[0], nrows=1)
columns = [str(c) for c in df.columns.tolist()]
xls.close()
break
except Exception:
continue
return sheet_names, columns
# ==================================================================
# 选择文件
# ==================================================================
def _on_select_files(self):
paths = filedialog.askopenfilenames(
title="选择 Excel 文件",
filetypes=[
("Excel 文件", "*.xlsx *.xls"),
("Excel 2007+", "*.xlsx"),
("Excel 97-2003", "*.xls"),
],
)
if not paths:
return
for p in paths:
if any(d["path"] == p for d in self._files_data):
continue
try:
sheet_names, columns = self._read_xlsx_info(p)
first_sheet = sheet_names[0] if sheet_names else None
self._files_data.append({
"path": p,
"filename": os.path.basename(p),
"sheet_names": sheet_names,
"columns": columns,
"selected_sheet": first_sheet,
"selected_col": columns[0] if columns else None,
"parse_mode": "横向排列",
})
except Exception as e:
messagebox.showerror("读取失败", f"无法读取文件:\n{os.path.basename(p)}\n\n{e}")
continue
n = len(self._files_data)
if n == 0:
self._path_entry.configure(state="normal")
self._path_entry.delete(0, tk.END)
self._path_entry.insert(0, "请选择 Excel 文件（.xlsx / .xls，可多选）")
self._path_entry.configure(state="readonly", fg=C["text_muted"])
self._status_label.configure(text=" 未选择任何文件")
elif n == 1:
self._path_entry.configure(state="normal")
self._path_entry.delete(0, tk.END)
self._path_entry.insert(0, self._files_data[0]["path"])
self._path_entry.configure(state="readonly", fg=C["text"])
self._status_label.configure(
text=f" 已选择 1 个文件 | 点击「浏览...」可继续添加",
fg=C["success"],
)
else:
self._path_entry.configure(state="normal")
self._path_entry.delete(0, tk.END)
self._path_entry.insert(0, f"已选择 {n} 个文件")
self._path_entry.configure(state="readonly", fg=C["text"])
self._status_label.configure(
text=f" 已选择 {n} 个文件 | 点击「浏览...」可继续添加",
fg=C["success"],
)
if self._files_data:
self._build_table()
else:
self._show_empty()
# ==================================================================
# 构建表格
# ==================================================================
def _build_table(self):
for w in self._table_area.winfo_children():
w.destroy()
self._table_frame = tk.Frame(self._table_area, bg=C["bg"])
self._table_frame.pack(fill=tk.BOTH, expand=True)
self._draw_header()
for fi, fd in enumerate(self._files_data):
self._draw_row(fi, fd)
# ── 确认按钮 ──
btn_bar = tk.Frame(self._table_frame, bg=C["bg"])
btn_bar.pack(fill=tk.X, pady=(16, 0))
self._confirm_btn = tk.Button(
btn_bar, text="✅ 开始解析",
font=("Microsoft YaHei", 12, "bold"),
bg=C["accent"], fg="white", relief=tk.FLAT,
activebackground=C["accent_hover"], activeforeground="white",
cursor="hand2", padx=28, pady=10, borderwidth=0,
command=self._on_confirm,
)
self._confirm_btn.pack(side=tk.RIGHT)
self._progress_label = tk.Label(
btn_bar, text="",
font=("Microsoft YaHei", 9),
bg=C["bg"], fg=C["text_muted"], anchor=tk.E,
)
self._progress_label.pack(side=tk.RIGHT, padx=(0, 12))
def _draw_header(self):
hdr = tk.Frame(self._table_frame, bg=C["white"], height=38)
hdr.pack(fill=tk.X)
hdr.pack_propagate(False)
headers = ["文件名", "工作表", "解析列", "解析方式"]
for i, title in enumerate(headers):
col_idx = i * 2
x, w = COL_X[col_idx], COL_W[col_idx]
tk.Label(
hdr, text=title, font=("Microsoft YaHei", 10, "bold"),
bg=C["white"], fg=C["text"], anchor=tk.W,
).place(x=x + CELL_PAD, y=0, width=w - CELL_PAD * 2, height=38)
if i < 3:
tk.Frame(hdr, bg=C["border"], width=1).place(
x=COL_X[col_idx + 1], y=6, height=26
)
tk.Frame(self._table_frame, bg=C["border"], height=1).pack(fill=tk.X)
def _draw_row(self, fi: int, fd: dict):
row = tk.Frame(self._table_frame, bg=C["white"], height=ROW_H)
row.pack(fill=tk.X)
row.pack_propagate(False)
# ── 文件名 ──
x, w = COL_X[0], COL_W[0]
tk.Label(
row, text=f"📄 {fd['filename']}",
font=("Microsoft YaHei", 10), bg=C["white"], fg=C["text"], anchor=tk.W,
).place(x=x + CELL_PAD, y=0, width=w - CELL_PAD * 2, height=ROW_H)
tk.Frame(row, bg=C["border"], width=1).place(x=COL_X[1], y=6, height=ROW_H - 12)
# ── 工作表（自定义 Dropdown） ──
x, w = COL_X[2], COL_W[2]
dd_sheet = Dropdown(
row,
values=fd["sheet_names"],
default=fd["selected_sheet"] or "",
on_select=lambda val, i=fi: self._on_sheet_change(i, val),
)
dd_sheet.place(x=x + CELL_PAD, y=6, width=w - CELL_PAD * 2, height=ROW_H - 12)
fd["sheet_dd"] = dd_sheet
tk.Frame(row, bg=C["border"], width=1).place(x=COL_X[3], y=6, height=ROW_H - 12)
# ── 解析列（自定义 Dropdown） ──
x, w = COL_X[4], COL_W[4]
dd_col = Dropdown(
row,
values=fd["columns"],
default=fd["selected_col"] or "",
on_select=lambda val, i=fi: self._on_col_change(i, val),
)
dd_col.place(x=x + CELL_PAD, y=6, width=w - CELL_PAD * 2, height=ROW_H - 12)
fd["col_dd"] = dd_col
tk.Frame(row, bg=C["border"], width=1).place(x=COL_X[5], y=6, height=ROW_H - 12)
# ── 解析方式（自定义 Dropdown） ──
x, w = COL_X[6], COL_W[6]
dd_mode = Dropdown(
row,
values=["横向排列", "纵向排序"],
default=fd["parse_mode"],
on_select=lambda val, i=fi: self._on_mode_change(i, val),
)
dd_mode.place(x=x + CELL_PAD, y=6, width=w - CELL_PAD * 2, height=ROW_H - 12)
fd["mode_dd"] = dd_mode
fd["row_frame"] = row
tk.Frame(self._table_frame, bg=C["border"], height=1).pack(fill=tk.X)
# ==================================================================
# 切换工作表 → 更新解析列
# ==================================================================
def _on_sheet_change(self, idx: int, new_sheet: str):
fd = self._files_data[idx]
fd["selected_sheet"] = new_sheet
try:
import pandas as pd
for engine in ["openpyxl", "xlrd"]:
try:
df = pd.read_excel(fd["path"], sheet_name=new_sheet, nrows=1, engine=engine)
columns = [str(c) for c in df.columns.tolist()]
break
except Exception:
continue
fd["columns"] = columns
first_col = columns[0] if columns else None
fd["selected_col"] = first_col
fd["col_dd"].set_values(columns, default=first_col)
except Exception as e:
messagebox.showerror("错误", f"读取工作表失败: {e}")
# ==================================================================
# 下拉选择回调
# ==================================================================
def _on_col_change(self, idx: int, val: str):
self._files_data[idx]["selected_col"] = val
def _on_mode_change(self, idx: int, val: str):
self._files_data[idx]["parse_mode"] = val
# ==================================================================
# 日志输出
# ==================================================================
def _log_write(self, text: str):
"""向底部日志面板追加一行"""
self._log_text.configure(state=tk.NORMAL)
self._log_text.insert(tk.END, text + "\n")
self._log_text.see(tk.END)
self._log_text.configure(state=tk.DISABLED)
def _log_clear(self):
"""清空日志面板"""
self._log_text.configure(state=tk.NORMAL)
self._log_text.delete("1.0", tk.END)
self._log_text.configure(state=tk.DISABLED)
# ==================================================================
# 确认解析
# ==================================================================
def _on_confirm(self):
if not self._files_data:
return
self._confirm_btn.configure(state=tk.DISABLED, text="⏳ 解析中...")
self._log_clear()
self._log_write("🚀 开始解析...")
# 捕获 print 输出到日志面板
class _LogWriter:
def __init__(self, root, log_func):
self._root = root
self._log = log_func
self._buf = ""
def write(self, s):
self._buf += s
if "\n" in self._buf:
lines = self._buf.split("\n")
self._buf = lines[-1]
for line in lines[:-1]:
stripped = line.strip()
if stripped:
self._root.after(0, lambda t=stripped: self._log(t))
def flush(self):
if self._buf.strip():
self._root.after(0, lambda t=self._buf.strip(): self._log(t))
self._buf = ""
def _run():
import sys
from app import finance_parse as fp
total = len(self._files_data)
success = 0
old_stdout = sys.stdout
old_stderr = sys.stderr
log_writer = _LogWriter(self.root, self._log_write)
sys.stdout = log_writer
sys.stderr = open(os.devnull, "w")
try:
for i, fd in enumerate(self._files_data):
fname = fd["filename"]
self.root.after(0, lambda n=i+1, fn=fname:
self._log_write(f"⏳ [{n}/{total}] 正在处理: {fn}"))
try:
save_path, amount = fp.parse_finance_data(
file_path=fd["path"],
target_index=fd["col_dd"].get(),
is_horizontal=(fd["mode_dd"].get() == "横向排列"),
sheet_name=fd["sheet_dd"].get(),
)
success += 1
fd["result"] = save_path
fd["amount"] = amount
self.root.after(0, lambda fn=fname, sp=save_path, amt=amount:
self._log_write(f"✅ [{success}/{total}] 完成: {fn} 共 {amt} 条"))
except Exception as e:
fd["error"] = str(e)
self.root.after(0, lambda fn=fname, err=str(e):
self._log_write(f"❌ [{i+1}/{total}] 失败: {fn} — {err}"))
finally:
sys.stdout = old_stdout
sys.stderr = old_stderr
self.root.after(0, lambda: self._on_parse_done(success, total))
threading.Thread(target=_run, daemon=True).start()
def _on_parse_done(self, success: int, total: int):
self._confirm_btn.configure(state=tk.NORMAL, text="✅ 开始解析")
# 从列表中移除已成功解析的文件，保留失败的
self._files_data = [fd for fd in self._files_data if "error" in fd]
if success == total:
self._log_write(f"🎉 全部 {total} 个文件解析完成！结果已保存到原文件所在目录。")
messagebox.showinfo("解析完成",
f"全部 {total} 个文件解析完成！\n\n结果已保存到原文件所在目录。")
else:
failed_count = len(self._files_data)
self._log_write(f"⚠️ 完成 {success}/{total}，有 {failed_count} 个文件失败。")
errors = []
for fd in self._files_data:
if "error" in fd:
errors.append(f"• {fd['filename']}: {fd['error']}")
msg = f"完成 {success}/{total}\n\n失败:\n" + "\n".join(errors[:5])
messagebox.showwarning("解析完成（部分失败）", msg)
# 刷新 UI：清除成功项后重建表格或显示空状态
if self._files_data:
self._build_table()
n = len(self._files_data)
self._status_label.configure(
text=f" 剩余 {n} 个失败文件 | 可调整配置后重新解析",
fg="#dc2626",
)
else:
self._show_empty()
self._path_entry.configure(state="normal")
self._path_entry.delete(0, tk.END)
self._path_entry.insert(0, "请选择 Excel 文件（.xlsx / .xls，可多选）")
self._path_entry.configure(state="readonly", fg=C["text_muted"])
self._status_label.configure(text=" 未选择任何文件")