使用flask在同一HTML页面上返回带有下载选项的响应

时间:2017-04-24 16:04:09

标签: python html file pandas flask

我有一个基本的烧瓶应用程序,其中数据框由两个CSV组成,并且发生了一些转换,在HTML页面上,最终结果数据框可以以表格格式看到。它工作正常,直到这里。

除此之外,我还希望用户可以选择以CSV格式下载同一个表格。

以下是我的烧瓶代码:

from flask import *
import pandas as pd
app = Flask(__name__)
@app.route("/tables")
def show_tables():
     df1 = pd.read_csv('daily.csv')
     df2 = pd.read_csv('companies.csv')
     df1['date']= pd.to_datetime(df1['date'], format='%m/%d/%y')
     df3 = pd.merge(df1,df2,how='left',on='id')
     dates = pd.DataFrame({"date": pd.date_range("2017-01-01", "2017-01-10")})
     df4 = (df3.groupby(['id', 'name'])['date', 'value']
 .apply(lambda g: g.merge(dates, how="outer"))
 .fillna(0)
 .reset_index(level=[0,1])
 .reset_index(drop=True))
     df4 = df4.sort_values(by=['id','date'])
     df4.value = df4.value.astype(int)
     df4['difference'] = df4.groupby('id')['value'].diff()
     return render_template('view.html',tables=[df4.to_html(classes='Company_data')],
     titles = [ 'Company_data'],filename=df4.to_csv())
@app.route('/tables_download/<filename>')
def tables_download(filename):
    return response(filename)  //--right way to pass the csv file?


if __name__ == "__main__":
    app.run()

以下是我的 HTML 代码:

      <!doctype html>
<title>Simple tables</title>
<link rel=stylesheet type=text/css href="{{ url_for('static', filename='style.css') }}">
<div class=page>
  <h1>Company data</h1>
  {% for table in tables %}
    <h2>{{titles[loop.index]}}</h2>
    {{ table|safe }}
  {% endfor %}
</div>

<a href="{{ url_for('tables_download', filename=filename) }}">Download</a>

在我的HTML页面上,我甚至看不到下载选项。

努力找出错误所以寻找帮助

1 个答案:

答案 0 :(得分:0)

正如Flask API中所述,我相信send_filesend_from_directory将是实现此目的的方法。

@app.route('/uploads/<path:filename>')
def download_file(filename):
    return send_from_directory(app.config['UPLOAD_FOLDER'], filename, as_attachment=True)

这些都记录在http://flask.pocoo.org/docs/0.12/api/

send_from_directory更安全(如果使用正确),因为它将可供下载的文件限制为仅限于特定目录中的文件,从而防止任何“黑客”攻击。从下载您的私人信息。