Чтение файлов данных во время очистки формы
Итак, я работаю над реализацией ответа на мой предыдущий вопрос.
Здесь моя модель:
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
Mutagen нужны файловые объекты или имена файлов на диске (я думаю) - первый случай (где загруженный файл больше чем размер файла, обрабатываемого в памяти) отлично работает, но я не знаю, как обрабатывать 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())
Что-то не так с мутагеном, или я делаю это неправильно?
После ответа rebus
Изменение настройки 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, когда я нажимаю submit:
Вы не можете установить обработчики загрузки после того, как загрузка была обработана.
хотя я делаю это как можно раньше!
Кроме того, я не уверен, что у меня есть метод save
для объекта, который я возвращаю (я смотрел в dir(self.files['mp3'].file)
и dir(self.files['mp3'])
).
Ответы
Ответ 1
Вы можете попытаться изменить FILE_UPLOAD_HANDLERS таким образом, чтобы Django всегда использовал обработчик временного файла:
FILE_UPLOAD_HANDLERS
default:
("django.core.files.uploadhandler.MemoryFileUploadHandler",
"django.core.files.uploadhandler.TemporaryFileUploadHandler",)
Таким образом, вы можете оставить только TemporaryFileUploadHandler
, переопределив настройку в настройках .py.
Edit:
Гораздо проще, должен был подумать об этом на первом месте: (:
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
.
Edit:
То же самое без экземпляра модели.
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')