Python中KeyboardInterrupt为何导致Pickle保存失败且指定代码未执行？

1. Pickle文件损坏（"ran out of input"错误） #

你推测的完全正确：当pickle.dump()执行过程中触发KeyboardInterrupt时，由于用"wb"模式直接打开目标文件，文件会被立即清空，而dump中途中断导致仅部分数据写入，最终生成不完整的损坏文件。后续读取时，pickle无法解析残缺数据，就会抛出"ran out of input"错误。

2. "help me"未执行的原因 #

有两种大概率情况：

• 触发第一次KeyboardInterrupt后，你又快速按了一次Ctrl+C，导致第二次中断发生在except KeyboardInterrupt: pass之后、print('help me')执行之前，程序直接强制退出。
• 当save_pickle函数内部（比如pickle.dump阶段）触发KeyboardInterrupt时，异常向上传递到func1的except块，执行pass后本应继续执行后续代码，但未正确关闭的文件句柄引发资源异常，导致程序直接终止，跳过了print和后续保存步骤。

1. 安全的Pickle保存方法 #

改用临时文件写入+原子替换的方式，确保只有完整数据写入完成后才替换原文件，从根源避免原文件损坏：

import pickle
import os
import tempfile

def save_pickle_safe(obj, file_path):
    # 在目标文件同目录创建临时文件，避免跨文件系统移动的原子性问题
    with tempfile.NamedTemporaryFile(mode='wb', dir=os.path.dirname(file_path), delete=False) as temp_file:
        pickle.dump(obj, temp_file)
    # 系统层面的原子替换操作，不会出现文件损坏的中间状态
    os.replace(temp_file.name, file_path)

即便dump过程中中断，损坏的也只是临时文件，原文件始终保持完好。

2. 完善异常处理与代码健壮性 #

• 用with语句管理文件句柄，确保无论是否发生异常，文件都会自动关闭。
• 调整异常捕获逻辑，确保中断后后续代码稳定执行，同时修复原代码中未定义全局变量的问题：

import time

class HelpMe():
    def __init__(self, save_obj, save_path):
        self.save_obj = save_obj  # 将需保存的对象和路径设为实例属性，避免依赖全局变量
        self.save_path = save_path

    def func1(self):
        try:
            self.func2()
        except KeyboardInterrupt:
            print("检测到中断，正在执行最终保存...")
        # 无论是否触发中断，都会执行以下代码
        print('help me')
        save_pickle_safe(self.save_obj, self.save_path)

    def func2(self):
        for i in range(600_000):
            time.sleep(1)
            if i % 100 == 0:
                save_pickle_safe(self.save_obj, self.save_path)

3. 额外建议 #

• 循环中频繁保存时，可增加保存成功的日志输出，方便后续排查问题。
• 如果待保存的obj体积较大，pickle.dump耗时较长，可考虑使用joblib等更高效的序列化库，或拆分数据分块保存。

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