python - 在 django 中验证多个文件

Question

我想允许上传某些特定的文件类型。我为某个文件编写了以下代码，它有效。

def validate_file_extension(value):
    if not value.name.endswith('.zip'):
       raise ValidationError(u'Error message')

但我想允许多个文件，所以我在 settings_dev 中设置了这些文件，并编写了以下代码，但无法正常工作。

def validate_file_extension(value):
    for f in settings_dev.TASK_UPLOAD_FILE_TYPES:
        if not value.name.endswith(f):
           raise ValidationError(u'Error message')

Settings_dev

TASK_UPLOAD_FILE_TYPES=['.pdf','.zip','.docx']

楷模：

up_stuff=models.FileField(upload_to="sellings",validators=[validate_file_extension])

我该怎么办？

Answer 1

如果 中有多个（不同的）文件类型TASK_UPLOAD_FILE_TYPES，则for循环将始终引发异常。因为任何一种文件类型都不匹配。

您不需要使用for，因为str.endswith接受元组作为参数。

>>> 'data.docx'.endswith(('.pdf','.zip','.docx'))
True
>>> 'data.py'.endswith(('.pdf','.zip','.docx'))
False

---

def validate_file_extension(value):
    if not value.name.endswith(tuple(settings_dev.TASK_UPLOAD_FILE_TYPES)):
       raise ValidationError(u'Error message')