Чтение данных файла при чистом методе формы - PullRequest
Новая
       53

Чтение данных файла при чистом методе формы

16 голосов
/ 11 мая 2010

Итак, я работаю над реализацией ответа на мой предыдущий вопрос .

Вот моя модель: 

class Talk(models.Model):
  title        = models.CharField(max_length=200)
  mp3          = models.FileField(upload_to = u'talks/', max_length=200)

Вот моя форма: 

class TalkForm(forms.ModelForm):
  def clean(self):
    super(TalkForm, self).clean()
    cleaned_data = self.cleaned_data

    if u'mp3' in self.files:
      from mutagen.mp3 import MP3
      if hasattr(self.files['mp3'], 'temporary_file_path'):
        audio = MP3(self.files['mp3'].temporary_file_path())
      else:
        # What goes here?
        audio = None # setting to None for now
      ...
    return cleaned_data

  class Meta:
    model = Talk

Мутагену нужны файловые объекты или имена файлов на диске (я думаю ) - первый случай (когда размер загружаемого файла превышает размер файла, обрабатываемого в памяти) хорошо, но я не знаю, как справиться с InMemoryUploadedFile, что я получаю иначе. Я пробовал: 

# TypeError (coercing to Unicode: need string or buffer, InMemoryUploadedFile found)
audio = MP3(self.files['mp3'])

# TypeError (coercing to Unicode: need string or buffer, cStringIO.StringO found)
audio = MP3(self.files['mp3'].file)

# Hangs seemingly indefinitely on my test file (~800KB)
audio = MP3(self.files['mp3'].file.read())

Что-то не так с мутагеном, или я делаю это неправильно?

После ответа ребуса

Изменение настройки FILE_UPLOAD_HANDLERS на лету в моем классе ModelAdmin следующим образом: 

def add_view(self, request, form_url='', extra_context=None):
  request.upload_handlers = [TemporaryFileUploadHandler()]
  return super(TalkAdmin, self).add_view(request, form_url, extra_context)

Получает следующую ошибку 500, когда я нажимаю "Отправить":

Невозможно установить обработчики загрузки после обработки загрузки.

хотя я делаю это как можно раньше!

Кроме того, я не уверен, что у меня есть save метод для объекта, который я возвращаю (я смотрел в dir(self.files['mp3'].file) и dir(self.files['mp3'])).

1 Ответ

24 голосов
/ 11 мая 2010

Вы можете попробовать изменить FILE_UPLOAD_HANDLERS таким образом, чтобы Django всегда использовал обработчик временного файла:

FILE_UPLOAD_HANDLERS по умолчанию: 

("django.core.files.uploadhandler.MemoryFileUploadHandler",
 "django.core.files.uploadhandler.TemporaryFileUploadHandler",)

Таким образом, вы можете оставить только TemporaryFileUploadHandler, переопределив настройку в вашем settings.py.

Изменить:

Гораздо проще, должен был бы подумать об этом в первую очередь: (: 

from your.models import Talk
mp3 = self.files['mp3']
f = Talk.mp3.save('somename.mp3', mp3)
MP3(f.mp3.path)
>>> {'TRCK': TRCK(encoding=0, text=[u'5'])}

Таким способом вы можете сохранить InMemoryUploadedFile на диске, а затем использовать путь к этому файлу для работы с mutagen.

Изменить:

То же самое без экземпляра модели. 

import os

from django.core.files.storage import default_storage
from django.core.files.base import ContentFile
from django.conf import settings

from mutagen.mp3 import MP3

mp3 = request.FILES['mp3'] # or self.files['mp3'] in your form

path = default_storage.save('tmp/somename.mp3', ContentFile(mp3.read()))
MP3(os.path.join(settings.MEDIA_ROOT, path))

Обратите внимание, что он сохраняет файл в MEDIA_ROOT, когда я пытаюсь сохранить его в другом месте, я получаю SuspiciousOperation , поскольку существуют пределы, куда вы можете писать ... Вы должны удалить этот файл после его изучения угадайте, настоящая вещь будет на вашей модели ... 

path = default_storage.delete('tmp/somename.mp3')
...