python-3.x 输出CSV文件以供下载，但不将CSV保存在服务器中

dddzy1tm 于 2023-01-06 发布在 Python

关注(0)|答案(2)|浏览(127)

我正在使用Python的Fast API框架开发一个API端点，它将提供给公众使用。它已经完成并在工作，但它的工作方式是我将创建它，然后保存到服务器的本地目录，然后读取该文件并返回一个csv文件作为对用户的响应。
我的问题是，如何直接将csv文件返回给用户，而不将其保存在服务器的目录中。

def export_client_invoice(client_id):

    invoice_doc = client_docs_db.view("client_invoice/by_client_id", key=int(client_id), include_docs=True)

    data = ["value %d" % i for i in range(1,4)]

    with open("./file.csv", 'w', newline='') as myfile:
    wr = csv.writer(myfile, quoting=csv.QUOTE_ALL)
    wr.writerow(data)

    file_like = open("./file.csv", mode="rb")
    response = StreamingResponse(file_like, media_type="text/csv")
    response.headers["Content-Disposition"] = "attachment; filename=export.csv"
    
return response

python-3.x

来源：https://stackoverflow.com/questions/66795949/output-csv-file-for-download-without-saving-the-csv-in-the-server

2条答案

按热度按时间

ss2ws0br1#

我不能用fastapi测试它，所以你可能需要稍微采用一下，使它在你的环境中工作。

from io import BytesIO
import csv
import codecs

data = ['value %d' % i for i in range(1,4)]

StreamWriter = codecs.getwriter('utf-8')
file_like = StreamWriter(BytesIO())

wr = csv.writer(file_like, quoting=csv.QUOTE_ALL)
wr.writerow(data)

print(file_like.getvalue())
# b'"value 1","value 2","value 3"\r\n'

response = StreamingResponse(file_like, media_type="text/csv")

赞(0）回复(0）举报 2023-01-06

8zzbczxx2#

@jackhammer013，您可以使用此代码返回带有FastAPI的csv文件

from fastapi.responses import StreamingResponse
... 
    return StreamingResponse(iter([file_like.getvalue()]),
                                             status_code=status.HTTP_200_OK,
                                             headers={
                                                 "Content-Disposition": f"attachment; filename=test.csv"
                                             },
                                             media_type="text/csv")

赞(0）回复(0）举报 2023-01-06

我来回答

python-3.x 输出CSV文件以供下载，但不将CSV保存在服务器中

2条答案

相关问题

热门标签

最新问答