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='主題圖片')
是不是很簡潔。