flask使用tablib导出数据库数据

原文地址：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)