最近为团队在公司内部建的站好多页面中需要导出为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')就可以了。