3

我正在尝试通过 Falcon 中的 GET 请求发送 CSV。我不知道从哪里开始。

下面是我的代码：

class LogCSV(object):
"""CSV generator.

This class responds to  GET methods.
"""
def on_get(self, req, resp):
    """Generates CSV for log."""

    mylist = [
        'one','two','three'
    ]

    myfile = open("testlogcsv.csv", 'w')
    wr = csv.writer(myfile, quoting=csv.QUOTE_ALL)
    wr.writerow(mylist)

    resp.status = falcon.HTTP_200
    resp.content_type = 'text/csv'
    resp.body = wr

我不想用勺子喂食，请让我知道我应该阅读/观看什么来帮助解决这个问题。谢谢

4

1 回答 1

1

您应该使用该Response.stream属性。在返回之前必须将其设置为类文件对象（具有read()方法的对象）。

因此，首先，您应该将 CSV 写入此对象，然后将其提供给 Falcon。在你的情况下：

resp.content_type = 'text/csv'
# Move the file pointer to the beginning
myfile.seek(0)
resp.stream = myfile

请记住使用 将文件指针移到开头seek(0)，以便 Falcon 可以读取它。

如果您的文件是临时文件并且足够小以存储在内存中，则可以使用内存文件BytesIO而不是普通文件。它的行为类似于普通文件，但从不写入文件系统。

myfile = BytesIO()
wr = csv.writer(myfile, quoting=csv.QUOTE_ALL)

...

resp.content_type = 'text/csv'
# Move the file pointer to the beginning
myfile.seek(0)
resp.stream = myfile

;)

于 2015-10-06T00:51:49.333 回答