Django之文件下载

到目前为止，我们的注意力都在HTML网页上，但是实际上，在网站上除了HTML外还有图片，文件，PDF等等。

首先来看下返回一张图片为例，读取本地磁盘上的一张图片并返回到网页上显示。

def test1(request):
    image_data=open(r'd:/django_test2/code.JPG','rb').read()
    return HttpResponse(image_data,content_type='image/JPG')

在网页上可以看到我们的代码截图显示在浏览器上

接下来看下如何生成文件，在网站上经常要下载后台的文件或者是显示后台文件的内容。

下面通过HttpResponse的方法可以直接将读取的内容显示在网页上

def test1(request):
    with open(r'd:/django_test2/django.txt') as f:
        c=f.read()
    return HttpResponse(c)

但是这种方法只适合小文件，如果遇到大的文件则会很耗内存。

Django中提供了StreamingHttpResponse可以以流的方式进行下载。代码如下。

from django.http import StreamingHttpResponse

def test1(request):
    def file_itertor(file_name,chunk_size=512):
        with open(file_name) as f:
            while True:
                c=f.read(chunk_size)
                if c:
                    yield c
                else:
                    break
    file_name=r'd:/django_test2/django.txt'
    download_file='django.txt'
    response=StreamingHttpResponse(file_itertor(file_name))
    response['Content-Type'] = 'application/octet-stream'
    response['Content-Disposition'] = 'attachment;filename="{0}"'.format(download_file)
    return response

如果没有设置Content-Type和Content-Disposition,那么文件流会直接显示在浏览器上，如果要下载到硬盘上则必须设置Content-Type和Content-Disposition。其中download_file为下载的时候显示的文件名。

刷新网页的时候，会自动下载该文件

最后来看下PDF文档的生成。

首先要安装一个reportlab。pip install reportlab

生成代码如下：

def test1(request):
    response = HttpResponse(content_type='application/pdf')
    response['Content-Disposition'] = 'attachment; filename="django-pdf.pdf"'
    p = canvas.Canvas(response)
    p.drawString(50, 50, "Hello django.")
    p.showPage()
    p.save()
    return response

刷新在浏览器中自动下载生成的PDF文件

生成的pdf文件内容。