Django实现pandas DataFrame转PNG下载的tempfile临时文件集成问题

你最初的tempfile版本逻辑存在3个核心错误，是导致字节类型报错、文件无法正常返回的原因：

• 误将df2img.plot_dataframe()返回的Plotly绘图对象直接写入临时文件，该方法返回值不是图片二进制流，不支持直接write操作
• df2img.save_dataframe()是无返回值的磁盘写入方法，把它的执行结果（None）作为响应体传给HttpResponse，不可能返回有效文件
• 跨平台场景下NamedTemporaryFile默认开启独占锁，未设置delete=False时文件句柄关闭就会自动删除，后续写入操作会直接报错

你后续改用的项目目录存文件+手动删除的方案存在明显隐患：请求处理抛异常、进程意外中断时删除逻辑不会执行，会在项目目录残留大量垃圾文件；并发场景下文件名冲突还会导致文件覆盖、返回内容错误。

基于tempfile的稳定实现 #

这个方案完全不会污染项目目录，且保证任何场景下临时文件都会被清理：

import os
import tempfile
import mimetypes
from django.http import FileResponse

def download_png(request, study):
    Names(study)  # 内部逻辑生成name变量
    Retrieve(study)  # 内部逻辑生成待导出的data变量
    # 自行补全current_time的生成逻辑
    pngfilename = f"{name}_{current_time}.png"
    mime_type, _ = mimetypes.guess_type(pngfilename)

    # 创建临时PNG文件，关闭后不自动删除，方便后续读取和手动清理
    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_fp:
        temp_file_path = temp_fp.name

    # 生成绘图对象并存入临时文件路径
    fig = df2img.plot_dataframe(data)
    df2img.save_dataframe(fig=fig, filename=temp_file_path)

    try:
        # 流式返回文件，大文件不占内存
        response = FileResponse(open(temp_file_path, "rb"), content_type=mime_type)
        response["Content-Disposition"] = f"attachment; filename={pngfilename}"
        return response
    finally:
        # 无论请求成功或报错，最终都删除临时文件
        os.unlink(temp_file_path)

更高性能的内存实现（无磁盘IO） #

如果追求更好的性能，可以完全跳过磁盘操作，直接在内存中生成图片返回，没有任何文件残留风险：

from io import BytesIO
import mimetypes
from django.http import HttpResponse

def download_png(request, study):
    Names(study)
    Retrieve(study)
    # 自行补全current_time的生成逻辑
    pngfilename = f"{name}_{current_time}.png"
    mime_type, _ = mimetypes.guess_type(pngfilename)

    fig = df2img.plot_dataframe(data)
    # 初始化内存字节流
    img_buffer = BytesIO()
    # 直接将图片写入内存流
    df2img.save_dataframe(fig=fig, filename=img_buffer)
    # 将流指针移到起始位置，方便读取
    img_buffer.seek(0)

    response = HttpResponse(img_buffer, content_type=mime_type)
    response["Content-Disposition"] = f"attachment; filename={pngfilename}"
    return response

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