django – 仅接受服务器端FileField中的某种文件类型

1. Django中的FileField

Django是一个流行的Python Web框架，它提供了许多功能强大的功能来处理文件上传和处理。其中之一是FileField，它是一种模型字段，用于在数据库中存储文件的引用。

使用FileField字段，您可以轻松地将上传的文件保存到服务器上的特定位置，并将其路径保存在数据库中。然后，您可以根据需要在模板中访问这些文件。

2. 限制文件类型

默认情况下，Django的FileField字段可以接受任何类型的文件。但是，在某些情况下，我们可能只想接受特定类型的文件。例如，在一个图片上传应用程序中，我们只想接受图像文件（如JPEG或PNG），而不是其他类型的文件。

要实现这样的限制，我们可以使用Django的validators模块。validators模块提供了一组方便的验证器，可以用于验证模型字段的值。我们可以通过创建一个自定义验证器来限制FileField字段的文件类型。

2.1 创建自定义验证器

我们可以通过继承FileExtensionValidator类来创建一个自定义验证器，将其应用于FileField字段。

from django.core.validators import BaseValidator

from django.forms.fields import FileField
from django.utils.deconstruct import deconstructible
@deconstructible
class AllowedFileTypesValidator(BaseValidator):
    message = '只允许上传%s类型的文件。'
    extension_message = '允许的文件类型为：%s。'
    code = 'invalid'
    def __init__(self, allowed_types):
        self.allowed_types = allowed_types
    def __call__(self, value):
        extension = value.name.split('.')[-1].lower()
        if extension not in self.allowed_types:
            message = self.extension_message % ', '.join(self.allowed_types)
            raise ValidationError(self.message, code=self.code, params={'extension': extension})

在上面的代码中，我们创建了一个名为AllowedFileTypesValidator的自定义验证器。它接受一个allowed_types参数，该参数是一个允许的文件类型列表。

在验证函数中，我们从上传文件的名称中提取扩展名，并将其与allowed_types进行比较。如果扩展名不在allowed_types中，我们将引发一个ValidationError异常。

2.2 在模型中使用自定义验证器

要在模型中使用自定义验证器，我们可以在FileField字段的validators参数中传递AllowedFileTypesValidator的实例。

假设我们有一个名为FileUpload的模型，它具有一个名为file的FileField字段。让我们通过实例化AllowedFileTypesValidator来限制该字段只能接受JPEG和PNG文件。

from django.db import models

class FileUpload(models.Model):
    file = models.FileField(upload_to='uploads/', validators=[AllowedFileTypesValidator(['jpg', 'jpeg', 'png'])])

在上面的代码中，我们在FileField字段的validators参数中传递了AllowedFileTypesValidator的实例，并传递了一个包含允许的文件类型的列表。

现在，如果上传的文件不是JPEG或PNG类型，将引发一个ValidationError异常，并且文件将不会保存在服务器上。

3. 结论

通过使用Django的validators模块，我们可以轻松地实现对FileField字段的文件类型进行限制。通过创建自定义验证器，并将其应用于FileField字段，我们可以确保只接受特定类型的文件。

这对于需要限制上传文件类型的应用程序非常有用，例如图像上传应用程序。通过限制文件类型，我们可以确保用户只能上传允许的文件类型，并提供更好的用户体验。