2016-08-09 6 views
0

Я создаю метод, который загружает файл, однако я хочу проверить размер файла, так как я просто хочу разрешить 5mb как максимальный предел.Как проверить ограничение размера файла в django

Я хочу сделать что-то вроде этого

def handle_uploaded_file(thisFile): 
    if thisFile > 5mb: 
     return "This file is more than 5mb" 
    else: 
     with open('some/file/' + str(thisFile), 'wb+') as destination: 
      for chunk in thisFile.chunks(): 
       destination.write(chunk) 
     return "File has successfully been uploaded" 
+0

Наряду с проверками Django, я также рекомендую добавить некоторые конфигурации на уровне сервера, чтобы ограничить размер загружаемых файлов (например, седений 'client_max_body_size' в Nginx). –

ответ

3

Использование ._size атрибут файла

if thisFile._size > 5242880: 
    return "This file is more than 5mb" 

._size представлена ​​в байтах. 5242880 - 5MB

def handle_uploaded_file(thisFile): 
    if thisFile._size > 5242880: 
     return "This file is more than 5mb" 
    else: 
     with open('some/file/' + str(thisFile), 'wb+') as destination: 
      for chunk in thisFile.chunks(): 
       destination.write(chunk) 
     return "File has successfully been uploaded" 
+0

Большое вам спасибо, но я получил это сообщение об ошибке unorderable types: int()> str() –

+0

@JamesReid Я исправил, мне плохо, вам нужно использовать 5242880 как int not string '5242880' – levi

+0

Большое вам спасибо. Это действительно большая помощь ... @levi –

4
# Add to your settings file 
CONTENT_TYPES = ['image', 'video'] 
# 2.5MB - 2621440 
# 5MB - 5242880 
# 10MB - 10485760 
# 20MB - 20971520 
# 50MB - 5242880 
# 100MB 104857600 
# 250MB - 214958080 
# 500MB - 429916160 
MAX_UPLOAD_SIZE = 5242880 

#Add to a form containing a FileField and change the field names accordingly. 
from django.template.defaultfilters import filesizeformat 
from django.utils.translation import ugettext_lazy as _ 
from django.conf import settings 
def clean_content(self): 
    content = self.cleaned_data['content'] 
    content_type = content.content_type.split('/')[0] 
    if content_type in settings.CONTENT_TYPES: 
     if content._size > settings.MAX_UPLOAD_SIZE: 
      raise forms.ValidationError(_('Please keep filesize under %s. Current filesize %s') % (filesizeformat(settings.MAX_UPLOAD_SIZE), filesizeformat(content._size))) 
    else: 
     raise forms.ValidationError(_('File type is not supported')) 
    return content 

Заслуга django snippet

+1

Будьте осторожны, 'MAX_UPLOAD_SIZE' является строкой, а' ._size' возвращает int. – levi

+0

Спасибо, что указали это, @levi. Я просто изменил MAX_UPLOAD_SIZE как int. – user4426017

Смежные вопросы