54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copied from https://github.com/Chive/django-multiupload/blob/master/multiupload/fields.py
|
|
from django import forms
|
|
from django.core.exceptions import ValidationError
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
|
|
class MultiFileInput(forms.FileInput):
|
|
def render(self, name, value, attrs=None):
|
|
attrs['multiple'] = 'multiple'
|
|
return super(MultiFileInput, self).render(name, value, attrs)
|
|
|
|
def value_from_datadict(self, data, files, name):
|
|
if hasattr(files, 'getlist'):
|
|
return files.getlist(name)
|
|
else:
|
|
return [files.get(name)]
|
|
|
|
|
|
class MultiFileField(forms.FileField):
|
|
widget = MultiFileInput
|
|
default_error_messages = {
|
|
'min_num': _(u'Ensure at least %(min_num)s files are uploaded (received %(num_files)s).'),
|
|
'max_num': _(u'Ensure at most %(max_num)s files are uploaded (received %(num_files)s).'),
|
|
'file_size': _(u'File %(uploaded_file_name)s exceeded maximum upload size.'),
|
|
}
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.min_num = kwargs.pop('min_num', 0)
|
|
self.max_num = kwargs.pop('max_num', None)
|
|
self.maximum_file_size = kwargs.pop('max_file_size', None)
|
|
super(MultiFileField, self).__init__(*args, **kwargs)
|
|
|
|
def to_python(self, data):
|
|
ret = []
|
|
for item in data:
|
|
i = super(MultiFileField, self).to_python(item)
|
|
if i:
|
|
ret.append(i)
|
|
return ret
|
|
|
|
def validate(self, data):
|
|
super(MultiFileField, self).validate(data)
|
|
num_files = len(data)
|
|
if len(data) and not data[0]:
|
|
num_files = 0
|
|
if num_files < self.min_num:
|
|
raise ValidationError(self.error_messages['min_num'] % {'min_num': self.min_num, 'num_files': num_files})
|
|
elif self.max_num and num_files > self.max_num:
|
|
raise ValidationError(self.error_messages['max_num'] % {'max_num': self.max_num, 'num_files': num_files})
|
|
for uploaded_file in data:
|
|
if self.maximum_file_size and uploaded_file.size > self.maximum_file_size:
|
|
raise ValidationError(self.error_messages['file_size'] % {'uploaded_file_name': uploaded_file.name})
|