原文地址:http://blog.kantli.com/article/46
在页面上导出数据是常见的需求,在flask中,要导出excel格式的数据,可以使用xlwt库,创建excel文件,逐行写入,不过,tablib库对这种操作做了很好的封装,用起来会更简单一些。
tablib可以通过pip直接安装。
tablib的使用,首先是构建基本数据对象:
import tablib
data = tablib.Dataset(*dataList, headers=headerList)
其中dataList是数据列表,一般是根据数据库查询结果构建的,列表中是每行数据的列表,例如:
[[1,2,3], [4,5,6], [7,8,9]]
headerList是标题列表,一般手动构建,例如:
[U'第一列', U'第二列', U'第三列']
获取数据对象之后,可以根据这个对象生成各种格式的数据文件,可以保存在服务器上供人下载,或者直接生成请求的响应。不需要多次下载的情况下,直接生成响应比较简单:
from flask import make_response
resp = make_response(data.xls)
filename = 'filename.xls' #用户下载默认文件名
resp.headers["Content-Disposition"] = "attachment; filename=" + filename #指定响应为下载文件
resp.headers['Content-Type'] = 'xls' #不指定的话会默认下载html格式,下载后还要改格式才能看
return resp
那么,一个综合的例子如下,根据用户给定的时间区间导出数据:
@main.route('/export_data', methods=['GET', 'POST'])
def export_data():
form = ExportDataForm() #供用户指定时间的表格,在forms中创建的简单类,导入使用
if form.validate_on_submit():
startDate = form.startDate.data
endDate = form.endDate.data
dataList = getData(startDate=startDate, endDate=endDate) #查询数据库并返回列表的函数,在其它文件创建并导入这里使用
headerList = [U"核导弹代码", U"类别", U"生产单位", U"入库日期"]
data = tablib.Dataset(*dataList, headers=headerList)
resp = make_response(data.xls)
resp.headers['Content-Type'] = 'xls'
filename = "核武库" + startDate.strftime("%Y%m%d") + "-" + endDate.strftime("%Y%m%d") + ".xls"
resp.headers["Content-Disposition"] = "attachment; filename=" + filename
return resp
return render_template('export_data.html', form=form)