python-3.x - 如何将 CSV 文件对象传递给猎鹰响应

Question

csv_file = "test.csv"
with open(csv_file, 'w') as csvfile:
    writer = csv.DictWriter(csvfile, fieldnames=csv_columns)
    writer.writeheader()
    writer.writerows(response_data)
    csvfile.close()

这里我们在本地生成了 CSV 文件。现在我们正在尝试使用 falcon 创建和下载文件。如何将 CSV 文件对象传递给 falcon 响应？

Answer 1

您可以按如下方式提供 CSV 文件:

此外，如果您使用的是 Falcon >= 1.4:

import falcon
import mimetypes


class StaticResource(object):
    def on_get(self, req, resp):
        csv_file_path = "test.csv"
        resp.content_type = mimetypes.guess_type(csv_file_path)[0]
        resp.stream, resp.stream_len = open(csv_file_path)


app = falcon.API()
app.add_route('/static', StaticResource())

如果您已经创建了文件，则创建文件对象:

import falcon


class StaticResource(object):
    def on_get(self, req, resp):
        csv_file_path = "test.csv"
        resp.status = falcon.HTTP_200
        resp.content_type = 'text/csv'
        with open(csv_file_path, 'r') as f:
            resp.body = f.read()


app = falcon.API()
app.add_route('/static', StaticResource())

如果你已经有了文件对象:

import falcon


class StaticResource(object):
    def on_get(self, req, resp):
        csv_file_path = "test.csv"
        with open(csv_file_path, 'w') as csvfile:
            writer = csv.DictWriter(csvfile, fieldnames=csv_columns)
            writer.writeheader()
            writer.writerows(response_data)
            csvfile.close()
        resp.status = falcon.HTTP_200
        resp.content_type = 'text/csv'
        resp.body = writer 


app = falcon.API()
app.add_route('/static', StaticResource())

希望我能正确回答您的问题。另外，您可以引用此文档https://falcon.readthedocs.io/en/stable/api/api.html#falcon.API.add_static_route服务目录。