Django：使用openpyxl将工作簿保存为附件

在本文中，我们将介绍如何使用Django和openpyxl库将工作簿保存为附件。Django是一个用于快速开发Web应用程序的高级Python框架，而openpyxl则是用于操作Excel文件的Python库。

阅读更多：Django 教程

准备工作

首先，我们需要安装Django和openpyxl库。在终端中执行以下命令：

pip install django openpyxl

安装完成后，我们需要创建一个Django项目。在终端中执行以下命令：

django-admin startproject myproject
cd myproject

现在，我们需要创建一个Django应用程序，以便于在项目中进行操作。在终端中执行以下命令：

python manage.py startapp excel

创建视图函数

接下来，我们需要在excel应用程序的views.py文件中创建一个视图函数。这个函数将负责生成工作簿并将其作为附件发送给用户。打开excel/views.py文件并添加以下代码：

from django.http import HttpResponse
from openpyxl import Workbook

def generate_workbook(request):
    response = HttpResponse(content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
    response['Content-Disposition'] = 'attachment; filename="workbook.xlsx"'

    workbook = Workbook()
    worksheet = workbook.active
    worksheet.append(['姓名', '年龄', '性别'])
    worksheet.append(['张三', 20, '男'])
    worksheet.append(['李四', 25, '女'])

    workbook.save(response)

    return response

在这个函数中，我们首先创建了一个HttpResponse对象，并将其content_type设置为application/vnd.openxmlformats-officedocument.spreadsheetml.sheet，这是一个Excel文件的MIME类型。然后，我们设置了Content-Disposition头部字段，告诉浏览器将响应内容作为附件下载，并指定了下载的文件名为workbook.xlsx。

接着，我们创建了一个工作簿对象和一个工作表对象。然后，我们使用append方法向工作表中添加数据。在这个示例中，我们添加了一个标题行和两行数据。最后，我们使用工作簿的save方法将工作簿保存到HttpResponse对象中。

配置URL路由

接下来，我们需要配置URL路由，以便用户可以访问我们的视图函数。打开myproject/urls.py文件并添加以下代码：

from django.urls import path
from excel.views import generate_workbook

urlpatterns = [
    path('generate-workbook/', generate_workbook, name='generate_workbook'),
]

在这个示例中，我们将/generate-workbook/映射到了generate_workbook视图函数。

创建模板和视图函数

最后，我们需要创建一个模板和一个视图函数，以便于用户访问我们的功能。在excel/views.py文件中添加以下代码：

from django.shortcuts import render

def generate_workbook(request):
    return render(request, 'excel/generate_workbook.html')

然后，我们需要创建一个模板文件。在项目的根目录下，创建一个名为templates的文件夹，然后在该文件夹下创建一个名为excel的子文件夹。在excel文件夹下创建一个名为generate_workbook.html的文件，并添加以下代码：

<h1>生成工作簿</h1>

<form action="{% url 'generate_workbook' %}" method="post">
  {% csrf_token %}
  <input type="submit" value="生成工作簿">
</form>

在这个模板中，我们创建了一个表单，其中包含一个提交按钮。当用户点击按钮时，表单将发送POST请求到generate_workbook视图函数，并生成工作簿。

运行项目

现在，我们已经完成了所有必要的配置。在终端中执行以下命令来运行Django项目：

python manage.py runserver

然后，在浏览器中访问http://localhost:8000/generate-workbook/。你应该能够看到一个页面上仅有一个“生成工作簿”的按钮。

点击按钮后，会生成一个名为workbook.xlsx的Excel文件。浏览器将提示你下载该文件。下载并打开Excel文件，你将看到一个包含姓名、年龄和性别的表格。

总结

本文介绍了如何使用Django和openpyxl库将工作簿保存为附件。我们首先创建了一个视图函数，该函数生成了一个包含数据的工作簿，并将其作为附件发送给用户。然后，我们配置了URL路由，并创建了一个模板和视图函数，以便用户可以访问生成工作簿的功能。

使用Django和openpyxl，你可以轻松地在Web应用程序中生成和下载Excel文件。这对于需要在应用程序中处理大量数据或导出报告的项目非常有用。通过使用这些工具，你可以更加灵活和高效地处理Excel文件，从而提高开发效率。

希望本文对你理解和使用Django和openpyxl库有所帮助。祝你使用这些工具开发出强大的Web应用程序和数据处理功能！