测试代码
import os
import re
import customtkinter as ctk
from datetime import datetime
from tkinter import filedialog, messagebox
# 设置界面主题
ctk.set_appearance_mode("System") # 自动切换深/浅色
ctk.set_default_color_theme("blue")
class WorkReporterGUI(ctk.CTk):
def __init__(self):
super().__init__()
self.title("Markdown 工作汇总生成器")
self.geometry("600x450")
# 变量
self.target_path = ctk.StringVar(value=os.getcwd())
# 创建布局
self.setup_u()
def setup_u(self):
# 标题
self.label_title = ctk.CTkLabel(self, text="个人工作周报汇总工具", font=ctk.CTkFont(size=20, weight="bold"))
self.label_title.pack(pady=20)
# 路径选择区
self.frame_path = ctk.CTkFrame(self)
self.frame_path.pack(pady=10, padx=20, fill="x")
self.entry_path = ctk.CTkEntry(self.frame_path, textvariable=self.target_path, width=400)
self.entry_path.pack(side="left", padx=10, pady=10)
self.btn_browse = ctk.CTkButton(self.frame_path, text="选择目录", width=80, command=self.browse_folder)
self.btn_browse.pack(side="right", padx=10)
# 选项
self.label_hint = ctk.CTkLabel(self, text="提示：将扫描目录下形如 251230.md 的文件", font=ctk.CTkFont(size=12))
self.label_hint.pack(pady=5)
# 运行按钮
self.btn_run = ctk.CTkButton(self, text="开始生成汇总表", command=self.process, height=50, font=ctk.CTkFont(size=16, weight="bold"))
self.btn_run.pack(pady=30)
# 日志输出区
self.log_text = ctk.CTkTextbox(self, height=150)
self.log_text.pack(pady=10, padx=20, fill="both")
def browse_folder(self):
path = filedialog.askdirectory()
if path:
self.target_path.set(path)
def log(self, message):
self.log_text.insert("end", f"[{datetime.now().strftime('%H:%M:%S')}] {message}\n")
self.log_text.see("end")
def get_weekday_cn(self, dt):
return ["周一", "周二", "周三", "周四", "周五", "周六", "周日"][dt.weekday()]
def process(self):
root_dir = self.target_path.get()
output_file = os.path.join(root_dir, "工作汇总表.md")
all_days_data = []
title_pattern = re.compile(r'^###\s+(.+)')
try:
self.log("正在扫描文件...")
for root, dirs, files in os.walk(root_dir):
for file in files:
if file == "工作汇总表.md" or not file.endswith(".md"): continue
if len(file) == 9: # 251230.md
date_part = file.split('.')[0]
try:
dt = datetime.strptime(date_part, "%y%m%d")
except: continue
file_path = os.path.join(root, file)
rel_path = os.path.relpath(file_path, start=root_dir)
events = []
with open(file_path, 'r', encoding='utf-8') as f:
in_code_block = False
for line in f:
if line.strip().startswith('```'):
in_code_block = not in_code_block
continue
if not in_code_block:
match = title_pattern.match(line)
if match: events.append(match.group(1).strip())
if events:
week_num = dt.strftime("%W")
all_days_data.append({
"dt": dt, "year": dt.year, "week_key": f"{dt.year}年-第{week_num}周",
"date_str": dt.strftime("%Y-%m-%d"), "weekday_cn": self.get_weekday_cn(dt),
"rel_path": rel_path, "events": events
})
if not all_days_data:
self.log("未找到有效记录！")
return
# 构建表格
weeks_dict = {}
for day in all_days_data:
weeks_dict.setdefault(day["week_key"], []).append(day)
sorted_week_keys = sorted(weeks_dict.keys(), reverse=True)
table_lines = [f"# 📝 工作汇总 ({datetime.now().strftime('%Y-%m-%d')})\n\n", "| 日期 | 周几 | 序号 | 标题内容 |", "| :--- | :--- | :--- | :--- |"]
last_year = None
for week_key in sorted_week_keys:
days = sorted(weeks_dict[week_key], key=lambda x: x["dt"])
curr_year = days[0]["year"]
if last_year and last_year != curr_year:
table_lines.append(f"| 🎊 | **{last_year} 年结束** | --- | **进入 {curr_year} 年度** |")
for d in days:
for idx, title in enumerate(d["events"], 1):
d_col = f"[{d['date_str']}]({d['rel_path']})" if idx == 1 else ""
w_col = d["weekday_cn"] if idx == 1 else ""
table_lines.append(f"| {d_col} | {w_col} | {idx} | {title} |")
table_lines.append(f"| --- | --- | --- | **{week_key} 回顾** |")
last_year = curr_year
with open(output_file, "w", encoding="utf-8") as f:
f.write("\n".join(table_lines))
self.log(f"成功！汇总表已生成。")
if messagebox.askyesno("完成", "汇总表已生成！是否立即打开查看？"):
os.startfile(output_file) if os.name == 'nt' else os.system(f'open "{output_file}"')
except Exception as e:
self.log(f"发生错误: {str(e)}")
messagebox.showerror("错误", str(e))
if __name__ == "__main__":
app = WorkReporterGUI()
app.mainloop()