feat: 添加脚本 merge_transactions.py 用于合并指定目录下的 Excel 文件中的“交易流水”工作表，并输出到同一目录下。

Co-authored-by: aider (ollama/gemma4:12b) <aider@aider.chat>
2026-06-08 06:41:26 +08:00
parent 9ba8306b8d
commit 4a43fbc895
1 changed files with 59 additions and 0 deletions
@@ -0,0 +1,59 @@
+import os
+import pandas as pd
+import glob
+
+def merge_excel_files(input_dir):
+    # 获取绝对路径并确保它是存在的
+    input_dir = os.path.abspath(input_dir)
+    if not os.path.isdir(input_dir):
+        print(f"错误：路径 '{input_dir}' 不是有效的目录。")
+        return
+
+    # 获取目录下所有的 excel 文件
+    # 支持 .xlsx 和 .xls 格式
+    file_pattern = os.path.join(input_dir, "*.xls*")
+    files = glob.glob(file_pattern)
+    
+    all_data = []
+    target_sheet = "交易流水"
+
+    for file_path in files:
+        # 跳过输出文件本身，防止循环读取
+        if "txn_summary" in os.path.basename(file_path):
+            continue
+            
+        try:
+            # 加载 Excel 文件
+            # 使用 ExcelFile 对象可以先检查 sheet 名称是否存在
+            xls = pd.ExcelFile(file_path)
+            if target_sheet in xls.sheet_names:
+                df = pd.read_excel(xls, sheet_name=target_sheet)
+                # 记录来源文件名（可选，方便核对）
+                df['source_file'] = os.path.basename(file_path)
+                all_data.append(df)
+                print(f"成功读取: {os.path.basename(file_path)}")
+            else:
+                print(f"跳过文件 {os.path.basename(file_path)}: 未找到名为 '{target_sheet}' 的工作表。")
+        except Exception as e:
+            print(f"处理文件 {os.path.basename(file_path)} 时出错: {e}")
+
+    if all_data:
+        # 合并所有 DataFrame
+        merged_df = pd.concat(all_data, ignore_index=True)
+        
+        # 生成输出路径
+        output_path = os.path.join(input_dir, "txn_summary.xlsx")
+        
+        # 保存到 Excel
+        with pd.ExcelWriter(output_path, engine='openpyxl') as writer:
+            merged_df.to_excel(writer, sheet_name='Summary', index=False)
+        
+        print("-" * 30)
+        print(f"合并完成！结果已保存至: {output_path}")
+    else:
+        print("未找到任何包含 '交易流水' 工作表的 Excel 文件。")
+
+if __name__ == "__main__":
+    print("请输入包含 Excel 文件的目录路径:")
+    user_path = input("> ").strip()
+    merge_excel_files(user_path)