django的ImageField没有提供控制上传图片的内置方法,我们可以在表单验证的过程中用clean函数来控制,后来在网上查了资料,可以通过重写ImageField的方法来控制上传图片大小,现在整理出来:
1、在APP内建立fields.py文件
from django.db.models importImageField
from django.forms import forms
from django.template.defaultfilters import filesizeformat
class RImageFiled(ImageField):
def __init__(self, *args, **kwargs):
self.max_upload_size = kwargs.pop("max_upload_size", [])
super().__init__(*args, **kwargs)
def clean(self, *args, **kwargs):
data = super().clean(*args, **kwargs)
file = data.file
try:
if file.size > self.max_upload_size:
raise forms.ValidationError('上传图片大小不能超过 {}, 当期图片大小 {}。'
.format(filesizeformat(self.max_upload_size),
filesizeformat(file.size)))
except AttributeError:
pass
return data
2、在models.py中引用:
from activities.fields import RImageFiled
创建models的时候字段如下写就OK了,
image = RImageFiled(upload_to=upload_path_handler, max_length=256, max_upload_size=2621440, verbose_name='主题图片')
是不是很简洁。