如何使用 FastAPI 从内存缓冲区返回 PDF 文件？

问题描述：

我想从 s3 获取一个 PDF 文件，然后从 FastAPI 后端将其返回到前端。

这是我的代码：

@router.post("/pdf_document")
def get_pdf(document : PDFRequest) :
    s3 = boto3.client('s3')
    file=document.name
    f=io.BytesIO()
    s3.download_fileobj('adm2yearsdatapdf', file,f)
    return StreamingResponse(f, media_type="application/pdf")

该API返回200状态码，但是不返回PDF文件作为响应。

解决方案 1：

由于整个文件数据已加载到内存中，因此没有实际理由使用StreamingResponse。您应该改用Response，通过传递文件字节（用于BytesIO.getvalue()获取包含缓冲区全部内容的字节）、定义media_type以及设置标题Content-Disposition，以便可以在浏览器中查看 PDF 文件或将其下载到用户的设备。有关更多详细信息和示例，请查看此答案以及此和此。相关答案也可以在这里找到。

此外，由于在调用方法buffer时会丢弃close()，因此您也可以使用 FastAPI/Starlette在返回响应后BackgroundTasks关闭buffer，以释放内存。或者，您可以使用获取字节pdf_bytes = buffer.getvalue()，然后使用关闭缓冲区buffer.close()，最后，return Response(pdf_bytes, headers=...。

例子

from fastapi import Response, BackgroundTasks

@app.get("/pdf")
def get_pdf(background_tasks: BackgroundTasks):
    buffer = io.BytesIO()  # BytesIO stream containing the pdf data
    # ...
    background_tasks.add_task(buffer.close)
    headers = {'Content-Disposition': 'inline; filename="out.pdf"'}
    return Response(buffer.getvalue(), headers=headers, media_type='application/pdf')

要下载 PDF 文件而不是在浏览器中查看，请使用：

headers = {'Content-Disposition': 'attachment; filename="out.pdf"'}

解决方案 2：

我的缓冲区和将 pdf 作为可下载链接发送的代码在http://127.0.0.1:8000/docs中
添加到标题字典“content-type”：“application/octet-stream”

async def convert_img_to_webp(img):
    image_io = BytesIO()
    image = Image.open(img)
    image.convert("RGB")
    image.save(image_io,"PDF")
    image_io.seek(0)
    # BackgroundTasks.add_task(image_io.close)
    return image_io


@router.post("/image/")
async def upload_file(file:UploadFile):
    if file:
        data = await convert_img_to_webp(file.file)
        headers = {'Content-Disposition': 'inline; filename="sample.pdf"',"content-type": "application/octet-stream"}
        return StreamingResponse(data,media_type='application/pdf',headers=headers)
    else:
        print("file not found")
        return