小编典典

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

django

如何限制FileField以优雅的方式仅在服务器端接受某种类型的文件(视频,音频,pdf等)?


阅读 514

收藏
2020-03-29

共1个答案

小编典典

一种非常简单的方法是使用自定义验证器。

在你的应用程序中validators.py

def validate_file_extension(value):
    import os
    from django.core.exceptions import ValidationError
    ext = os.path.splitext(value.name)[1]  # [0] returns path+filename
    valid_extensions = ['.pdf', '.doc', '.docx', '.jpg', '.png', '.xlsx', '.xls']
    if not ext.lower() in valid_extensions:
        raise ValidationError('Unsupported file extension.')

然后在你的models.py:

from .validators import validate_file_extension

…并在表单字段中使用验证器:

class Document(models.Model):
    file = models.FileField(upload_to="documents/%Y/%m/%d", validators=[validate_file_extension])
2020-03-29