Python return csv file in response

最新推荐文章于 2024-05-14 06:58:02 发布

Edison Dont

最新推荐文章于 2024-05-14 06:58:02 发布

阅读量470

点赞数 1

文章标签： csv python

原文链接：https://qiita.com/hirohuntexp/items/3a1bf3f195a50424c211

版权

import StringIO
import csv
from flask import make_response

@app.route('/download/<obj>/')
def download(obj):

    f = StringIO()
    writer = csv.writer(f, quotechar='"', quoting=csv.QUOTE_ALL, lineterminator="\n")

    if obj == 'users':
        writer.writerow(['id','username','gender','age','created_at'])
        for u in User.query.all():
            writer.writerow([u.id, u.username,u.gender,u.age,u.created_at])

    res = make_response()
    res.data = f.getvalue()
    res.headers['Content-Type'] = 'text/csv'
    res.headers['Content-Disposition'] = 'attachment; filename='+ obj +'.csv'
    return res