Django实现下载100G的超大CSV文件

最新推荐文章于 2025-05-11 18:21:50 发布

Python_P叔

最新推荐文章于 2025-05-11 18:21:50 发布

阅读量1k

点赞数 21

CC 4.0 BY-SA版权

文章标签： django sqlite 数据库

本文链接：https://blog.youkuaiyun.com/Saki_Python/article/details/135670325

本文介绍了如何使用Django的StreamingHttpResponse处理大数据集，避免内存不足问题。通过创建StreamingHttpResponse对象和定制的生成器函数，实现大型CSV文件的流传输，降低内存使用，提高效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在处理大数据集的时候，我们经常遇到的一个问题是内存使用。当我们试图生成一个大型文件，如 CSV，并尝试将其全部加载到内存中时，可能会遇到内存不足的问题。幸运的是，Django 提供了一个解决方案：流传输。本文将详细的讲述如何利用 Django 进行大型 CSV 文件的流传输。

流传输的基础概念

流传输是一种技术，允许你一次发送一小部分响应，而不是一次性加载整个响应到内存中。这种技术对于服务大文件，如图像，音频和大型数据集非常有用。

创建 Django StreamingHttpResponse 对象

StreamingHttpResponse 对象允许我们将响应内容按部分，或者说“流”的形式发送给客户端。试看以下代码：

from django.http import StreamingHttpResponse

def big_file_download(request):
    def file_iterator(file_name, chunk_size=512):
        with open(file_name, "rb") as f:
            while True:
                c = f.read(chunk_size)
                if c:
                    yield c
                else:
                    break
    response = StreamingHttpResponse(file_iterator('big_file.csv'))
    return response

在这里，我们首先创建一个生成器函数 file_iterator，它以 chunk_size 为单位读取文件，然后用这个生成器函数作为 StreamingHttpResponse 的参数，从而创建出一个可以流传输的响应对象。

实现 Django 流传输大型 CSV 文件

现在，我们知道了如何创建 StreamingHttpResponse 对象，接着看一下如何用它来流传输大型 CSV 文件:

import csv
from django.http import StreamingHttpResponse
from .models import Person

class Echo:
    def write(self, value):
        return value

def streaming_csv_view(request):
    persons = Person.objects.all().values_list('name', 'age', 'city')
    pseudo_buffer = Echo()
    writer = csv.writer(pseudo_buffer)
    
    response = StreamingHttpResponse((writer.writerow(person) for person in persons),
                                     content_type="text/csv")
    response['Content-Disposition'] = 'attachment; filename="persons.csv"'
    return response

在上述代码中,我们创建了一个 Echo 类，并将其实例 pseudo_buffer 作为 csv.writer 的参数。这样，我们可以以迭代的方式写入 CSV 行，而不必把它们全部加载到内存中。之后，我们把这个 writerow 方法的迭代器作为 StreamingHttpResponse 的参数。注意，我们此时仍需要设置正确的 MIME 类型和 Content-Disposition 头部。