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

<a 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

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

如果想了解更多关于xlwt模块的内容,请点击:official documentation


相关推荐:django-import-export插件使用教程


码字不容易,转载请加上本文链接和注明出处。如果上面的内容帮到你了,可以打赏作者喝杯茶。

117.jpg

文章评论 4