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