1. 程式人生 > >Django實現大檔案下載

Django實現大檔案下載

基於Django建立的網站,如果提供檔案下載功能,最簡單的方式莫過於將靜態檔案交給Nginx等處理,但有些時候,由於網站本身邏輯,需要通過Django提供下載功能,如頁面資料匯出功能(下載動態生成的檔案)、先檢查使用者許可權再下載檔案等。因此,有必要研究一下檔案下載功能在Django中的實現。

最簡單的檔案下載功能的實現

將檔案流放入HttpResponse物件即可,如:

def file_download(request):
  # do something...
  with open('file_name.txt') as f:
    c = f.read()
  return
HttpResponse(c)

這種方式簡單粗暴,適合小檔案的下載,但如果這個檔案非常大,這種方式會佔用大量的記憶體,甚至導致伺服器崩潰。

更合理的檔案下載功能

Django的HttpResponse物件允許將迭代器作為傳入引數,將上面程式碼中的傳入引數c換成一個迭代器,便可以將上述下載功能優化為對大小檔案均適合;而Django更進一步,推薦使用 StreamingHttpResponse物件取代HttpResponse物件,StreamingHttpResponse物件用於將檔案流傳送給瀏覽器,與HttpResponse物件非常相似,對於檔案下載功能,使用StreamingHttpResponse物件更合理。

因此,更加合理的檔案下載功能,應該先寫一個迭代器,用於處理檔案,然後將這個迭代器作為引數傳遞給StreaminghttpResponse物件,如:

from django.http import StreamingHttpResponse

def big_file_download(request):
  # do something...

  def file_iterator(file_name, chunk_size=512):
    with open(file_name) as f:
      while True:
        c = f.read(chunk_size)
        if
c: yield c else: break the_file_name = "file_name.txt" response = StreamingHttpResponse(file_iterator(the_file_name)) return response

檔案下載功能再次優化

上述的程式碼,已經完成了將伺服器上的檔案,通過檔案流傳輸到瀏覽器,但檔案流通常會以亂碼形式顯示到瀏覽器中,而非下載到硬碟上,因此,還要在做點優化,讓檔案流寫入硬碟。優化很簡單,給StreamingHttpResponse物件的Content-Type和Content-Disposition欄位賦下面的值即可,如:

response['Content-Type'] = 'application/octet-stream'
response['Content-Disposition'] = 'attachment;filename="test.pdf"'

完整程式碼如下:

from django.http import StreamingHttpResponse

def big_file_download(request):
  # do something...

  def file_iterator(file_name, chunk_size=512):
    with open(file_name) as f:
      while True:
        c = f.read(chunk_size)
        if c:
          yield c
        else:
          break

  the_file_name = "big_file.pdf"
  response = StreamingHttpResponse(file_iterator(the_file_name))
  response['Content-Type'] = 'application/octet-stream'
  response['Content-Disposition'] = 'attachment;filename="{0}"'.format(the_file_name)

  return response