Django - Получить загруженный тип файла / mimetype - PullRequest
15 голосов
/ 31 января 2011

Есть ли способ получить тип содержимого загружаемого файла при перезаписи метода сохранения моделей? Я пробовал это:

def save(self):
    print(self.file.content_type)
    super(Media, self).save()

Но это не сработало. В этом примере self.file является моделью. FileField:

file = models.FileField(upload_to='uploads/%m-%Y/')

Редактировать: я хочу иметь возможность сохранять тип контента в базе данных, поэтому он мне понадобится до того, как сохранение будет фактически завершено:)

Ответы [ 2 ]

27 голосов
/ 31 января 2011
class MyForm(forms.ModelForm):

    def clean_file(self):
        file = self.cleaned_data['file']
        try:
            if file:
                file_type = file.content_type.split('/')[0]
                print file_type

                if len(file.name.split('.')) == 1:
                    raise forms.ValidationError(_('File type is not supported'))

                if file_type in settings.TASK_UPLOAD_FILE_TYPES:
                    if file._size > settings.TASK_UPLOAD_FILE_MAX_SIZE:
                        raise forms.ValidationError(_('Please keep filesize under %s. Current filesize %s') % (filesizeformat(settings.TASK_UPLOAD_FILE_MAX_SIZE), filesizeformat(file._size)))
                else:
                    raise forms.ValidationError(_('File type is not supported'))
        except:
            pass

        return file

settings.py

TASK_UPLOAD_FILE_TYPES = ['pdf', 'vnd.oasis.opendocument.text','vnd.ms-excel','msword','application',]
TASK_UPLOAD_FILE_MAX_SIZE = "5242880"
6 голосов
/ 31 января 2011

Согласно документам, ваш код вида, который обрабатывает загрузку формы, должен иметь возможность получить mimetype (с предупреждениями) в битах request.FILES

Последний кусок головоломки обработка фактических данных файла из request.FILES. Каждая запись в этом словарь является объектом UploadedFile - простая обертка вокруг загруженного файла. Вы обычно будете использовать один из них методы доступа к загруженному Содержание:

класс UploadedFile¶ В дополнение к все унаследованные от File, все Объекты UploadedFile определяют следующие методы / атрибуты:

UploadedFile.content_type заголовок типа контента, загруженный с файл (например, текст / простой или применение / PDF). Как и любые данные предоставленный пользователем, вы не должны доверять, что загруженный файл на самом деле этот тип. Вам все еще нужно проверить, что файл содержит контент, который заголовок типа контента претензии - "доверяй, но проверяй".

Это все здесь: http://docs.djangoproject.com/en/1.2/topics/http/file-uploads/ (хотя я вырезал пасту из документации выше ...)

Я не вижу content_type на модели FileField с в документах ...

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...