生成CSV文件:
有時候我們做的網(wǎng)站,需要將一些數(shù)據(jù),生成有一個 CSV 文件給瀏覽器熏瞄,并且是作為附件的形式下載下來。以下將講解如何生成 CSV 文件谬以。
生成小的CSV文件:
這里將用一個生成小的 CSV 文件為例强饮,來把生成 CSV 文件的技術(shù)要點(diǎn)講到位。我們用 Python 內(nèi)置的 csv 模塊來處理 csv 文件为黎,并且使用 HttpResponse 來將 csv 文件返回回去邮丰。示例代碼如下:
代碼解釋:
1. 在初始化 HttpResponse 的時候,指定了 Content-Type 為 text/csv 铭乾,這將告訴瀏覽器剪廉,這是一個 csv 格式的文件而不是一個 HTML 格式的文件,如果用默認(rèn)值炕檩,默認(rèn)值就是 html 斗蒋,那么瀏覽器將把 csv 格式的文件按照 html 格式輸出,這肯定不是我們想要的。
2. 第二個我們還在 response 中添加一個 Content-Disposition 頭泉沾,這個東西是用來告訴瀏覽器該如何處理這個文件捞蚂,我們給這個頭的值設(shè)置為 attachment; ,那么瀏覽器將不會對這個文件進(jìn)行顯示跷究,而是作為附件的形式下載姓迅,第二個 filename="somefilename.csv" 是用來指定這個 csv 文件的名字。
3. 使用 csv 模塊的 writer 方法揭朝,將相應(yīng)的數(shù)據(jù)寫入到 response 中队贱。
將 csv 文件定義成模板:
還可以將 csv 格式的文件定義成模板,然后使用 Django 內(nèi)置的模板系統(tǒng)潭袱,并給這個模板傳入一個 Context 對象,這樣模板系統(tǒng)就會根據(jù)傳入的 Context 對象锋恬,生成具體的 csv 文件屯换。示例代碼如下:
模板文件:
視圖函數(shù):
生成大的CSV文件:
以上的例子是生成的一個小的 csv 文件,如果想要生成大型的 csv 文件与学,那么以上方式將有可能會發(fā)生超時的情況(服務(wù)器要生成一個大型csv文件彤悔,需要的時間可能會超過瀏覽器默認(rèn)的超時時間)。這時候我們可以借助另外一個類索守,叫做 StreamingHttpResponse 對象晕窑,這個對象是將響應(yīng)的數(shù)據(jù)作為一個流返回給客戶端,而不是作為一個整體返回卵佛。示例代碼如下:
這里我們構(gòu)建了一個非常大的數(shù)據(jù)集 rows 杨赤,并且將其變成一個迭代器。然后因?yàn)?StreamingHttpResponse 的第一個參數(shù)只能是一個生成器截汪,因此我們使用圓括號 (writer.writerow(row) for row in rows) 疾牲,并且因?yàn)槲覀円獙懙奈募?csv 格式的文件,因此需要調(diào)用 writer.writerow 將 row 變成一個 csv 格式的字符串衙解。而調(diào)用 writer.writerow 又需要一個中間的容器阳柔,因此這里我們定義了一個非常簡單的類 Echo ,這個類只實(shí)現(xiàn)一個 write 方法蚓峦,以后在執(zhí)行 csv.writer(pseudo_buffer) 的時候舌剂,就會調(diào)用 Echo.writer 方法。
注意: StreamingHttpResponse 會啟動一個進(jìn)程來和客戶端保持長連接暑椰,所以會很消耗資源霍转。所以如果不是特殊要求,盡量少用這種方法干茉。
關(guān)于StreamingHttpResponse:
這個類是專門用來處理流數(shù)據(jù)的谴忧。使得在處理一些大型文件的時候,不會因?yàn)榉?wù)器處理時間過長而到時連接超時。這個類不是繼承自 HttpResponse 沾谓,并且跟 HttpResponse 對比有以下幾點(diǎn)區(qū)別:
1. 這個類沒有屬性 content 委造,相反是 streaming_content 。
2. 這個類的 streaming_content 必須是一個可以迭代的對象均驶。
3. 這個類沒有 write 方法昏兆,如果給這個類的對象寫入數(shù)據(jù)將會報(bào)錯。注意: StreamingHttpResponse 會啟動一個進(jìn)程來和客戶端保持長連接妇穴,所以會很消耗資源爬虱。所以. ,_2. 這個類的 streaming_content 必須是一個可以迭代的對
3. 這個類沒有 write 方法腾它,如果給這個類的對象寫入數(shù)意: StreamingHttpResponse 會啟動一個進(jìn)程來和客戶端如果不是特殊要求跑筝,盡量少用這種方法。