在保存对象之前处理文件上传

Question

我有一个这样的模型：

class Talk(BaseModel):
  title        = models.CharField(max_length=200)
  mp3          = models.FileField(upload_to = u'talks/', max_length=200)
  seconds      = models.IntegerField(blank = True, null = True)

我想在保存之前验证上传的文件是 MP3，如下所示：

def is_mp3(path_to_file):
  from mutagen.mp3 import MP3
  audio = MP3(path_to_file)
  return not audio.info.sketchy

一旦我确定我有 MP3，我想以秒为单位保存演讲的长度属性，如下所示：

audio = MP3(path_to_file)
self.seconds = audio.info.length

问题是，在保存之前，上传的文件没有路径（请参阅 这张票，已关闭为 wontfix），因此我无法处理 MP3。

我想提出一个很好的验证错误，以便 ModelForm 可以显示有用的错误（“你这个白痴，你没有上传 MP3”之类的）。

知道如何在保存文件之前访问该文件吗？

^{ps 如果有人知道验证 MP3 文件的更好方法，我洗耳恭听 - 我也希望能够处理 ID3 数据（设置艺术家、专辑、标题，可能还有专辑封面，所以我需要它可由诱变剂处理。}

Answer 1

您可以在 request.FILES 中访问文件数据 在你看来。

我认为最好的方法是将上传的文件绑定到表单，覆盖表单干净的方法，获取UploadedFile 对象 来自 clean_data，以您喜欢的方式验证它，然后覆盖 保存方法 并使用有关文件的信息填充模型实例，然后保存它。

Answer 2

在保存之前获取文件的更简洁的方法如下：

from django.core.exceptions import ValidationError

#this go in your class Model
def clean(self):
    try:
        f = self.mp3.file #the file in Memory
    except ValueError:
        raise ValidationError("A File is needed")
    f.__class__ #this prints <class 'django.core.files.uploadedfile.InMemoryUploadedFile'>
    processfile(f)

如果我们需要路径，答案是 在另一个问题中

Answer 3

您可以遵循 ImageField< 使用的技术/code>验证文件头，然后返回到文件的开头。

class ImageField(FileField):
    # ...    
    def to_python(self, data):
        f = super(ImageField, self).to_python(data)
        # ...
        # We need to get a file object for Pillow. We might have a path or we might
        # have to read the data into memory.
        if hasattr(data, 'temporary_file_path'):
            file = data.temporary_file_path()
        else:
            if hasattr(data, 'read'):
                file = BytesIO(data.read())
            else:
                file = BytesIO(data['content'])

        try:
            # ...
        except Exception:
            # Pillow doesn't recognize it as an image.
            six.reraise(ValidationError, ValidationError(
                self.error_messages['invalid_image'],
                code='invalid_image',
            ), sys.exc_info()[2])
        if hasattr(f, 'seek') and callable(f.seek):
            f.seek(0)
        return f