有時候我們做的網站,需要將一些數據,生成有一個 CSV 文件給瀏覽器,并且是作為附件的形式下載下來。下面我們看一下如何生成 CSV 文件。
下載小的CSV文件我們先看一下怎么下載小的 CSV 文件。我們用 Python 內置的 csv 模塊來處理 csv 文件,并且使用 HttpRespon 來將 csv 文件返回回去。示例代碼如下:
import csvfrom django.http import HttpRespondef csv_view(request): respon = HttpRespon(content_type='text/csv') respon['Content-Disposition'] = 'attachment; filename="somefilename.csv"' writer = csv.writer(respon) writer.writerow(['name', 'age', 'x']) writer.writerow(['張三', '28', '女']) return respon我們在初始化 HttpRespon 的時候,指定了 Content-Type 為 text/csv,這樣瀏覽器就會把它當做一個 csv 格式的文件而不是一個 HTML 格式的文件,因為默認值就是 html。respon 中添加一個 Content-Disposition 頭,這將告訴瀏覽器該如何處理這個文件,我們給這個頭的值設置為 attachment;,那么瀏覽器將不會對這個文件進行顯示,而是作為附件的形式下載,第二個 filename="somefilename.csv" 是用來指定這個csv文件的名字。我們使用 csv 模塊的 writer 方法,將相應的數據寫入到 respon 中。將csv文件定義成模板
我們還可以將 csv 格式的文件定義成模板,然后使用 Django 內置的模板系統,并給這個模板傳入一個 Context 對象,這樣模板系統就會根據傳入的 Context 對象,生成具體的 csv 文件。示例代碼如下:
模板文件:
{% for row in data %}"{{ row.0|addslashes }}", "{{ row.1|addslashes }}", "{{ row.2|addslashes }}", "{{ row.3|addslashes }}", "{{ row.4|addslashes }}"{% endfor %}
視圖函數:
from django.http import HttpResponfrom django.template import loader, Contextdef some_view(request): respon = HttpRespon(content_type='text/csv') respon['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') respon.content = t.render({"data": csv_data}) return respon如何下載超大的CSV文件
以上的例子是生成的一個小的csv文件,如果想要生成大型的csv文件,那么以上方式將有可能會發生超時的情況(服務器要生成一個大型csv文件,需要的時間可能會超過瀏覽器默認的超時時間)。這時候我們可以借助另外一個類,叫做StreamingHttpRespon對象,這個對象是將響應的數據作為一個流返回給客戶端,而不是作為一個整體返回。示例代碼如下:
class Echo: """ 定義一個可以執行寫操作的類,以后調用csv.writer的時候,就會執行這個方法 """ def write(lf, value): return valuedef large_csv(request): rows = (["Row {}".format(idx), str(idx)] for idx in range(655360)) pudo_buffer = Echo() writer = csv.writer(pudo_buffer) respon = StreamingHttpRespon((writer.writerow(row) for row in rows),content_type="text/csv") respon['Content-Disposition'] = 'attachment; filename="somefilename.csv"' return respon
這里我們構建了一個非常大的數據集rows,并且將其變成一個迭代器。然后因為StreamingHttpRespon的第一個參數只能是一個生成器,因此我們使用圓括號(writer.writerow(row) for row in rows),并且因為我們要寫的文件是csv格式的文件,因此需要調用writer.writerow將row變成一個csv格式的字符串。而調用writer.writerow又需要一個中間的容器,因此這里我們定義了一個非常簡單的類Echo,這個類只實現一個write方法,以后在執行csv.writer(pudo_buffer)的時候,就會調用Echo.writer方法。
關于StreamingHttpRespon:這個類是專門用來處理流數據的。使得在處理一些大型文件的時候,不會因為服務器處理時間過長而到時連接超時。這個類不是繼承自HttpRespon,并且跟HttpRespon對比有以下幾點區別:
這個類沒有屬性content,相反是streaming_content。這個類的streaming_content必須是一個可以迭代的對象。這個類沒有write方法,如果給這個類的對象寫入數據將會報錯。注意:StreamingHttpRespon會啟動一個進程來和客戶端保持長連接,所以會很消耗資源。所以如果不是特殊要求,盡量少用這種方法。
本文發布于:2023-02-28 21:14:00,感謝您對本站的認可!
本文鏈接:http://m.newhan.cn/zhishi/a/1677740996108114.html
版權聲明:本站內容均來自互聯網,僅供演示用,請勿用于商業和其他非法用途。如果侵犯了您的權益請與我們聯系,我們將在24小時內刪除。
本文word下載地址:電影集結號下載(集結號電影 下載).doc
本文 PDF 下載地址:電影集結號下載(集結號電影 下載).pdf
| 留言與評論(共有 0 條評論) |