将django的HTTP响应保存为excel

最近为团队在公司内部建的站好多页面中需要导出为excel格式，在网上查了查之后终于实现了，在这里记录一下，以便日后需要时查看一下。这里需要python中安装xlwt模块，代码大致如下：

#encoding=utf-8

import xlwt
from django.http import HttpResponse
from django.template.loader import get_template
from django.db import connection


def save_asset_xls(request):
        sql = 'select * from book';

        cursor = connection.cursor()   
        cursor.execute(data_sql)       
        row_all = cursor.fetchall()

        _lst = []
        _lst.extend(row_all[:])        
        _lst.insert(0, ['列1', '列2', '列3'])

        book = xlwt.Workbook(encoding='utf8') 
        sheet = book.add_sheet('untitled')

        for row, rowdata in enumerate(_lst):
            for col, val in enumerate(rowdata):
                sheet.write(row, col, val, style=xlwt.Style.default_style)

        response = HttpResponse(mimetype='application/vnd.ms-excel')
        response['Content-Disposition'] = 'attachment; filename=example.xls'
        book.save(response)
        return response

这里给出了一个从数据库查找数据然后导出excel的代码，并以HTTP响应的方式返回给浏览器。当然也可以将数据直接存成本地文件，只要把book.save(response)改为book.save('/tmp/test.xls')就可以了。