Pandas ExcelWriter在try except语句中仍生成损坏文件如何处理

问题根因 #

pd.ExcelWriter在实例化时就会直接创建指定路径的空文件，哪怕后续逻辑报错、没有写入有效内容也没有正常关闭写入流，这个空文件已经完成了对原文件的覆盖，就会导致原文件丢失、生成损坏的0KB文件。

最优解决方案：临时文件中转 #

所有写入操作先在临时文件中完成，确认所有逻辑执行无报错、临时文件有效后，再用临时文件替换目标文件。全程操作不会修改原文件，彻底避免覆盖损坏风险。

实现代码 #

import os
import shutil
import pandas as pd
import openpyxl as op

def update_excel(master_file, save_name, table1):
    # 创建同目录下的临时文件路径
    temp_save_path = f"{save_name}.tmp.xlsx"
    try:
        # 加载原工作簿
        wb = op.load_workbook(master_file)
        # 所有写入操作先指向临时文件
        with pd.ExcelWriter(temp_save_path, engine='openpyxl') as writer:
            writer.book = wb
            writer.sheets = {ws.title: ws for ws in wb.worksheets}
            
            table1.to_excel(writer, sheet_name="New Sheet")
            # 其余数据表处理逻辑
            # ...
            
            wb.remove(wb["Data Sheet"])
            wb["New Sheet"].title = "Data Sheet"
        
        # 校验临时文件有效后替换目标文件
        if os.path.exists(temp_save_path) and os.path.getsize(temp_save_path) > 0:
            # 同分区下move为原子操作，不会出现半写入损坏
            shutil.move(temp_save_path, save_name)
        return True
    except Exception as e:
        # 异常时清理临时文件
        if os.path.exists(temp_save_path):
            os.remove(temp_save_path)
        return False

补充说明 #

• 该方案支持覆盖原文件场景：哪怕save_name和master_file路径完全一致，也不会在操作过程中修改原文件，只有全部逻辑执行成功才会替换
• 用上下文管理器with声明ExcelWriter会自动处理写入流关闭，无需手动调用close()，避免资源泄漏
• 如果需要更高安全性，可以在替换文件前新增一步打开临时文件校验逻辑，确认文件可正常读取后再执行替换

内容的提问来源于stack exchange，提问作者Gingerhaze