09-文件的上传与下载

d:\python\demo3

目标

  • 简介
  • 文件上传完整实例
  • 文件下载关键代码
  • 文件下载完整实例

简介

Django在处理文件上传的时候,文件数据被保存在request.FILES。
可以通过request.FILES['file']来读取数据
注意request.FILES 只有在请求方法为POST,并且发送请求的

拥有enctype="multipart/form-data" 属性时,才会包含数据。否则request.FILES 为空。
废话少说,开始编码

简单文件上传(略)

参考:http://www.ywnds.com/?p=11834

基于表单上传文件

  1. form.py

     class UploadFileForm(forms.Form):
         title = forms.CharField(max_length=50, label='标题')
         file = forms.FileField(label='文件')
    
  2. views.py

     from demo3.settings import BASE_DIR
     from demo3.form import RegisterForm, RegisterFormChinese,UploadFileForm
     import uuid
     
     FILE_DIR = BASE_DIR + '/files/'
     def upload(request):
         if request.method == 'POST':
             form = UploadFileForm(request.POST, request.FILES)
             if form.is_valid():
                 handle_uploaded_file(request.FILES['file'])
                 return render(request, 'success.html')
         else:
             form = UploadFileForm()
         return render(request, 'upload.html', {'form': form})
     
     
     def handle_uploaded_file(file):
         # file_path = FILE_DIR + str(uuid.uuid1())
         file_path = FILE_DIR + file.name
         with open(file_path, 'wb+') as destination:
             for chunk in file.chunks():
                 destination.write(chunk)
    
  3. urls.py里新增路径映射

     path('upload/',views.upload())
    
  4. upload.html

     
         {% csrf_token %}
         {{ form }}
  5. 工程里新建files目录


    09-文件的上传与下载_第1张图片
    image.png

测试地址:http://127.0.0.1:8000/forms/upload/

  1. 成功后看一下目录下的文件


    09-文件的上传与下载_第2张图片
    image.png

文件下载

基于Django建立的网站,如果提供文件下载功能,最简单的方式莫过于将静态文件交给Nginx等处理,但有些时候,由于网站本身逻辑,需要通过Django提供下载功能,如页面数据导出功能(下载动态生成的文件)、先检查用户权限再下载文件等。因此,有必要研究一下文件下载功能在Django中的实现。

  • 最简单的文件下载功能的实现
    将文件流放入HttpResponse对象即可,如:

def download_file(request):
# do something...
with open('/tmp/file_name.txt', 'rb') as f:
c = f.read()
return HttpResponse(c)
这种方式简单粗暴,适合小文件的下载,但如果这个文件非常大,这种方式会占用大量的内存,甚至导致服务器崩溃。

  • 合理的文件下载功能(未完待续)
    Django的HttpResponse对象允许将迭代器作为传入参数,将上面代码中的传入参数c换成一个迭代器,便可以将上述下载功能优化为对大小文件均适合;而Django更进一步,推荐使用 StreamingHttpResponse对象取代HttpResponse对象,StreamingHttpResponse对象用于将文件流发送给浏览器,与HttpResponse对象非常相似,对于文件下载功能,使用StreamingHttpResponse对象更合理。

因此,更加合理的文件下载功能,应该先写一个迭代器,用于处理文件,然后将这个迭代器作为参数传递给StreaminghttpResponse对象,如:

    from django.http import StreamingHttpResponse
    
    def download_file(request):
        def file_iterator(file, chunk_size=512):
            with open(file) as f:
                while True:
                    c = f.read(chunk_size)
                    if c:
                        yield c
                    else:
                        break
    
        file = "file_name.txt"
        response = StreamingHttpResponse(file_iterator(file))
        return response

文件下载功能再次优化

上述的代码,已经完成了将服务器上的文件,通过文件流传输到浏览器,但文件流通常会以乱码形式显示到浏览器中,而非下载到硬盘上,因此,还要在做点优化,让文件流写入硬盘。优化很简单,给StreamingHttpResponse对象的Content-Type和Content-Disposition字段赋下面的值即可,如:

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

所以,文件下载views.py完整代码如下:

from django.http import StreamingHttpResponse

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

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

代码实例

创建一个download.html,当访问/download/地址时,将上传后的文件在页面中以列表的形式显示出来,每一行列出上传文件的文件名、上传时间,可以点击文件名下载该文件。

  1. views.py

    from django.http import StreamingHttpResponse
    def download(request, filename):
    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_DIR + filename
    response = StreamingHttpResponse(file_iterator(the_file_name))
    response['Content-Type'] = 'application/octet-stream'
    response['Content-Disposition'] = 'attachment;filename="{0}"'.format(filename)
    return response

    def download_page(request):
    return render(request, 'download.html')

  2. download.html


    文件下载: test.txt 点我下载

  1. urls.py新增

    path('download/', views.download_page),
    path('', views.download)

  2. 测试,进入以入地址,点击下载

    http://127.0.0.1:8000/forms/download/

09-文件的上传与下载_第3张图片
image.png
image.png

你可能感兴趣的:(09-文件的上传与下载)