程序崩溃时自动删除安全文件的Python实现方法咨询

当然可以用Python实现，针对不同操作系统有不同的方案，下面是具体实现方式：

Unix/Linux/macOS 下的极简方案 #

Unix类系统的文件机制允许我们在打开文件后立即删除它，只要文件句柄保持打开，程序就能正常读写内容；一旦进程终止（不管是正常退出还是崩溃），系统会自动清理这个已被标记删除的文件，完全不需要手动处理。

示例代码：

import os

def create_secure_file():
    # 创建独占式文件，权限设为仅当前用户可读写
    fd, file_path = os.open(
        "/tmp/secure_temp",
        os.O_RDWR | os.O_CREAT | os.O_EXCL,
        0o600
    )
    # 将文件描述符转为文件对象
    secure_file = os.fdopen(fd, "r+")
    
    # 立即删除文件路径
    os.unlink(file_path)
    
    # 正常读写文件
    secure_file.write("你的敏感内容")
    secure_file.seek(0)
    print(secure_file.read())
    
    # 无需手动关闭删除，进程结束后系统自动回收

跨平台兼容方案（覆盖多数崩溃场景） #

如果需要支持Windows（Windows不允许删除正在打开的文件），可以结合atexit和信号处理来覆盖大部分程序终止场景，虽然无法处理被强制杀死（如kill -9）或系统崩溃的极端情况，但能应对绝大多数异常崩溃：

示例代码：

import atexit
import os
import signal
import tempfile

secure_file_path = None

def clean_secure_file():
    """注册的清理函数，负责删除安全文件"""
    if secure_file_path and os.path.exists(secure_file_path):
        os.remove(secure_file_path)

def handle_termination_signal(signum, frame):
    """捕获终止信号，执行清理后退出"""
    clean_secure_file()
    exit(1)

# 注册程序退出时的清理函数
atexit.register(clean_secure_file)
# 捕获常见的终止信号（如Ctrl+C、kill命令）
signal.signal(signal.SIGINT, handle_termination_signal)
signal.signal(signal.SIGTERM, handle_termination_signal)

# 创建临时文件（不自动删除）
with tempfile.NamedTemporaryFile(delete=False, mode="r+", dir="/tmp") as f:
    secure_file_path = f.name
    f.write("敏感数据内容")
    # 模拟程序崩溃：raise Exception("意外崩溃！")

关于tempfile模块的补充说明 #

你提到tempfile需要显式关闭才删除，其实tempfile.NamedTemporaryFile(delete=True)在Unix环境下，即使程序崩溃（如未捕获的异常），进程终止时文件句柄会被系统自动关闭，文件也会被删除；但在Windows环境下，若程序异常崩溃导致文件句柄未正常关闭，delete=True的临时文件不会自动删除，这也是为什么你觉得它不适用的原因。

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