diff --git a/merge_transactions.py b/merge_transactions.py new file mode 100644 index 0000000..7573825 --- /dev/null +++ b/merge_transactions.py @@ -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)