Django框架中的文件上传技巧-Python教程-PHP中文网

Django框架中的文件上传技巧

WBOY

发布： 2023-06-18 08:24:06

原创

4617 人浏览过

近年来，Web应用程序逐渐流行，而其中许多应用程序都需要文件上传功能。在 Django 框架中，实现上传文件功能并不困难，但是在实际开发中，我们还需要处理上传的文件，其他操作包括更改文件名、限制文件大小等问题。本文将分享一些 Django 框架中的文件上传技巧。

一、配置文件上传项
在 Django 项目中，要配置文件上传需要在 settings.py 文件中进行配置。下面我们介绍其中的几个重要配置项。

文件上传路径
在 settings.py 中，我们可以定义存储上传文件的路径，如下：
```
MEDIA_ROOT = os.path.join(BASE_DIR, 'uploads')
```
登录后复制
在这个配置项中，我们定义了存储上传文件的根目录为 uploads，媒体文件的路径是相对于 settings.py 文件所在的目录计算的。我们还可以在 Django 项目的 urls.py 中添加相应的 URL 规则，这样用户就可以通过访问该 URL 来获取上传的文件。
支持文件格式
文件格式的控制是一个非常重要的内容，我们可以通过下面的配置项指定上传文件的格式。
```
ALLOWED_FILE_TYPES = ['jpg', 'jpeg', 'png', 'gif', 'pdf']
```
登录后复制
注意，我们必须将所有的文件格式转换成小写字母。
文件大小限制
Django 框架默认限制所有文件的大小都不能超过 2.5MB。如果我们需要修改这个限制值，可以在 settings.py 配置项中进行修改。
```
MAX_UPLOAD_SIZE = 10485760 # 10MB
```
登录后复制
在这个配置中，我们将最大上传文件大小限制为 10MB。

二、文件上传
我们可以通过 Django 表单组件实现文件上传。下面是一个简单的文件上传表单示例：

<form method="POST" enctype="multipart/form-data">
    {% csrf_token %}
    <input type="file" name="document" accept=".pdf">
    <input type="submit" value="Submit">
</form>

登录后复制

注意，enctype 属性必须为 multipart/form-data 才能支持文件上传。

在视图函数中，我们可以使用 request.FILES.get() 方法获取上传的文件。这个方法返回的是一个文件对象。下面是一个简单的示例：

def upload(request):
    if request.method == 'POST':
        file = request.FILES.get('document')
        if file:
            handle_uploaded_file(file)
            messages.success(request, '上传成功')
            return redirect('home')
        else:
            messages.error(request, '上传失败')
            return redirect('upload')
    else:
        return render(request, 'upload.html')

登录后复制

在这个示例中，我们首先获取上传的文件对象，然后调用 handle_uploaded_file() 函数将其保存到指定的路径。在没有上传文件的情况下，我们可以通过 messages 模块将错误信息提示给用户。

三、文件命名
为了更好地管理上传到服务器的文件，我们建议将上传的文件命名为含有时间戳的唯一文件名。下面是一个简单的实例：

def handle_uploaded_file(file):
    date = datetime.now().strftime("%Y%m%d%H%M%S")
    filename = f"{date}_{file.name}"
    path = os.path.join(MEDIA_ROOT, filename)
    with open(path, 'wb+') as destination:
        for chunk in file.chunks():
            destination.write(chunk)

登录后复制

首先，我们使用 datetime 模块获取当前日期和时间，并将其转换为字符串格式。然后，我们将这个时间戳和原始文件名结合生成一个新的文件名。最后，我们将新文件名与上传文件的路径相结合，并使用文件对象的 chunks() 方法将上传的文件内容写入文件中。

四、文件大小
为了防止服务器出现问题，我们需要对上传的文件大小进行限制。下面是一个简单的文件大小检查函数：

def check_file_size(file):
    if file.size > MAX_UPLOAD_SIZE:
        raise ValidationError(f'上传文件大小最大为 {MAX_UPLOAD_SIZE / 1048576} MB')

登录后复制

在这个函数中，我们检查上传文件的大小是否超过最大上传大小。如果超过了，我们就抛出 ValidationError 异常并提示用户。在视图函数中处理这个异常：

def upload(request):
    if request.method == 'POST':
        file = request.FILES.get('document')
        if file:
            try:
                check_file_size(file)
            except ValidationError as e:
                messages.error(request, e)
                return redirect('upload')
            else:
                handle_uploaded_file(file)
                messages.success(request, '上传成功')
                return redirect('home')
        else:
            messages.error(request, '上传失败')
            return redirect('upload')
    else:
        return render(request, 'upload.html')

登录后复制

通过这种方式，我们可以更好地控制上传文件的大小。

总结：