博客

Python django导出excel详解

  • Post category:Python

请允许我详细讲解一下“Python Django 导出 Excel 详解”的完整实例教程。

1. 简介

在 Django 的开发中，经常需要从网站中导出数据，常见的导出格式是 Excel 格式。这里将介绍如何使用 Python 的 openpyxl 库完成 Django 的 Excel 导出。

2. 安装openpyxl

首先，我们需要安装 openpyxl 库，通过执行以下命令安装：

pip install openpyxl

3. 创建视图函数

接下来，我们需要创建一个视图函数，用于响应导出 Excel 的请求。一个简单的导出视图如下所示：

from django.http import HttpResponse
from openpyxl import Workbook

def export_excel(request):
    # 创建一个 workbook 对象
    wb = Workbook()

    # 获取数据并写入 worksheet
    ws = wb.active
    ws['A1'] = 'Hello'
    ws['B1'] = 'World'

    # 设置 response header
    response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
    response['Content-Disposition'] = 'attachment; filename=export.xlsx'

    # 将 workbook 对象写入 response
    wb.save(response)

    return response

在这个例子中，我们导出了一个包含 “Hello” 和 “World” 两行数据的 Excel 文件。注意到 HttpResponse 的 MIME 类型为 application/vnd.openxmlformats-officedocument.spreadsheetml.sheet，这是 Excel 文件的 MIME 类型。

4. 配置URL

在 Django 中，必须将视图函数配置到 URL 上，以便能够通过 HTTP 请求访问该视图函数。将上一步中创建的视图函数配置到一个 URL 上，可以通过如下代码实现：

from django.urls import path
from . import views

urlpatterns = [
    path('export-excel/', views.export_excel, name='export_excel'),
]

这个例子中，将 export_excel 视图配置到了 /export-excel/ URL 上。

5. 发送请求

配置好了 URL 和视图函数之后，我们就可以在浏览器中发送一个 GET 请求，来访问这个页面，并导出数据为 Excel 文件。在示例中，使用 localhost:8000/export-excel/ 访问该视图。

6. 示例说明

下面通过两个示例说明如何使用 openpyxl 来完成复杂的 Django 导出 Excel 功能。

示例1：导出数据库中的数据

from django.http import HttpResponse
from openpyxl import Workbook
from .models import MyModel

def export_excel(request):
    # 创建一个 workbook 对象
    wb = Workbook()

    # 获取数据并写入 worksheet
    ws = wb.active
    ws['A1'] = 'ID'
    ws['B1'] = 'Name'
    ws['C1'] = 'Email'

    # 查询 MyModel 中的所有数据，并写入Worksheet
    data = MyModel.objects.all()
    for i, item in enumerate(data):
        ws.cell(row=i+2, column=1, value=item.id)
        ws.cell(row=i+2, column=2, value=item.name)
        ws.cell(row=i+2, column=3, value=item.email)

    # 设置 response header
    response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
    response['Content-Disposition'] = 'attachment; filename=my_model_data.xlsx'

    # 将 workbook 对象写入 response
    wb.save(response)

    return response

在这个例子中，通过查询 MyModel 中的所有数据，并将数据写入 Excel 文件中。可以看到在写入 Worksheet 中的数据时，我们使用了 ws.cell() 方法来处理每个单元格的数据。因为使用了 enumerate() 函数，所以我们可以获取每个元素的索引值，从而指定这一行数据的行数。

相关模型和 URL 的配置，需要根据项目的实际情况进行修改。

示例2：导出筛选数据

from django.http import HttpResponse
from openpyxl import Workbook
from .models import MyModel

def export_excel(request):
    # 创建一个 workbook 对象
    wb = Workbook()

    # 获取数据并写入 worksheet
    ws = wb.active
    ws['A1'] = 'ID'
    ws['B1'] = 'Name'
    ws['C1'] = 'Email'

    # 查询 MyModel 中 name 包含 'John' 的数据，并将数据写入 Worksheet
    data = MyModel.objects.filter(name__icontains='John')
    for i, item in enumerate(data):
        ws.cell(row=i+2, column=1, value=item.id)
        ws.cell(row=i+2, column=2, value=item.name)
        ws.cell(row=i+2, column=3, value=item.email)

    # 设置 response header
    response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
    response['Content-Disposition'] = 'attachment; filename=john_data.xlsx'

    # 将 workbook 对象写入 response
    wb.save(response)

    return response

在这个例子中，查询了 MyModel 中名字中包含 “John” 的数据，并将其写入 Excel 文件中。这里使用了 filter() 函数来获取符合条件的数据。

这个例子中的相关模型和 URL 需要根据实际情况进行修改。

7. 结论

本文演示了如何使用 Python 的 openpyxl 库完成 Django 中的 Excel 导出功能。通过这种方法，可以轻松地导出数据到 Excel 文件，为数据分析和数据交换提供便利。

Tags: ,