Python：写入文件并使用缓冲区

Question

我正在使用 django 生成个性化文件，但是这样做时会生成一个文件，并且就空间而言，使用它是一件很糟糕的事情。

这就是我现在的做法：

with open(filename, 'wb') as f:
    pdf.write(f) #pdf is an object of pyPDF2 library

with open(filename, 'rb') as f:
    return send_file(data=f, filename=filename) #send_file is a HTTPResponse parametted to download file data

所以在上面的代码中生成了一个文件。

简单的解决方法是下载后删除文件，但我记得在java中使用流对象来处理这种情况。

在 Python 中可以这样做吗？

编辑：

def send_file(data, filename, mimetype=None, force_download=False):
    disposition = 'attachment' if force_download else 'inline'
    filename = os.path.basename(filename)
    response = HttpResponse(data, content_type=mimetype or 'application/octet-stream')
    response['Content-Disposition'] = '%s; filename="%s"' % (disposition, filename)
    return response

Answer 1

在不知道 pdf.write 和 send_file 函数的确切细节的情况下，我希望在这两种情况下它们都会采用符合 BinaryIO 接口。因此，您可以尝试使用 BytesIO 来存储内容在内存缓冲区中，而不是写入文件：

with io.BytesIO() as buf:
    pdf.write(buf)
    buf.seek(0)
    send_file(data=buf, filename=filename)

取决于上述函数的确切性质，YMMV。