將數據導出到excel是許多web應用程序的常見需求。Python使得一切變得更加容易。但是,無論如何,每當我需要自己去實現時,有一個例子做參考,做起來,將會更加快捷。
實現這個目標有兩種方法:
-
使用Python的csv模塊將數據導出到.csv文件;
-
使用名為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>