如何实现临时文件随对象垃圾回收销毁？解决Python临时文件提前删除问题

原代码中from_data方法使用with tempfile.NamedTemporaryFile()创建临时文件，with块结束后文件会被立即关闭并删除，导致返回的A实例访问data属性时文件已不存在。

核心思路是让临时文件的生命周期与A实例绑定，只有当实例被垃圾回收时，临时文件才被删除。以下是两种适配不同场景的实现方案：

方案一：通用实现（非Windows系统优先） #

通过保留临时文件句柄，让句柄与实例绑定，依赖文件句柄被销毁时自动删除文件的特性：

from pickle import load
import tempfile


class A:
    def __init__(self, file_path, tmp_handle=None):
        self.file_path = file_path
        self._tmp_handle = tmp_handle  # 保留临时文件句柄，阻止提前回收

    @property
    def data(self):
        with open(self.file_path, "rb") as f2r:
            return load(f2r)

    @classmethod
    def from_data(cls, data):
        # 不使用with语句，避免句柄提前关闭
        tmp_file = tempfile.NamedTemporaryFile(delete=True)
        tmp_file.write(data)
        tmp_file.seek(0)  # 重置文件指针到开头，确保读取正常
        return cls(tmp_file.name, tmp_handle=tmp_file)

说明： #

• _tmp_handle属性保存临时文件的句柄，只要A实例存在，该句柄就不会被垃圾回收，临时文件也不会被关闭删除。
• 当A实例被垃圾回收时，_tmp_handle被销毁，临时文件句柄关闭，NamedTemporaryFile默认会自动删除文件。
• 写入数据后必须调用seek(0)，否则后续读取文件时指针在末尾，会导致load失败。

方案二：Windows兼容实现 #

Windows系统中NamedTemporaryFile默认以独占模式打开，不关闭句柄无法再次读取文件，因此需要手动管理文件删除：

import os
from pickle import load
import tempfile


class A:
    def __init__(self, file_path, is_temporary=False):
        self.file_path = file_path
        self._is_temporary = is_temporary

    @property
    def data(self):
        with open(self.file_path, "rb") as f2r:
            return load(f2r)

    @classmethod
    def from_data(cls, data):
        # 设置delete=False，避免文件被自动删除
        tmp_file = tempfile.NamedTemporaryFile(delete=False)
        tmp_file.write(data)
        tmp_file.seek(0)
        tmp_file.close()  # 关闭句柄，允许后续读取
        return cls(tmp_file.name, is_temporary=True)

    def __del__(self):
        # 实例被回收时手动删除临时文件
        if self._is_temporary:
            try:
                os.remove(self.file_path)
            except (OSError, FileNotFoundError):
                # 文件已不存在时忽略异常
                pass

说明： #

• 使用delete=False创建临时文件，关闭句柄后文件不会自动删除，确保后续能正常读取。
• 通过_is_temporary标记实例是否由临时文件创建，在__del__方法中手动删除文件。
• 捕获删除时的异常，避免因文件已被删除等情况引发报错。

可选优化：手动资源清理 #

若需要更可靠的资源控制，可结合上下文管理器实现手动清理：

class A:
    # ... 继承上述方案的代码 ...

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self._is_temporary:
            try:
                os.remove(self.file_path)
            except (OSError, FileNotFoundError):
                pass

使用方式：

with A.from_data(serialized_data) as a:
    print(a.data)
# 退出with块时自动删除临时文件

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