Django：验证上传文件的文件类型

  • Z时代
  • 2024-01-10
  • 分类：问答

我有一个可以让人们上传文件的应用程序，表示为UploadedFiles。但是，我要确保用户仅上传xml文件。我知道我可以使用进行此操作magic，但是我不知道将检查放在何处-据我所知，clean由于文件在clean运行时尚未上传，因此无法将其放在函数中。

这是UploadedFile模型：

class UploadedFile(models.Model):
    """This represents a file that has been uploaded to the server."""
    STATE_UPLOADED = 0
    STATE_ANNOTATED = 1
    STATE_PROCESSING = 2
    STATE_PROCESSED = 4
    STATES = (
        (STATE_UPLOADED, "Uploaded"),
        (STATE_ANNOTATED, "Annotated"),
        (STATE_PROCESSING, "Processing"),
        (STATE_PROCESSED, "Processed"),
    )
    status = models.SmallIntegerField(choices=STATES,
        default=0, blank=True, null=True) 
    file = models.FileField(upload_to=settings.XML_ROOT)
    project = models.ForeignKey(Project)
    def __unicode__(self):
        return self.file.name
    def name(self):
        return os.path.basename(self.file.name)
    def save(self, *args, **kwargs):
        if not self.status:
            self.status = self.STATE_UPLOADED
        super(UploadedFile, self).save(*args, **kwargs)
    def delete(self, *args, **kwargs):
        os.remove(self.file.path)
        self.file.delete(False)
        super(UploadedFile, self).delete(*args, **kwargs)
    def get_absolute_url(self):
        return u'/upload/projects/%d' % self.id
    def clean(self):
        if not "XML" in magic.from_file(self.file.url):
            raise ValidationError(u'Not an xml file.')
class UploadedFileForm(forms.ModelForm):
    class Meta:                
        model = UploadedFile
        exclude = ('project',)

回答：

对于后代：解决方案是使用read方法并将其传递给magic.from_buffer。

class UploadedFileForm(ModelForm):
    def clean_file(self):
        file = self.cleaned_data.get("file", False)
        filetype = magic.from_buffer(file.read())
        if not "XML" in filetype:
            raise ValidationError("File is not XML.")
        return file
    class Meta:
        model = models.UploadedFile
        exclude = ('project',)

以上是 Django：验证上传文件的文件类型 的全部内容， 来源链接： utcz.com/qa/433099.html

回到顶部