Django实用技巧:如何把数据导出到Excel

将数据导出到excel是许多web应用程序的常见需求。Python使得一切变得更加容易。但是,无论如何,每当我需要自己去实现时,有一个例子做参考,做起来,将会更加快捷。

实现这个目标有两种方法:

  1. 使用Python的csv模块将数据导出到.csv文件;

  2. 使用名为xlwt的第三方模块将数据导出到.xls文件。

文中,我们将用导出django.contrib.auth.models.User的数据做为示范。


导出数据到CSV文件

简单的方法。不需要安装依赖项,这是一件很好的事情。如果您不需要任何花哨的格式,请使用它。

views.py

import csv
from django.http import HttpResponse
from django.contrib.auth.models import User
def export_users_csv(request):     response = HttpResponse(content_type='text/csv')     response['Content-Disposition'] = 'attachment; filename="users.csv"'     writer = csv.writer(response)     writer.writerow(['Username', 'First name', 'Last name', 'Email address'])     users = User.objects.all().values_list('username', 'first_name', 'last_name', 'email')     for user in users:         writer.writerow(user)     return response

urls.py

import views
urlpatterns = [
    ...
    url(r'^export/csv/$', views.export_users_csv, name='export_users_csv'),
]

template.html

<href="{% url 'export_users_csv' %}">Export all users</a>

在上面的示例中,我在QuerySet中使用values_list有两个原因:第一是只查询需要导出的字段,第二是因为它将返回元组中的数据,而不是模型实例。writerow方法需要一个列表/元组。

另一种方法是:

writer.writerow([user.username, user.first_name, user.last_name, user.email, ])

将数据导出到XLS文件

如果您确实需要导出到.xls文件,请使用这个方法。你将能够添加格式为粗体字体、字体大小、定义列大小等。

首先,安装xlwt模块。最简单的方法是使用pip。

pip install xlwt

views.py

import xlwt
from django.http import HttpResponse
from django.contrib.auth.models import User
def export_users_xls(request):     response = HttpResponse(content_type='application/ms-excel')     response['Content-Disposition'] = 'attachment; filename="users.xls"'     wb = xlwt.Workbook(encoding='utf-8')     ws = wb.add_sheet('Users')     # Sheet header, first row     row_num = 0     font_style = xlwt.XFStyle()     font_style.font.bold = True     columns = ['Username', 'First name', 'Last name', 'Email address', ]     for col_num in range(len(columns)):         ws.write(row_num, col_num, columns[col_num], font_style)     # Sheet body, remaining rows     font_style = xlwt.XFStyle()     rows = User.objects.all().values_list('username', 'first_name', 'last_name', 'email')     for row in rows:         row_num += 1         for col_num in range(len(row)):             ws.write(row_num, col_num, row[col_num], font_style)     wb.save(response)     return response

urls.py

import views
urlpatterns = [
    ...
    url(r'^export/xls/$', views.export_users_xls, name='export_users_xls'),
]

template.html

<href="{% url 'export_users_xls' %}">Export all users</a>

猜你喜欢

转载自www.cnblogs.com/pcent/p/10809130.html
今日推荐