django文件上传

1,226 阅读3分钟

文件上传

这是我参与8月更文挑战的第12天,活动详情查看:8月更文挑战

1、上传数据的存储

在保存上传的文件之前,数据需要保存到某处。
​
默认情况下,如果上传的文件小于2.5兆,Django 将把文件的所有内容保存到内存里。这意味着保存文件只涉及从内存中读取和写入磁盘,因此这很快。
​
但如果上传的文件很大,Django 会把文件写入系统临时目录的临时文件里存储。在类 Unix 平台里这意味着 Django 会生成一个类似名为 `/tmp/tmpzfp6I6.upload` 的文件。如果上传的文件非常大,你可以查看这个文件的大小增长,因为 Django 将数据流式传输到磁盘上
​
Django 处理文件上传时, 文件最终会位于 `request.FILES`

2、简单文件上传

考虑一个包含 FileField 的表单:

from django import forms
​
class UploadFile(forms.Form):
    file = forms.FileField()
处理这个表单的视图将通过:attr:request.FILES获取到文件数据,

注意

只有在请求是通过 POST 提交且提交的 <form> 表单有 enctype="multipart/form-data" 属性的时候,request.FILES才会包含文件数据,否则的话, request.FILES 是空的。

大多数情况下,你需要像这样将文件数据从 request 传递给表单。示例如下:

from django.http import HttpResponseRedirect
from django.shortcuts import render
from .forms import UploadFileForm
​
from somewhere import handle_uploaded_file
​
def upload_file(request):
    if request.method == 'POST':
        form = UploadFile(request.POST, request.FILES)
        if form.is_valid():
            handle_uploaded_file(request.FILES['file'])
            return HttpResponseRedirect('')
    else:  # GET
        form = UploadFile()
    return render(request, 'test.html', {'form': form})

注意我们必须将 request.FILES传入到表单的 构造方法中,只有这样文件数据才能绑定到表单中。

3、通过模型来处理上传的文件

如果想要在 FileField 上的 Model保存文件,使用 ModelForm会让这一过程变得简单。当调用 form.save() 时,文件对象将会被保存在对相应 FileFieldupload_to参数所指定的地方:

from django.http import HttpResponseRedirect
from django.shortcuts import render
from .forms import ModelFormWithFileField
​
def upload_file(request):
    if request.method == 'POST':
        form = ModelFormWithFileField(request.POST, request.FILES)
        if form.is_valid():
            form.save()
            return HttpResponseRedirect('')
    else:
        form = ModelFormWithFileField()
    return render(request, 'test.html', {'form': form})

如果你准备手工构建对象,你可以指定来自 request.FILES 的文件对象到模型里的文件对象:

from django.http import HttpResponseRedirect
from django.shortcuts import render
from .forms import UploadFileForm
from .models import ModelWithFileField
​
def upload_file(request):
    if request.method == 'POST':
        form = UploadFileForm(request.POST, request.FILES)
        if form.is_valid():
            instance = ModelWithFileField(file_field=request.FILES['file'])
            instance.save()
            return HttpResponseRedirect('/success/url/')
    else:
        form = UploadFileForm()
    return render(request, 'upload.html', {'form': form})

4、上传多个文件

如果你想使用一个表单字段上传多个文件,则需要设置字段的 widget 的 multiple=True HTML 属性。

from django import forms
​
class FileFieldForm(forms.Form):
    file_field = forms.FileField(widget=forms.ClearableFileInput(attrs={'multiple': True}))

然后覆盖 FormView子类的 post 方法来控制多个文件上传:

from django.views.generic.edit import FormView
from .forms import FileFieldForm
​
class FileFieldView(FormView):
    form_class = FileField
    template_name = 'test.html'  
    success_url = '...' 
​
    def post(self, request, *args, **kwargs):
        form_class = self.get_form_class()
        form = self.get_form(form_class)
        files = request.FILES.getlist('file_field')
        if form.is_valid():
            for f in files:
                pass
            return self.form_valid(form)
        else:
            return self.form_invalid(form)

5、上传 Handlers

当一个用户上传文件时,Django 会把文件数据传递给 upload handler —— 这是一个很小的类,它用来在上传时处理文件数据。上传处理模块最初定义在 FILE_UPLOAD_HANDLERS里,默认为:

["django.core.files.uploadhandler.MemoryFileUploadHandler",
 "django.core.files.uploadhandler.TemporaryFileUploadHandler"]

实例

MemoryFileUploadHandlerTemporaryFileUploadHandler 提供 Django 默认文件上传行为,小文件读入内存,大文件存在磁盘上。

你只能在访问 request.POSTrequest.FILES 之前修改上传处理程序,开始上传处理后修改上传处理程序没有意义。如果你从读取 request.POSTrequest.FILES 之后尝试修改 request.upload_handlers ,Django 会报错。

因此,你要尽早在视图里修改上传处理程序。

而且, request.POSTCsrfViewMiddleware 访问,默认情况下已开启。这意味着你需要在视图上使用 csrf_exempt() 来允许你改变上传处理程序。然后你需要在实际处理请求的函数上使用 csrf_protect()。注意这可能会让处理程序在 CSRF 检测完成之前开始接受文件上传。示例:

from django.views.decorators.csrf import csrf_exempt, csrf_protect
​
@csrf_exempt
def upload_file_view(request):
    request.upload_handlers.insert(0, ProgressBarUploadHandler(request))
    return _upload_file_view(request)
​
@csrf_protect
def _upload_file_view(request):
    pass