Django 如何使用视图动态输出 CSV 以及 PDF

简介: 这一篇我们需要用到 python 的 csv 和 reportLab 库,通过django视图来定义输出我们需要的 csv 或者 pdf 文件。

这一篇我们需要用到 pythoncsvreportLab 库,通过django视图来定义输出我们需要的 csv 或者 pdf 文件。

csv文件

打开我们的视图文件 testsite/members/views.py 。新增一个视图方法:

import csv  # 导入python的csv包
def some_view(request):
    # Create the HttpResponse object with the appropriate CSV header.
    response = HttpResponse(
        content_type="text/csv",
        headers={"Content-Disposition": 'attachment; filename="somefilename.csv"'},
    )
    writer = csv.writer(response)
    writer.writerow(["第一行", "Foo", "Bar", "Baz"])
    writer.writerow(["第二行", "A", "B", "C", '"Testing"', "Here's a quote"])
    return response

在上面代码里面

  • 响应会获得一个特殊的 MIME 类型 text/csv。会告诉浏览器该文档是一个 CSV 文件,而不是 HTML 文件。
  • 响应会获得一个附加 Content-Disposition 标头,其中包含 CSV 文件的名称。此文件名是任意的;您可以随意命名。浏览器会在“另存为…”对话框中使用它。
  • response 您可以通过将作为第一个参数传递给 来挂接 CSV 生成 API csv.writer 。该 csv.writer 函数需要一个类似文件的对象,并且 HttpResponse 对象符合要求。
  • 对于 CSV 文件中的每一行,调用 writer.writerow ,并将一个 可迭代的传递给它。
  • CSV 模块会为您处理引号,因此您不必担心转义带引号或逗号的字符串。传递 writerow() 您的原始字符串,它会做正确的事情。


然后打开我们的路由文件 testsite/members/urls.py ,添加一个路由:

path('csv/', views.some_view, name='csv'),

访问我们的 http://127.0.0.1:8000/members/csv 地址,可以得到一个 csv 文件。如下图所示:

1719572094605.jpg

打开文件里面就是我们自定义格式的 csv

1719572105982.jpg

如果当我们在使用流式传输大型 csv 文件的时候,在处理生成非常大响应的视图时,就要改用  Django StreamingHttpResponse


为啥要改成这个呢?


举个例子,通过流式传输需要很长时间才能生成的文件,您可以避免负载平衡器在服务器生成响应时丢弃可能超时的连接。


在下面例子中,我们就可以充分利用 Python 生成器来高效处理大型 CSV 文件的组装和传输,打开 testsite/members/views.py 文件:

import csv
from django.http import StreamingHttpResponse
class Echo:
    """An object that implements just the write method of the file-like
    interface.
    """
    def write(self, value):
        """Write the value by returning it, instead of storing in a buffer."""
        return value
def some_streaming_csv_view(request):
    """A view that streams a large CSV file."""
    rows = (["Row {}".format(idx), str(idx)] for idx in range(65536))
    pseudo_buffer = Echo()
    writer = csv.writer(pseudo_buffer)
    return StreamingHttpResponse(
        (writer.writerow(row) for row in rows),
        content_type="text/csv",
        headers={"Content-Disposition": 'attachment; filename="somefilename.csv"'},
    )

添加我们的路由,打开我们的 testsite/members/urls.py 文件:

path('csv-stream/', views.some_streaming_csv_view, name='csv'),

浏览器访问 http://127.0.0.1:8080/members/csv-stream 流式输出文件:

1719572180225.jpg

打开这个文件如图示:

1719572202315.jpg

当然还一种方式来生成 csv ,通过 django 的模版系统。下面跟着代码来试一下这种方法:


依然还是先打开视图文件 testsite/members/views.py

from django.http import HttpResponse
from django.template import loader
def some_view(request):
    response = HttpResponse(
        content_type="text/csv",
        headers={"Content-Disposition": 'attachment; filename="somefilename.csv"'},
    )
    csv_data = (
        ("First row", "Foo", "Bar", "Baz"),
        ("Second row", "A", "B", "C", '"Testing"', "Here's a quote"),
    )
    t = loader.get_template("my_template_name.txt")
    c = {"data": csv_data}
    response.write(t.render(c))
    return response

添加我们的路由,打开我们的 testsite/members/urls.py 文件:

path('csv-template/', views.some_view_tem, name='csv'),

在当前目录下面的 templates 文件夹创建我们的模板文件 my_template_name.txt

{% for row in data %}"{{ row.0|addslashes }}", "{{ row.1|addslashes }}", "{{ row.2|addslashes }}", "{{ row.3|addslashes }}", "{{ row.4|addslashes }}"
{% endfor %}

浏览器访问 http://127.0.0.1:8080/members/csv-template 流式输出文件:

1719572377365.jpg

就可以得到我们自定义模板的 csv 文件。

如何创建 PDF 文件

由于 python 拥有出色的开源 ReportLab Python PDF 库,使得我们动态生成 pdf 非常有优势。


首先我们安装ReportLab 库:

py -m pip install reportlab

