使用Flask下载多个CSV？ [英] Download multiple CSVs using Flask?

问题描述

我有一个应用程序需要一些信息，使用熊猫进行一些计算，并将最终的熊猫数据框转换为CSV，然后使用Flask应用程序下载。如何在一个视图中下载多个CSV？看来，我一次只能返回一个响应。

一个示例片段：

  def serve_csv（dataframe，filename）：
 buffer = StringIO.StringIO（）
 dataframe.to_csv（buffer，encoding ='utf-8'，index = False）
 buffer.seek（0）
 return send_file（buffer，
 attachment_filename = filename，
 mimetype ='text / csv'）
 $ b $ def make_calculation（arg1， arg2）：
'''做一些计算。 
输入：arg1  - 字符串，arg2-字符串
返回：一个熊猫数据框'''
 
 @ app.route（'test_app'，methods = ['GET'， 'POST'] 
 def test_app（）：
 form = Form1（）
 if form.validate_on_submit（）：
 calculated_dataframe = make_calculation（str（form.input_1.data）， str（form.input_2.data））
 return serve_csv（calculated_dataframe，'Your_final_output.csv'）
 return render_template（'test_app.html'，form = form）
  
 
 
解决方案
您可以返回 MIME Multipart 响应，一个zip文件或一个TAR球（请注意，链接的RFC有点过时，但由于它是HTML格式，所以更容易快速上手;官方的是< a href =http：// ww如果你选择做一个MIME multipart响应，那么你可以使用一个MIME多部分响应，一个好的起点可能是查看  MultipartEncoder  和请求工具栏中的nofollow>  MultipartDecoder   ;你可以直接使用它们，或者至少使用子类/组合来获得你想要的行为。  Zip文件和 TAR球可以使用标准库模块来实现。另外一种方法是设计您的API，以便您返回JSON，使用标头（或XML元素或JSON字段）来指示其他CSV可以通过其他请求或类似获得。
 
I have an app that takes in some information, performs some calculations using pandas, and turns the final pandas data frame into a CSV that is then downloaded using the Flask app. How do I download multiple CSVs within one view? It seems that I can only return a single response at a time.

An example snippet:
def serve_csv(dataframe,filename):
    buffer = StringIO.StringIO()
    dataframe.to_csv(buffer, encoding='utf-8', index=False)
    buffer.seek(0)
    return send_file(buffer,
             attachment_filename=filename,
             mimetype='text/csv')

def make_calculation(arg1, arg2):
   '''Does some calculations.
   input: arg1 - string, arg2- string
   returns: a pandas data frame'''

@app.route('test_app', methods=['GET', 'POST']
def test_app():
    form = Form1()
    if form.validate_on_submit():
    calculated_dataframe = make_calculation(str(form.input_1.data), str(form.input_2.data))
        return serve_csv(calculated_dataframe, 'Your_final_output.csv')
    return render_template('test_app.html', form=form)
So let's say in that example above that make_calculation returned two pandas data frames. How would I print both of them to a CSV?
解决方案
You could return a MIME Multipart response, a zip file, or a TAR ball (please note the linked RFC is somewhat out of date, but is easier to quickly get up to speed with because it's in HTML; the official one is here).

If you choose to do a MIME multipart response, a good starting point might be to look at the MultipartEncoder and MultipartDecoder in requests toolbelt; you may be able to use them directly, or at least subclass/compose using those to get your desired behavior. Zip files and TAR balls can be implemented using standard library modules.

An alternative would be to design your API so that you were returning JSON, use a header (or XML element or JSON field) to indicate that additional CSVs could be obtained by another request, or similar.

                        
这篇关于使用Flask下载多个CSV？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！