将django的HTTP响应保存为excel

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

[python] view plain copy

1. #encoding=utf-8  
2.   
3. import xlwt  
4. from django.http import HttpResponse  
5. from django.template.loader import get_template  
6. from django.db import connection  
7.   
8.   
9. def save_asset_xls(request):  
10.         sql = 'select * from book';  
11.   
12.         cursor = connection.cursor()     
13.         cursor.execute(data_sql)         
14.         row_all = cursor.fetchall()  
15.   
16.         _lst = []  
17.         _lst.extend(row_all[:])          
18.         _lst.insert(0, ['列1', '列2', '列3'])  
19.   
20.         book = xlwt.Workbook(encoding='utf8')   
21.         sheet = book.add_sheet('untitled')  
22.   
23.         for row, rowdata in enumerate(_lst):  
24.             for col, val in enumerate(rowdata):  
25.                 sheet.write(row, col, val, style=xlwt.Style.default_style)  
26.   
27.         response = HttpResponse(mimetype='application/vnd.ms-excel')  
28.         response['Content-Disposition'] = 'attachment; filename=example.xls'  
29.         book.save(response)  
30.         return response  

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