Django教程：动态生成和优化CSV文件导出

这篇文档详细介绍了如何通过Django视图动态输出CSV（逗号分隔值）文件。你可以使用Python的CSV库或Django的模板系统来实现这一目标。

使用Python CSV库

Python内置了CSV库，即csv。在Django中使用它的关键在于，csv模块的CSV创建功能可以作用于类似文件的对象，而Django的HttpResponse对象正好是这样的一个对象。

以下是一个示例：

import csv from django.http import HttpResponse def some_view(request): # 创建带有适当CSV标头的HttpResponse对象。 response = HttpResponse(content_type='text/csv') response['Content-Disposition'] = 'attachment; filename="somefilename.csv"' writer = csv.writer(response) writer.writerow(['First row', 'Foo', 'Bar', 'Baz']) writer.writerow(['Second row', 'A', 'B', 'C', '"Testing"', "Here's a quote"]) return response

这里的代码和注释很清晰，但需要注意一些事项：

• 响应对象获得了特殊的MIME类型text/csv，以告知浏览器该文档是CSV文件而不是HTML文件。
• 响应对象获取了附加的Content-Disposition协议头，其中包含CSV文件的名称，用于浏览器的”另存为”对话框。

在Python 2中处理Unicode

在Python 2中，csv模块不支持Unicode输入。对于这个问题，你可以手动将Unicode对象编码为兼容的编码，使用csv模块示例中提供的UnicodeWriter类，或者使用python-unicodecsv模块，它是csv模块的替代方案，能够优雅地处理Unicode。

流式传输大尺寸CSV文件

当处理生成大尺寸响应的视图时，可以考虑使用Django的StreamingHttpResponse类。下面是一个示例，利用Python的生成器来有效处理大尺寸CSV文件的拼接和传输。

import csv from django.utils.six.moves import range from django.http import StreamingHttpResponse class Echo(object): def write(self, value): return value def some_streaming_csv_view(request): rows = (["Row {}".format(idx), str(idx)] for idx in range(65536)) pseudo_buffer = Echo() writer = csv.writer(pseudo_buffer) response = StreamingHttpResponse((writer.writerow(row) for row in rows), content_type="text/csv") response['Content-Disposition'] = 'attachment; filename="somefilename.csv"' return response

使用模板系统

另一种生成CSV的方法是使用Django模板系统。虽然相对于方便的Python csv模块而言较为低级，但为了完整性，这个解决方案也在这里展示。

以下是一个示例，它使用模板系统生成相同的CSV文件：

from django.http import HttpResponse from django.template import loader, Context def some_view(request): response = HttpResponse(content_type='text/csv') response['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 = Context({'data': csv_data}) response.write(t.render(c)) return response

以上例子和前一个例子唯一的不同在于，这个例子使用模板加载而不是csv模块。模板的结果，例如content_type='text/csv'，都是相同的。然后，创建模板my_template_name.txt，其中包含适当的模板代码。

这个文档还提供了一些关于处理Unicode和流式传输大尺寸CSV文件的实用信息。最后，文档强调了这些技巧不仅适用于CSV，你可以使用类似的方法生成其他基于文本或二进制数据的格式。