我们可以看到下列结果:

1719572448256.jpg

使用 django 动态生成  pdf 的关键在于 ReportLab API 作用于类似文件的对象,而 DjangoFileResponse 对象接受类似文件的对象。


打开我们的 testsite/members/views.py 视图文件:

import io
from django.http import FileResponse
from reportlab.pdfgen import canvas
def some_view_pdf(request):
    buffer = io.BytesIO()
    p = canvas.Canvas(buffer)
    p.drawString(100, 100, "Hello world.")
    p.showPage()
    p.save()
    buffer.seek(0)
    return FileResponse(buffer, as_attachment=True, filename="hello.pdf")

代码里面我讲一下大概的含义:

  • 响应将根据文件扩展名自动设置 MIME 类型 application/pdf 。这就告诉浏览器该文档是 PDF 文件,而不是 HTML 文件或通用 application/octet-stream 二进制内容。
  • as_attachment=True 传递给时 FileResponse ,它会设置适当的 Content-Disposition 标头,并告诉网络浏览器弹出一个对话框,提示/确认如何处理文档,即使机器上设置了默认值。如果as_attachment 省略该参数,浏览器将使用已配置用于 PDF 的任何程序/插件来处理 PDF
  • filename 定义你需要输出的 pdf 名字。浏览器将在“另存为…”对话框中使用它。
  • 可以挂接到 ReportLab API :作为第一个参数传递的相同缓冲区 canvas.Canvas 可以提供给该类 FileResponse
  • 所有后面的 PDF 生成方法均在 PDF 对象(上面例子中为p)上调用,而不是在 上调用 buffer
  • showPage() 最后,调用并 save() 查看 PDF 文件非常重要。


添加我们的路由,打开我们的 testsite/members/urls.py 文件:

path('csv-stream/', views.some_streaming_csv_view, name='csv'),

浏览器访问 http://127.0.0.1:8080/members/pdf

1719572547033.jpg

打开 pdf 文件,就可以看到我们自动输出的 pdf 文件:

1719572595337.jpg

ReportLab 并不是线程安全的。


所以一些小伙伴在构建 PDF 生成 Django 视图时出现的奇怪问题,这些视图就是由于许多人同时访问造成的。

总结

虽然 csv 文件和 pdf 文件的动态生成技术很常见,也是业务中经常用到的,我们文中所用到的只是其中的某个包,比如 reportlab


django 官方的 packages 有更详细不同包的比较,以及他们的使用方式,https://djangopackages.org/grids/g/pdf/


利用这两个工具结合我们的 django 视图,就完全的能和我们自己开发的系统结合起来了。


根据相应的显示字段动态生成一切我们想要的文件,是非常方便的。

相关文章
|
24天前
|
Python
Django——视图
Django——视图
|
4天前
|
存储 安全 网络协议
Django路由与会话深度探索:静态、动态路由分发,以及Cookie与Session的奥秘
Django路由与会话深度探索:静态、动态路由分发,以及Cookie与Session的奥秘
|
4天前
|
Go 调度 Python
Django 视图探秘:FBV与CBV注册方式的异同,揭秘as_view()的执行魔法
Django 视图探秘:FBV与CBV注册方式的异同,揭秘as_view()的执行魔法
|
4天前
|
JSON 缓存 前端开发
Django视图层探索:GET/POST请求处理、参数传递与响应方式详解
Django视图层探索:GET/POST请求处理、参数传递与响应方式详解
|
19天前
|
API 网络架构 Python
在django使用视图集和路由集
【6月更文挑战第11天】本文介绍Viewsets是Django REST框架中将多个视图逻辑整合到单个类的工具,减少了重复代码。当项目API变得复杂且有重复模式时,考虑使用它们;否则,保持视图和URL模式的简洁性。
19 3
|
20天前
|
API 数据库 网络架构
在django中应用视图和路由集
【6月更文挑战第10天】 本文介绍viewsets`和`Routers`是Django REST framework中用于简化API视图和路由的工具。它们提供了一个抽象层,允许用更少的代码替代多个相关视图,并能自动生成URL。定义`UserList`和`UserDetail`视图集,分别用于列表和详情展示。
15 3
|
4天前
|
数据库 开发者 Python
Django ORM入门指南:从概念到实践,掌握模型创建、迁移与视图操作
Django ORM入门指南:从概念到实践,掌握模型创建、迁移与视图操作
|
24天前
|
Python
创建Django项目,实现视图,路由
创建Django项目,实现视图,路由
|
2月前
|
开发者 Python
Django模板系统的强大之处:动态渲染与扩展性
【4月更文挑战第15天】Django模板系统是Web开发中的强大工具,支持动态渲染和扩展性。动态渲染包括变量、标签和过滤器的使用,实现内容根据上下文数据动态生成。模板继承和自定义标签则提升了扩展性,减少代码重复,增强可维护性。通过这些特性,Django模板系统助力开发者构建高效、动态的Web应用。
|
2月前
|
XML JSON 前端开发
Django视图函数技巧,从入门到实战
Django视图函数技巧,从入门到实战